Efficient Inference for Dynamic Topic Modeling with Large Vocabularies

Abstract

Dynamic topic modeling is a well established tool for capturing the temporal dynamics of the topics of a corpus. Currently, dynamic topic models can only consider a small set of frequent words because of their computational complexity and insufficient data for less frequent words. In this work, we de- velop a scalable dynamic topic model by utilizing the correlation among the words in the vocabulary. By correlating previously independent temporal processes for words, our new model allows us to reliably estimate the topic representations contain- ing less frequent words. We develop an amortised variational inference method with self-normalised importance sampling approximation to the word distribution that dramatically reduces the compu- tational complexity and the number of variational parameters in order to handle large vocabularies. With extensive experiments on text datasets, we show that our method significantly outperforms the previous works by modeling word correlations, and it is able to handle real world data with a large vocabulary (80K words) which could not be pro- cessed by previous continuous dynamic topic mod- els. With qualitative analyses, we show that our method can perform inference on infrequent but representative keywords much more reliably than previous methods.

Related

March 2023 | Nature Machine Intelligence

Estimating categorical counterfactuals via deep twin networks

Athanasios Vlontzos, Bernhard Kainz, Ciarán M. Gilligan-Lee

February 2023 | WSDM

Calibrated Recommendations as a Minimum-Cost Flow Problem

Himan Abdollahpouri, Zahra Nazari, Alex Gain,Clay Giibson, Maria Dimakopoulou, Jesse Anderton, Benjamin Carterette, Mounia Lalmas, Tony Jebara

November 2022 | NeurIPS

Society of Agents: Regrets Bounds of Concurrent Thompson Sampling

Yan Chen, Perry Dong, Qinxun Bai, Maria Dimakopoulou, Wei Xu, Zhengyuan Zhou