Alleviating "Posterior Collapse" in Deep Topic Models via Policy Gradient
Yewen Li, Chaojie Wang, Zhibin Duan, Dongsheng Wang, Bo Chen, Bo An, Mingyuan Zhou
Abstract
Deep topic models have been proven as a promising way to extract hierarchical latent representations from documents represented as high-dimensional bag-of-words vectors. However, the representation capability of existing deep topic models is still limited by the phenomenon of "posterior collapse", which has been widely criticized in deep generative models, resulting in the higher-level latent representations exhibiting similar or meaningless patterns. To this end, in this paper, we first develop a novel deep-coupling generative process for existing deep topic models, which incorporates skip connections into the generation of documents, enforcing strong links between the document and its multi-layer latent representations. After that, utilizing data augmentation techniques, we reformulate the deep-coupling generative process as a Markov decision process and develop a corresponding Policy Gradient (PG) based training algorithm, which can further alleviate the information reduction at higher layers. Extensive experiments demonstrate that our developed methods can effectively alleviate "posterior collapse" in deep topic models, contributing to providing higher-quality latent document representations. Recently, benefiting from the development of deep neural networks (DNNs), there has been an emerging research interest to develop neural topic models (NTMs) to boost the performance, efficiency, and usability of topic modeling with DNNs. Specifically, following the framework of variational autoencoder (VAE) [10], most NTMs [11, 12, 13 ] construct a variational inference network (encoder) to project each document into its stochastic latent representation, and then reconstruct the corresponding BoW observation with a stochastic/deterministic decoder. By modeling the inference/generative process with DNNs, these NTMs are more flexible and scalable than traditional Bayesian PTMs, contributing to performing large-scale downstream tasks, especially in NLP tasks [14, 15] .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a836d325-7dc5-45cb-89a0-248c9ad5cbc9Cited by top-tier papers4
- Neural Topic Modeling with Large Language Models in the LoopXiaohao Yang, He Zhao, Weijie Xu, Yuanyuan Qi et al.ACL 2025 · 13 citations
- Bayesian Progressive Deep Topic Model with Knowledge Informed Textual Data Coarsening ProcessZhibin Duan, Xinyang Liu, Yudi Su, Yishi Xu et al.ICML 2023 · 7 citations
- Controlling Posterior Collapse by an Inverse Lipschitz Constraint on the Decoder NetworkYuri Kinoshita, Kenta Oono, Kenji Fukumizu, Yuichi Yoshida et al.ICML 2023 · 6 citations
- Improving Unsupervised Hierarchical Representation With Reinforcement LearningRuyi An, Yewen Li, Xu He, Pengjie Gu et al.CVPR 2024
Builds on6
- Representing Mixtures of Word Embeddings with Mixtures of Topic EmbeddingsDongsheng Wang, Dandan Guo, He Zhao, Huangjie Zheng et al.ICLR 2022 · 56 citations
- Sawtooth Factorial Topic Embeddings Guided Gamma Belief NetworkZhibin Duan, Dongsheng Wang, Bo Chen, Chaojie Wang et al.ICML 2021 · 49 citations
- Deep Relational Topic Modeling via Graph Poisson Gamma Belief NetworkChaojie Wang, Hao Zhang, Bo Chen, Dongsheng Wang et al.NeurIPS 2020 · 20 citations
- Knowledge-Aware Bayesian Deep Topic ModelDongsheng Wang, Yishi Xu, Miaoge Li, Zhibin Duan et al.NeurIPS 2022 · 19 citations
- Variational Hetero-Encoder Randomized GANs for Joint Image-Text ModelingHao Zhang, Bo Chen, Long Tian, Zhengjue Wang et al.ICLR 2020 · 9 citations
Related papers
- Neural Attention-Aware Hierarchical Topic ModelYuan Jin, He Zhao, Ming Liu, Lan Du et al.EMNLP 2021
- Effective Estimation of Deep Generative Language ModelsTom Pelsmaeker, Wilker AzizACL 2020 · 5 citations
- Topic Modeling Revisited: A Document Graph-based Neural Network PerspectiveDazhong Shen, Chuan Qin, Chao Wang, Zheng Dong et al.NeurIPS 2021 · 50 citations
- Tree-Structured Topic Modeling with Nonparametric Neural Variational InferenceZiye Chen, Cheng Ding, Zusheng Zhang, Yanghui Rao et al.ACL 2021
- S2WTM: Spherical Sliced-Wasserstein Autoencoder for Topic ModelingSuman Adhya, Debarshi Kumar SanyalACL 2025
