Efficient Training of LDA on a GPU by Mean-for-Mode Estimation

Jean-Baptiste Tristan, Joseph Tassarotti, Guy L. Steele Jr.

Abstract: We introduce Mean-for-Mode estimation, a variant of an uncollapsed Gibbs sampler that we use to train LDA on a GPU. The algorithm combines benefits of both uncollapsed and collapsed Gibbs samplers....