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.

Compositional Recurrent Neural Networks for Chinese Short Text Classification

Bi-directional LSTM Recurrent Neural Network for Chinese Word Segmentation

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

Empirical Study on Character Level Neural Network Classifier for Chinese Text.

Long Short-Term Memory Neural Networks for Chinese Word Segmentation.

Hierarchical Convolutional Recurrent Neural Network for Chinese Text Classification

Hybrid Attention Networks for Chinese Short Text Classification

Chinese Text Classification Without Word Segmentation

Chinese Short Text Classification with Mutual-Attention Convolutional Neural Networks

Coattention-Based Recurrent Neural Networks for Sentiment Analysis of Chinese Texts.

A Chinese Character-Level and Word-Level Complementary Text Classification Method.

Chinese Text Classification Based on Hybrid Model of CNN and LSTM

Hyper-Gated Recurrent Neural Networks for Chinese Word Segmentation.

Neural Graph Matching Networks for Chinese Short Text Matching.

Chinese Text Classification Model Based on Deep Learning

LM Enhanced BiRNN-CRF for Joint Chinese Word Segmentation and POS Tagging.

Recurrent Neural Word Segmentation with Tag Inference

Word-character attention model for Chinese text classification

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

Character-level Neural Networks for Short Text Classification