Knowledge-Aware Bayesian Deep Topic Model
Dongsheng Wang, Yishi Xu, Miaoge Li, Zhibin Duan, Chaojie Wang, Bo Chen, Mingyuan Zhou
Abstract
We propose a Bayesian generative model for incorporating prior domain knowledge into hierarchical topic modeling. Although embedded topic models (ETMs) and its variants have gained promising performance in text analysis, they mainly focus on mining word co-occurrence patterns, ignoring potentially easy-to-obtain prior topic hierarchies that could help enhance topic coherence. While several knowledge-based topic models have recently been proposed, they are either only applicable to shallow hierarchies or sensitive to the quality of the provided prior knowledge. To this end, we develop a novel deep ETM that jointly models the documents and the given prior knowledge by embedding the words and topics into the same space. Guided by the provided knowledge, the proposed model tends to discover topic hierarchies that are organized into interpretable taxonomies. Besides, with a technique for adapting a given graph, our extended version allows the provided prior topic structure to be finetuned to match the target corpus. Extensive experiments show that our proposed model efficiently integrates the prior knowledge and improves both hierarchical topic discovery and document representation.
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 574f2803-15dd-44fb-a0d3-19f639b3c705Cited by top-tier papers6
- FASTopic: Pretrained Transformer is a Fast, Adaptive, Stable, and Transferable Topic ModelXiaobao Wu, Thong Nguyen, Delvin Zhang, William Yang Wang et al.NeurIPS 2024 · 67 citations
- HyperMiner: Topic Taxonomy Mining with Hyperbolic EmbeddingYishi Xu, Dongsheng Wang, Bo Chen, Ruiying Lu et al.NeurIPS 2022 · 38 citations
- On the Affinity, Rationality, and Diversity of Hierarchical Topic ModelingXiaobao Wu, Fengjun Pan, Thong Nguyen, Yichao Feng et al.AAAI 2024 · 35 citations
- Alleviating "Posterior Collapse" in Deep Topic Models via Policy GradientYewen Li, Chaojie Wang, Zhibin Duan, Dongsheng Wang et al.NeurIPS 2022 · 11 citations
- Context-guided Embedding Adaptation for Effective Topic Modeling in Low-Resource RegimesYishi Xu, Jianqiao Sun, Yudi Su, Xinyang Liu et al.NeurIPS 2023 · 9 citations
Builds on10
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- Adaptive Graph Convolutional Recurrent Network for Traffic ForecastingLei Bai, Lina Yao, Can Li, Xianzhi Wang et al.NeurIPS 2020 · 2,206 citations
- NVAE: A Deep Hierarchical Variational AutoencoderArash Vahdat, Jan KautzNeurIPS 2020 · 1,141 citations
- Iterative Deep Graph Learning for Graph Neural Networks: Better and Robust Node EmbeddingsYu Chen, Lingfei Wu, Mohammed J. ZakiNeurIPS 2020 · 559 citations
- Bayesian Attention ModulesXinjie Fan, Shujian Zhang, Bo Chen, Mingyuan ZhouNeurIPS 2020 · 78 citations
Related papers
- TopicNet: Semantic Graph-Guided Topic DiscoveryZhibin Duan, Yishi Xu, Bo Chen, Dongsheng Wang et al.NeurIPS 2021 · 19 citations
- Nonlinear Structural Equation Model Guided Gaussian Mixture Hierarchical Topic ModelingHegang Chen, Pengbo Mao, Yuyin Lu, Yanghui RaoACL 2023 · 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
- Bayesian Deep Embedding Topic Meta-LearnerZhibin Duan, Yishi Xu, Jianqiao Sun, Bo Chen et al.ICML 2022 · 7 citations
- Sawtooth Factorial Topic Embeddings Guided Gamma Belief NetworkZhibin Duan, Dongsheng Wang, Bo Chen, Chaojie Wang et al.ICML 2021 · 49 citations
