Alert button

Diversity-Aware Coherence Loss for Improving Neural Topic Models

May 26, 2023
Raymond Li, Felipe González-Pizarro, Linzi Xing, Gabriel Murray, Giuseppe Carenini

Figure 1 for Diversity-Aware Coherence Loss for Improving Neural Topic Models
Figure 2 for Diversity-Aware Coherence Loss for Improving Neural Topic Models
Figure 3 for Diversity-Aware Coherence Loss for Improving Neural Topic Models
Figure 4 for Diversity-Aware Coherence Loss for Improving Neural Topic Models

Share this with someone who'll enjoy it:

The standard approach for neural topic modeling uses a variational autoencoder (VAE) framework that jointly minimizes the KL divergence between the estimated posterior and prior, in addition to the reconstruction loss. Since neural topic models are trained by recreating individual input documents, they do not explicitly capture the coherence between topic words on the corpus level. In this work, we propose a novel diversity-aware coherence loss that encourages the model to learn corpus-level coherence scores while maintaining a high diversity between topics. Experimental results on multiple datasets show that our method significantly improves the performance of neural topic models without requiring any pretraining or additional parameters.

* Minor Fixes, 11 pages, Camera-Ready for ACL 2023 (Short Paper)  
View paper onarxiv icon

Share this with someone who'll enjoy it: