Back to Search Start Over

Semi-supervised Text Classification Based On Graph Attention Neural Networks

Authors :
Jian Huang
Nana Tao
Wei Wang
Qingshan Deng
Hui Chen
Jing Wang
Source :
2021 4th International Conference on Artificial Intelligence and Big Data (ICAIBD).
Publication Year :
2021
Publisher :
IEEE, 2021.

Abstract

The text classification task is the most basic and essential in natural language processing. It can find the rules about documents and distinguish the categories of different papers. Nowadays, graph convolutional neural network is widely used in text classification task because of its scalability and high efficiency. However, the GCN model has the best effect at layer 2. As the number of network layers increases, problems such as over smoothing will occur. This paper proposes a semi-supervised text classification model based on a graph attention neural network, which makes full use of the characteristics of information transmission along with the graph by adding residual connections to deepen the number of network layers. Simultaneously, the attention mechanism can be used to give different weights according to the importance of neighboring nodes to target nodes. A dual-layer attention mechanism at the topic level and node level is proposed to learn the importance of different neighboring nodes and extra neighboring node information to target nodes. Experimental results show that this model is superior to other methods on four benchmark datasets.

Details

Database :
OpenAIRE
Journal :
2021 4th International Conference on Artificial Intelligence and Big Data (ICAIBD)
Accession number :
edsair.doi...........c3a64108cb01926ee5c3e96da10aecd4
Full Text :
https://doi.org/10.1109/icaibd51990.2021.9459003