login

Nested Hierarchical Dirichlet Processes

IEEE Transactions on Pattern Analysis and Machine IntelligencePublished 18 April 2014Open access
John Paisley, Chong Wang, David M. Blei, Michael I. Jordan
Citations216
View PDF

TL;DR

A stochastic variational inference algorithm is derived for the model, which enables efficient inference for massive collections of text documents and alleviates the rigid, single-path formulation of the nCRP.

Abstract

We develop a nested hierarchical Dirichlet process (nHDP) for hierarchical topic modeling. The nHDP generalizes the nested Chinese restaurant process (nCRP) to allow each word to follow its own path to a topic node according to a per-document distribution over the paths on a shared tree. This alleviates the rigid, single-path formulation assumed by the nCRP, allowing documents to easily express complex thematic borrowings. We derive a stochastic variational inference algorithm for the model, which enables efficient inference for massive collections of text documents. We demonstrate our algorithm on 1.8 million documents from The New York Times and 2.7 million documents from Wikipedia.

Keywords

Computer ScienceMathematics