Affiliations: [a] Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming, China
| [b] Yunnan Key Laboratory of Artificial Intelligence, Kunming University of Science and Technoloy, Kunming, China
Correspondence:
[*]
Corresponding author. Dangguo Shao, Kunming University of Science and Technology, Kunming, 650500, China. E-mail: [email protected].
Abstract: Aiming at the low effectiveness of short texts feature extraction, this paper proposes a short texts classification model based on the improved Wasserstein-Latent Dirichlet Allocation (W-LDA), which is a neural network topic model based on the Wasserstein Auto-Encoder (WAE) framework. The improvements of W-LDA are as follows: Firstly, the Bag of Words (BOW) input in the W-LDA is preprocessed by Term Frequency–Inverse Document Frequency (TF-IDF); Subsequently, the prior distribution of potential topics in W-LDA is replaced from the Dirichlet distribution to the Gaussian mixture distribution, which is based on the Variational Bayesian inference; And then the sparsemax function layer is introduced after the hidden layer inferred by the encoder network to generate a sparse document-topic distribution with better topic relevance, the improved W-LDA is named the Sparse Wasserstein-Variational Bayesian Gaussian mixture model (SW-VBGMM); Finally, the document-topic distribution generated by SW-VBGMM is input to BiGRU (Bidirectional Gating Recurrent Unit) for the deep feature extraction and the short texts classification. Experiments on three Chinese short texts datasets and one English dataset represent that our model is better than some common topic models and neural network models in the four evaluation indexes (accuracy, precision, recall, F1 value) of text classification.
Keywords: Short texts classification, neural network topic model, Variational Bayesian Gaussian mixture model (VBGMM), sparsemax, BiGRU (Bidirectional Gating Recurrent Unit)
