Text Classification Based on Graph Convolution Neural Network and Attention Mechanism
S Zhai, W Zhang, D Cheng, X Bai - Proceedings of the 2022 5th …, 2022 - dl.acm.org
S Zhai, W Zhang, D Cheng, X Bai
Proceedings of the 2022 5th International Conference on Artificial …, 2022•dl.acm.orgExtracting and representing text features is the most important part of text classification.
Aiming at the problem of incomplete feature extraction in traditional text classification
methods, a text classification model based on graph convolution neural network and
attention mechanism is proposed. Firstly, the text is input into BERT (Bi-directional Encoder
Representations from Transformers) model to obtain the word vector representation, the
context semantic information of the given text is learned by the BiGRU (Bi-directional Gated …
Aiming at the problem of incomplete feature extraction in traditional text classification
methods, a text classification model based on graph convolution neural network and
attention mechanism is proposed. Firstly, the text is input into BERT (Bi-directional Encoder
Representations from Transformers) model to obtain the word vector representation, the
context semantic information of the given text is learned by the BiGRU (Bi-directional Gated …
Extracting and representing text features is the most important part of text classification. Aiming at the problem of incomplete feature extraction in traditional text classification methods, a text classification model based on graph convolution neural network and attention mechanism is proposed. Firstly, the text is input into BERT (Bi-directional Encoder Representations from Transformers) model to obtain the word vector representation, the context semantic information of the given text is learned by the BiGRU (Bi-directional Gated Recurrent Unit), and the important information is screened by attention mechanism and used as node features. Secondly, the dependency syntax diagram and the corresponding adjacency matrix of the input text are constructed. Thirdly, the GCN (Graph Convolution Neural Network) is used to learn the node features and adjacency matrix. Finally, the obtained text features are input into the classifier for text classification. Experiments on two datasets show that the proposed model achieves a good classification effect, and better accuracy is achieved in comparison with baseline models.

Showing the best result for this search. See all results