1. Research on few-shot text classification techniques based on text-level-graph neural networks
- Author
-
Xiangcheng AN, Baozhu LIU, and Jingwei GAN
- Subjects
natural language processing ,few-shot text classification ,pre-trained model ,graph neural network ,prototype network ,Technology - Abstract
In order to solve the problem of poor accuracy of text classification in text graph neural network with small samples, a text level graph neural network-prototypical (LGNN-Proto) was designed. An advanced pre-training language model was adopted, and the text graph neural network was used to construct the graph for each input text, then the global parameters were shared. The result of the text graph neural network was used as the input of the prototype network to classify the unlabeled text, and the validity of the new model on multiple text classification data sets was verified. The results show that the accuracy of unlabeled text classification is improved by 1% ~ 3% compared with that of supervised learning, which requires a large number of labeled documents, and the new model is validated on multiple text classification data sets with advanced performance and lower memory consumption. The research results can provide reference for solving the problem of text classification with small sample size.
- Published
- 2024
- Full Text
- View/download PDF