Abstract:Text classification is an important and classic application in natural language processing (NLP). Recent studies have shown that graph neural networks (GNNs) are effective in tasks with rich structural relationships and serve as effective transductive learning approaches. Text representation learning methods based on large-scale pretraining can learn implicit but rich semantic information from text. However, few studies have comprehensively utilized the contextual semantic and structural information for Chinese text classification. Moreover, the existing GNN methods for text classification did not consider the applicability of their graph construction methods to long or short texts. In this work, we propose Chinese-BERTology-wwm-GCN, a framework that combines Chinese bidirectional encoder representations from transformers (BERT) series models with whole word masking (Chinese-BERTology-wwm) and the graph convolutional network (GCN) for Chinese text classification. When building text graph, we use documents and words as nodes to construct a heterogeneous graph for the entire corpus. Specifically, we use the term frequency-inverse document frequency (TF-IDF) to construct the word-document edge weights. For long text corpora, we propose an improved pointwise mutual information (PMI*) measure for words according to their word co-occurrence distances to represent the weights of word-word edges. For short text corpora, the co-occurrence information between words is often limited. Therefore, we utilize cosine similarity to represent the word-word edge weights. During the training stage, we effectively combine the cross-entropy and hinge losses and use them to jointly train Chinese-BERTology-wwm and GCN. Experiments show that our proposed framework significantly outperforms the baselines on three Chinese benchmark datasets and achieves good performance even with few labeled training sets.

Chinese Text Classification Based on the BVB Model

Chinese Text Classification Using BERT and Flat-Lattice Transformer.

A multiclass classification framework for document categorization

A text classification method based on a convolutional and bidirectional long short-term memory model

Fast text categorization based on collaborative work in the semantic and class spaces

Collaborative Work with Linear Classifier and Extreme Learning Machine for Fast Text Categorization

A Chinese Text Classification Method Based on BERT and Convolutional Neural Network

Feature-enhanced text-inception model for Chinese long text classification

Chinese text classification by combining Chinese-BERTology-wwm and GCN

Feature-Enhanced Nonequilibrium Bidirectional Long Short-Term Memory Model for Chinese Text Classification

Chinese Text Classification Model Based on Deep Learning

A RBF Network for Chinese Text Classification Based on Concept Feature Extraction

Bag-of-Embeddings for Text Classification.

A Method for Chinese Text Classification Based on Three-Dimensional Vector Space Model

An Automatic Chinese-Text Classifier Based on Vector Space Model

Chinese Text Classification System on Regulatory Information Based on SVM

Attention-based BILSTM network with part-of-speech features for Chinese text classification

Chinese Text Classification Method Based on BERT Word Embedding