Topic Modeling as Multi-Objective Contrastive Optimization

Thong Nguyen,Xiaobao Wu,Xinshuai Dong,Cong-Duy T Nguyen,See-Kiong Ng,Anh Tuan Luu
2024-03-09
Abstract:Recent representation learning approaches enhance neural topic models by optimizing the weighted linear combination of the evidence lower bound (ELBO) of the log-likelihood and the contrastive learning objective that contrasts pairs of input documents. However, document-level contrastive learning might capture low-level mutual information, such as word ratio, which disturbs topic modeling. Moreover, there is a potential conflict between the ELBO loss that memorizes input details for better reconstruction quality, and the contrastive loss which attempts to learn topic representations that generalize among input documents. To address these issues, we first introduce a novel contrastive learning method oriented towards sets of topic vectors to capture useful semantics that are shared among a set of input documents. Secondly, we explicitly cast contrastive topic modeling as a gradient-based multi-objective optimization problem, with the goal of achieving a Pareto stationary solution that balances the trade-off between the ELBO and the contrastive objective. Extensive experiments demonstrate that our framework consistently produces higher-performing neural topic models in terms of topic coherence, topic diversity, and downstream performance.
Computer Science
What problem does this paper attempt to address?
This paper addresses the problem of neural topic models paying excessive attention to low-level common features and potential conflicts in contrastive learning. Traditional methods use document-level contrastive learning, which captures low-level mutual information and interferes with topic modeling. To solve these problems, the paper proposes two innovative approaches: 1) set-guided contrastive learning, which captures useful semantics through document sets and reduces the influence of low-level mutual information; 2) rephrasing contrastive topic modeling as a multi-objective optimization problem to find Pareto optimal solutions that balance multiple objectives. Experimental results show that these methods improve the coherence, diversity, and performance on downstream tasks of the topics.