16 天前

文本级别的图神经网络用于文本分类

Lianzhe Huang, Dehong Ma, Sujian Li, Xiaodong Zhang, Houfeng WANG
文本级别的图神经网络用于文本分类
摘要

近年来,研究者们开始探索图神经网络(GNN)在文本分类中的应用,因为GNN在处理复杂结构以及保留全局信息方面表现出色。然而,以往基于GNN的方法主要面临两个实际问题:一是采用固定的语料库级图结构,难以支持在线测试;二是内存消耗较高。为解决上述问题,本文提出一种新型GNN模型,该模型为每篇输入文本动态构建图结构,并通过全局参数共享机制替代传统在整个语料库上使用单一图结构的方式。该方法消除了单个文本与整个语料库之间的强依赖关系,从而支持在线测试,同时仍能有效保留全局信息。此外,我们采用更小的文本窗口构建图结构,不仅能够提取更丰富的局部特征,还能显著减少边的数量,从而大幅降低内存开销。实验结果表明,所提模型在多个文本分类数据集上均优于现有方法,且在内存消耗更少的情况下实现了更优性能。

文本级别的图神经网络用于文本分类 | 最新论文 | HyperAI超神经