Expectation propagation as a way of life: A framework for Bayesian inference on partitioned data

Loading...
Thumbnail Image
Journal Title
Journal ISSN
Volume Title
A1 Alkuperäisartikkeli tieteellisessä aikakauslehdessä
Date
2020
Major/Subject
Mcode
Degree programme
Language
en
Pages
1-53
Series
Journal of Machine Learning Research, Volume 21
Abstract
A common divide-and-conquer approach for Bayesian computation with big data is to partition the data, perform local inference for each piece separately, and combine the results to obtain a global posterior approximation. While being conceptually and computationally appealing, this method involves the problematic need to also split the prior for the local inferences; these weakened priors may not provide enough regularization for each separate computation, thus eliminating one of the key advantages of Bayesian methods. To resolve this dilemma while still retaining the generalizability of the underlying local inference method, we apply the idea of expectation propagation (EP) as a framework for distributed Bayesian inference. The central idea is to iteratively update approximations to the local likelihoods given the state of the other approximations and the prior. The present paper has two roles: we review the steps that are needed to keep EP algorithms numerically stable, and we suggest a general approach, inspired by EP, for approaching data partitioning problems in a way that achieves the computational benefits of parallelism while allowing each local update to make use of relevant information from the other sites. In addition, we demonstrate how the method can be applied in a hierarchical context to make use of partitioning of both data and parameters. The paper describes a general algorithmic framework, rather than a specific algorithm, and presents an example implementation for it.
Description
Keywords
Bayesian computation, Data partitioning, Expectation propagation, Hierarchical models, Statistical computing
Other note
Citation
Vehtari, A, Gelman, A, Sivula, T, Jylänki, P, Tran, D, Sahai, S, Blomstedt, P, Cunningham, J P, Schiminovich, D & Robert, C P 2020, ' Expectation propagation as a way of life : A framework for Bayesian inference on partitioned data ', Journal of Machine Learning Research, vol. 21, pp. 1-53 . < https://arxiv.org/abs/1412.4869 >