ClusterCluster: Parallel Markov Chain Monte Carlo for Dirichlet Process Mixtures
arXiv:1304.2302
Abstract
The Dirichlet process (DP) is a fundamental mathematical tool for Bayesian nonparametric modeling, and is widely used in tasks such as density estimation, natural language processing, and time series modeling. Although MCMC inference methods for the DP often provide a gold standard in terms asymptotic accuracy, they can be computationally expensive and are not obviously parallelizable. We propose a reparameterization of the Dirichlet process that induces conditional independencies between the atoms that form the random measure. This conditional independence enables many of the Markov chain transition operators for DP inference to be simulated in parallel across multiple cores. Applied to mixture modeling, our approach enables the Dirichlet process to simultaneously learn clusters that describe the data and superclusters that define the granularity of parallelization. Unlike previous approaches, our technique does not require alteration of the model and leaves the true posterior distribution invariant. It also naturally lends itself to a distributed software implementation in terms of Map-Reduce, which we test in cluster configurations of over 50 machines and 100 cores. We present experiments exploring the parallel efficiency and convergence properties of our approach on both synthetic and real-world data, including runs on 1MM data vectors in 256 dimensions.
12 pages, 10 figures. Submitted to ICML 2013 during third submission cycle
References in corpus (1)
Cited by in corpus (6)
- Distributed Variational Inference in Sparse Gaussian Process Regression and Latent Variable Models
- d-blink: Distributed End-to-End Bayesian Entity Resolution
- Parallel Clustering of Single Cell Transcriptomic Data with Split-Merge Sampling on Dirichlet Process Mixtures
- Optimistic Concurrency Control for Distributed Unsupervised Learning
- Accelerated Parallel Non-conjugate Sampling for Bayesian Non-parametric Models
- Distributed Collapsed Gibbs Sampler for Dirichlet Process Mixture Models in Federated Learning