主题模型中用于并行推断的稀疏部分折叠MCMC

Sparse Partially Collapsed MCMC for Parallel Inference in Topic Models

Journal of Computational and Graphical Statistics · 2017
被引 15
ABS 3

中文导读

提出一种稀疏部分折叠吉布斯采样器,用于主题模型的并行推断,在五个文本语料上实验表明,部分折叠带来的统计效率损失小于预期,且可通过并行化和稀疏性在大语料上获得更快的速度。

Abstract

Topic models, and more specifically the class of latent Dirichlet allocation (LDA), are widely used for probabilistic modeling of text. Markov chain Monte Carlo (MCMC) sampling from the posterior distribution is typically performed using a collapsed Gibbs sampler. We propose a parallel sparse partially collapsed Gibbs sampler and compare its speed and efficiency to state-of-the-art samplers for topic models on five well-known text corpora of differing sizes and properties. In particular, we propose and compare two different strategies for sampling the parameter block with latent topic indicators. The experiments show that the increase in statistical inefficiency from only partial collapsing is smaller than commonly assumed, and can be more than compensated by the speedup from parallelization and sparsity on larger corpora. We also prove that the partially collapsed samplers scale well with the size of the corpus. The proposed algorithm is fast, efficient, exact, and can be used in more modeling situations than the ordinary collapsed sampler. Supplementary materials for this article are available online.

主题模型潜在狄利克雷分配马尔可夫链蒙特卡洛并行计算文本建模