Improving Contextualized Topic Models with Negative Sampling

Code

• github.com/adhyasuman/ctmneg
  OfficialIn paperpytorch★ 2
• github.com/adhyasuman/ntms_dropout_analysis
  pytorch★ 4

Abstract

Topic modeling has emerged as a dominant method for exploring large document collections. Recent approaches to topic modeling use large contextualized language models and variational autoencoders. In this paper, we propose a negative sampling mechanism for a contextualized topic model to improve the quality of the generated topics. In particular, during model training, we perturb the generated document-topic vector and use a triplet loss to encourage the document reconstructed from the correct document-topic vector to be similar to the input document and dissimilar to the document reconstructed from the perturbed vector. Experiments for different topic counts on three publicly available benchmark datasets show that in most cases, our approach leads to an increase in topic coherence over that of the baselines. Our model also achieves very high topic diversity.

Tasks

Reproductions