Applied Mathematics and Nonlinear Sciences
Journal license

Journal

Applied Mathematics and Nonlinear Sciences


Volume
& Issue

Volume 9, Issue 1


Published
on

July 2, 2024


Pages


DOI

Article

Zero-sample text classification algorithm based on BERT and graph convolutional neural network


Authors

Ying Qiao Affiliation:
School of Computer Science and Software Engineering, Southwest Petroleum University, Chengdu, Sichuan, 610500, China.
, Yu Li Affiliation:
School of Computer Science and Software Engineering, Southwest Petroleum University, Chengdu, Sichuan, 610500, China.
, Liangzhi Zhou Affiliation:
PetroChina Changqing Oilfield Company Oil Production PLANT NO.7, Xi’an, Shaanxi, 710000, China.
and Xu Shang Affiliation:
PetroChina Changqing Oilfield Company Oil Production PLANT NO.7, Xi’an, Shaanxi, 710000, China.


Abstract

In this study, we undertake a comprehensive examination of zero-shot text classification and its associated implications. We propose the adoption of the BERT model as a method for text feature representation. Subsequently, we utilize the Pointwise Mutual Information (PMI) metric to adjust the weight values within a graph convolutional neural network, thereby facilitating the construction of a text graph. Additionally, we incorporate an attention mechanism to transform this text graph, enabling it to represent the output labels of zero-shot text classification effectively. The experimental environment is set up, and the comparison and ablation experiments of the text classification model based on BERT and graph convolutional neural network with the baseline models are carried out in several different types of datasets, and the parameter settings of λ are adjusted according to the experimental results, and the convergence of the BERT model is compared to test the robustness of the model performance and the classification effect. When λ was set to 0.60, the model achieved the best results in each dataset. When the task is set to 5-way-5-shot, the convergence rate of the model for the Snippets dataset using the penultimate layer of features can reach 74%-80% of the training accuracy at the 5,000th step. The training accuracy gradually flattens out in the first 10,000 steps, and the model achieves classification accuracy in all four learning scenarios, with good stability.


Keywords

BERT model, Graph convolutional neural network, Attention mechanism, Baseline model, Text classification, 97P10


Citation

Qiao, Y., Li, Y., Zhou, L., & Shang, X. (2024). Zero-sample text classification algorithm based on BERT and graph convolutional neural network. Applied Mathematics and Nonlinear Sciences, 9(1). https://doi.org/10.2478/amns-2024-1560
1 Total citations
0.23 FWCI
1 Recent citations
(2 years)
23 References
Open Access Yes
View full metrics

Published by: Engineering Journals

Engineering Journals Logo