A CORRELATED TOPIC MODEL OF SCIENCE

被引:950
作者
Blei, David M. [1 ]
Lafferty, John D. [2 ]
机构
[1] Princeton Univ, Dept Comp Sci, Princeton, NJ 08540 USA
[2] Carnegie Mellon Univ, Dept Comp Sci, Machine Learning DEpt, Pittsburgh, PA 15213 USA
关键词
Hierarchical models; approximate posterior inference; variational methods; text analysis;
D O I
10.1214/07-AOAS114
中图分类号
O21 [概率论与数理统计]; C8 [统计学];
学科分类号
020208 ; 070103 ; 0714 ;
摘要
Topic models. such as latent Dirichlet allocation (LDA), call he useful tools for the statistical analysis of document collections and other discrete data. The LDA model assumes that the words of each document arise from a mixture of topics, each of which is it distribution over the vocabulary. 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 Sterns 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 [J. Roy. Statist. Soc. Ser. B 44 (1982) 139-177]. We derive a fast variational inference algorithm for approximate posterior inference in this model. which is complicated by the fact that the logistic normal is not conjugate to the multinomial. We apply the CTM to the articles from Science Published from 1990-1999, a data set that comprises 57M words. The CTM gives a better fit of the data than LDA, and we demonstrate its Use as an exploratory tool of large document collections.
引用
收藏
页码:17 / 35
页数:19
相关论文
共 31 条
  • [1] Airoldi EM, 2007, LECT NOTES COMPUT SC, V4503, P57
  • [2] AITCHISON J, 1982, J ROY STAT SOC B, V44, P139
  • [3] LOGISTIC-NORMAL DISTRIBUTIONS - SOME PROPERTIES AND USES
    AITCHISON, J
    SHEN, SM
    [J]. BIOMETRIKA, 1980, 67 (02) : 261 - 272
  • [4] AITCHISON J, 1985, J ROY STAT SOC B MET, V47, P136
  • [5] [Anonymous], AUTHOR RECIPIENT TOP
  • [6] BISHOP C, 2003, ADV NEURAL INFORM PR, V15, P777
  • [7] Blei D., 2003, P 26 ANN INT ACM SIG, P127, DOI DOI 10.1145/860435.860460
  • [8] Blei D. M, 2006, ADV NEURAL INFORM PR, V18
  • [9] Variational Inference for Dirichlet Process Mixtures
    Blei, David M.
    Jordan, Michael I.
    [J]. BAYESIAN ANALYSIS, 2006, 1 (01): : 121 - 143
  • [10] Latent Dirichlet allocation
    Blei, DM
    Ng, AY
    Jordan, MI
    [J]. JOURNAL OF MACHINE LEARNING RESEARCH, 2003, 3 (4-5) : 993 - 1022