Self-Taught convolutional neural networks for short text clustering.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Short text clustering is a challenging problem due to its sparseness of text representation. Here we propose a flexible Self-Taught Convolutional neural network framework for Short Text Clustering (dubbed STC), which can flexibly and successfully incorporate more useful semantic features and learn non-biased deep text representation in an unsupervised manner. In our framework, the original raw text features are firstly embedded into compact binary codes by using one existing unsupervised dimensionality reduction method. Then, word embeddings are explored and fed into convolutional neural networks to learn deep feature representations, meanwhile the output units are used to fit the pre-trained binary codes in the training process. Finally, we get the optimal clusters by employing K-means to cluster the learned representations. Extensive experimental results demonstrate that the proposed framework is effective, flexible and outperform several popular clustering methods when tested on three public short text datasets.

Authors

  • Jiaming Xu
    Institute of Automation, Chinese Academy of Sciences (CAS), Beijing, PR China.
  • Bo Xu
    State Key Laboratory of Cardiovascular Disease, Fuwai Hospital, National Center for Cardiovascular Diseases, Chinese Academy of Medical Sciences and Peking Union Medical College, Beijing 100037, China.
  • Peng Wang
    Neuroengineering Laboratory, School of Biomedical Engineering and Technology, Tianjin Medical University, Tianjin, China.
  • Suncong Zheng
    Institute of Automation, Chinese Academy of Sciences (CAS), Beijing, PR China.
  • Guanhua Tian
    Institute of Automation, Chinese Academy of Sciences (CAS), Beijing, PR China.
  • Jun Zhao