arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1906.02416stat.MLcs.CLcs.IRcs.LG

Sparse Parallel Training of Hierarchical Dirichlet Process Topic Models

Imperial College London(帝国理工学院);Uppsala University(乌普萨拉大学);Aalto University(阿尔托大学);Ericsson AB(爱立信公司);Linköping University(林雪平大学)

Alexander Terenin, Måns Magnusson, Leif Jonsson

英文摘要:

To scale non-parametric extensions of probabilistic topic models such as Latent Dirichlet allocation to larger data sets, practitioners rely increasingly on parallel and distributed systems. In this work, we study data-parallel training for the hierarchical Dirichlet process (HDP) topic model. Based upon a representation of certain conditional distributions within an HDP, we propose a doubly sparse data-parallel sampler for the HDP topic model. This sampler utilizes all available sources of sparsity found in natural language - an important way to make computation efficient. We benchmark our method on a well-known corpus (PubMed) with 8m documents and 768m tokens, using a single multi-core machine in under four days.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑