Efficient methods for topic model inference on streaming document collections

Abstract

Topic models provide a powerful tool for analyzing large text collections by representing high dimensional data in a low dimensional subspace. Fitting a topic model given a set of training documents requires approximate inference techniques that are computationally expensive. With today's large-scale, constantly expanding document collections, it is useful to be able to infer topic distributions for new documents without retraining the model. In this paper, we empirically evaluate the performance of several methods for topic inference in previously unseen documents, including methods based on Gibbs sampling, variational inference, and a new method inspired by text classification. The classification-based inference method produces results similar to iterative inference methods, but requires only a single matrix multiplication. In addition to these inference methods, we present SparseLDA, an algorithm and data structure for evaluating Gibbs sampling distributions. Empirical results indicate that SparseLDA can be approximately 20 times faster than traditional LDA and provide twice the speedup of previously published fast sampling methods, while also using substantially less memory.

DOI: 10.1145/1557019.1557121

Extracted Key Phrases

9 Figures and Tables

0204060200920102011201220132014201520162017
Citations per Year

310 Citations

Semantic Scholar estimates that this publication has 310 citations based on the available data.

See our FAQ for additional information.

Cite this paper

@inproceedings{Yao2009EfficientMF, title={Efficient methods for topic model inference on streaming document collections}, author={Limin Yao and David M. Mimno and Andrew McCallum}, booktitle={KDD}, year={2009} }