Abstract

Topic models, such as latent Dirichlet allocation (LDA), can be useful tools for the statistical analysis of document collections and other dis- crete data. The LDA model assumes that the words of each document arise from a mixture of topics, each of which is a distribution over the vo- cabulary. A limitation of LDA is the inability to model topic correlation even though, for example, a document about genetics is more likely to also be about disease than x-ray astronomy. This limitation stems from the use of the Dirichlet distribution to model the variability among the topic proportions. In this paper we develop the correlated topic model (CTM), where the topic proportions exhibit correlation via the logistic normal distribution 1. We derive a mean-field variational inference al- gorithm for approximate posterior inference in this model, which is com- plicated by the fact that the logistic normal is not conjugate to the multi- nomial. The CTM gives a better fit than LDA on a collection of OCRed articles from the journal Science. Furthermore, the CTM provides a nat- ural way of visualizing and exploring this and other unstructured data sets.

Links and resources

Tags

community

  • @ghagerer
  • @ans
  • @tberg
  • @josephausterwei
@ans's tags highlighted