arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.13826cs.CL
收藏

每个文档都有自己的结构:基于图神经网络的归纳文本分类

Every Document Owns Its Structure: Inductive Text Classification via Graph Neural Networks

Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Xi’an Jiaotong University(西安交通大学)

Yufeng Zhang, Xueli Yu, Zeyu Cui, Shu Wu, Zhongzhen Wen, Liang Wang

摘要:

文本分类是自然语言处理的基础,图神经网络近年来被应用于文本分类。然而,现有的基于图的作品既不能捕捉每个文档中的上下文单词关系,也不能实现新单词的归纳学习。在这项工作中,为了克服这些问题,我们提出了通过GNN进行归纳文本分类的TextING。我们首先为每个文档构建单独的图,然后使用GNN根据它们的局部结构来学习细粒度的单词表示,这也可以有效地为新文档中的未看到的单词生成嵌入。最后,词节点被聚合作为文档嵌入。在四个基准数据集上的大量实验表明,我们的方法优于最先进的文本分类方法。

英文摘要:

Text classification is fundamental in natural language processing (NLP), and Graph Neural Networks (GNN) are recently applied in this task. However, the existing graph-based works can neither capture the contextual word relationships within each document nor fulfil the inductive learning of new words. In this work, to overcome such problems, we propose TextING for inductive text classification via GNN. We first build individual graphs for each document and then use GNN to learn the fine-grained word representations based on their local structures, which can also effectively produce embeddings for unseen words in the new document. Finally, the word nodes are aggregated as the document embedding. Extensive experiments on four benchmark datasets show that our method outperforms state-of-the-art text classification methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑