Skip to main navigation Skip to search Skip to main content

Attentive Representation Learning With Adversarial Training for Short Text Clustering

  • Wei Zhang*
  • , Chao Dong
  • , Jianhua Yin
  • , Jianyong Wang
  • *Corresponding author for this work
  • East China Normal University
  • Shandong University
  • Tsinghua University
  • Jiangsu Normal University

Research output: Contribution to journalArticlepeer-review

Abstract

Short text clustering has far-reaching effects on semantic analysis, showing its importance for multiple applications such as corpus summarization and information retrieval. However, it inevitably encounters the severe sparsity of short text representations, making the previous clustering approaches still far from satisfactory. In this paper, we present a novel attentive representation learning model for shot text clustering, wherein cluster-level attention is proposed to capture the correlations between text representations and cluster representations. Relying on this, the representation learning and clustering for short texts are seamlessly integrated into a unified model. To further ensure robust model training for short texts, we apply adversarial training to the unsupervised clustering setting, by injecting perturbations into the cluster representations. The model parameters and perturbations are optimized alternately through a minimax game. Extensive experiments on four real-world short text datasets demonstrate the superiority of the proposed model over several strong competitors, verifying that robust adversarial training yields substantial performance gains.

Original languageEnglish
Pages (from-to)5196-5210
Number of pages15
JournalIEEE Transactions on Knowledge and Data Engineering
Volume34
Issue number11
DOIs
StatePublished - 1 Nov 2022

Keywords

  • Short text clustering
  • attention mechanisms
  • representation learning
  • robust adversarial training

Fingerprint

Dive into the research topics of 'Attentive Representation Learning With Adversarial Training for Short Text Clustering'. Together they form a unique fingerprint.

Cite this