arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2003.11644cs.CLcs.LGstat.ML
收藏

基于注意力的图神经网络多标签文本分类

Multi-Label Text Classification using Attention-based Graph Neural Network

Saama AI Research(萨马人工智能研究院)

Ankit Pal, Muru Selvakumar, Malaikannan Sankarasubbu

摘要:

在多标签文本分类(MLTC)中,一个样本可以属于多个类。据观察,大多数MLTC任务,有依赖关系或相关性之间的标签。现有的方法往往忽略了标签之间的关系。在本文中,提出了一个基于图的注意力网络模型来捕捉标签之间的注意依赖结构。图注意力网络使用特征矩阵和相关矩阵来捕获和探索标签之间的关键依赖关系,并为任务生成分类器。将生成的分类器应用于从文本特征提取网络(BiLSTM)获得的句子特征向量,以实现端到端训练。注意力允许系统为每个标签的邻居节点分配不同的权重,从而允许它隐式地学习标签之间的依赖关系。所提出的模型的结果进行了验证五个真实世界的MLTC数据集。该模型实现了类似或更好的性能相比,以前的国家的最先进的模型。

英文摘要:

In Multi-Label Text Classification (MLTC), one sample can belong to more than one class. It is observed that most MLTC tasks, there are dependencies or correlations among labels. Existing methods tend to ignore the relationship among labels. In this paper, a graph attention network-based model is proposed to capture the attentive dependency structure among the labels. The graph attention network uses a feature matrix and a correlation matrix to capture and explore the crucial dependencies between the labels and generate classifiers for the task. The generated classifiers are applied to sentence feature vectors obtained from the text feature extraction network (BiLSTM) to enable end-to-end training. Attention allows the system to assign different weights to neighbor nodes per label, thus allowing it to learn the dependencies among labels implicitly. The results of the proposed model are validated on five real-world MLTC datasets. The proposed model achieves similar or better performance compared to the previous state-of-the-art models.

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

↑