Distributed Bayesian Posterior Sampling via Moment Sharing
Xu, Minjie, Lakshminarayanan, Balaji, Teh, Yee Whye, Zhu, Jun, Zhang, Bo
–Neural Information Processing Systems
We propose a distributed Markov chain Monte Carlo (MCMC) inference algorithm for large scale Bayesian posterior simulation. We assume that the dataset is partitioned and stored across nodes of a cluster. Our procedure involves an independent MCMC posterior sampler at each node based on its local partition of the data. Moment statistics of the local posteriors are collected from each sampler and propagated across the cluster using expectation propagation message passing with low communication costs. The moment sharing scheme improves posterior estimation quality by enforcing agreement among the samplers.
Neural Information Processing Systems
Feb-14-2020, 12:41:29 GMT