Abstract:With the emergence and development of deep generative models, such as the variational auto-encoders (VAEs), the research on topic modeling successfully extends to a new area: neural topic modeling, which aims to learn disentangled topics to understand the data better. However, the original VAE framework had been shown to be limited in disentanglement performance, bringing their inherent defects to a neural topic model (NTM). In this paper, we put forward that the optimization objectives of contrastive learning are consistent with two important goals (alignment and uniformity) of well-disentangled topic learning. Also, the optimization objectives of contrastive learning are consistent with two key evaluation measures for topic models, topic coherence and topic diversity. So, we come to the important conclusion that alignment and uniformity of disentangled topic learning can be quantified with topic coherence and topic diversity. Accordingly, we are inspired to propose the Contrastive Disentangled Neural Topic Model (CNTM). By representing both words and topics as low-dimensional vectors in the same embedding space, we apply contrastive learning to neural topic modeling to produce factorized and disentangled topics in an interpretable manner. We compare our proposed CNTM with strong baseline models on widely-used metrics. Our model achieves the best topic coherence scores under the most general evaluation setting (100% proportion topic selected) with 25.0%, 10.9%, 24.6%, and 51.3% improvements above the second-best models’ scores reported on four datasets of 20 Newsgroups, Web Snippets, Tag My News, and Reuters, respectively. Our method also gets the second-best topic diversity scores on the dataset of 20Newsgroups and Web Snippets. Our experimental results show that CNTM can effectively leverage the disentanglement ability from contrastive learning to solve the inherent defect of neural topic modeling and obtain better topic quality.

Coherence-Aware Neural Topic Modeling

Diversity-Aware Coherence Loss for Improving Neural Topic Models

A Model of Coherence Based on Distributed Sentence Representation.

Text Coherence Analysis Based on Deep Neural Network.

Neural Topic Model with Reinforcement Learning.

Visualizing and Understanding Neural Models in NLP

Is Automated Topic Model Evaluation Broken?: The Incoherence of Coherence

Extracting nonlinear neural topics with neural variational bayes

Improving Topic Disentanglement Via Contrastive Learning

Topic-based Coherence Modeling for Statistical Machine Translation

Neural Net Models for Open-Domain Discourse Coherence

A Discrete Variational Recurrent Topic Model without the Reparametrization Trick

TAN-NTM: Topic Attention Networks for Neural Topic Modeling

Modeling Coherence for Neural Machine Translation with Dynamic and Topic Caches

Rethinking Self-Supervision Objectives for Generalizable Coherence Modeling

Dependency-Aware Neural Topic Model

Nonparametric Topic Modeling with Neural Inference

Modeling Coherence for Discourse Neural Machine Translation

Topic Modeling Revisited: A Document Graph-based Neural Network Perspective

A Bayesian Nonparametric Topic Model with Variational Auto-Encoders

Neural Variational Correlated Topic Modeling