Gaussian Ensemble Belief Propagation for Efficient Inference in High-Dimensional Systems
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909486544846848 |
|---|---|
| author | MacKinlay, Dan Tsuchida, Russell Pagendam, Dan Kuhnert, Petra |
| author_facet | MacKinlay, Dan Tsuchida, Russell Pagendam, Dan Kuhnert, Petra |
| contents | Efficient inference in high-dimensional models is a central challenge in machine learning. We introduce the Gaussian Ensemble Belief Propagation (GEnBP) algorithm, which combines the strengths of the Ensemble Kalman Filter (EnKF) and Gaussian Belief Propagation (GaBP) to address this challenge. GEnBP updates ensembles of prior samples into posterior samples by passing low-rank local messages over the edges of a graphical model, enabling efficient handling of high-dimensional states, parameters, and complex, noisy, black-box generation processes. By utilizing local message passing within a graphical model structure, GEnBP effectively manages complex dependency structures and remains computationally efficient even when the ensemble size is much smaller than the inference dimension -- a common scenario in spatiotemporal modeling, image processing, and physical model inversion. We demonstrate that GEnBP can be applied to various problem structures, including data assimilation, system identification, and hierarchical models, and show through experiments that it outperforms existing belief propagation methods in terms of accuracy and computational efficiency.
Supporting code is available at https://github.com/danmackinlay/GEnBP |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2402_08193 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Gaussian Ensemble Belief Propagation for Efficient Inference in High-Dimensional Systems MacKinlay, Dan Tsuchida, Russell Pagendam, Dan Kuhnert, Petra Machine Learning 62-07 (Primary) 62F15, 62M40, 68T05, 68W25 I.2.6; H.2.4; I.2.8; J.2 Efficient inference in high-dimensional models is a central challenge in machine learning. We introduce the Gaussian Ensemble Belief Propagation (GEnBP) algorithm, which combines the strengths of the Ensemble Kalman Filter (EnKF) and Gaussian Belief Propagation (GaBP) to address this challenge. GEnBP updates ensembles of prior samples into posterior samples by passing low-rank local messages over the edges of a graphical model, enabling efficient handling of high-dimensional states, parameters, and complex, noisy, black-box generation processes. By utilizing local message passing within a graphical model structure, GEnBP effectively manages complex dependency structures and remains computationally efficient even when the ensemble size is much smaller than the inference dimension -- a common scenario in spatiotemporal modeling, image processing, and physical model inversion. We demonstrate that GEnBP can be applied to various problem structures, including data assimilation, system identification, and hierarchical models, and show through experiments that it outperforms existing belief propagation methods in terms of accuracy and computational efficiency. Supporting code is available at https://github.com/danmackinlay/GEnBP |
| title | Gaussian Ensemble Belief Propagation for Efficient Inference in High-Dimensional Systems |
| topic | Machine Learning 62-07 (Primary) 62F15, 62M40, 68T05, 68W25 I.2.6; H.2.4; I.2.8; J.2 |
| url | https://arxiv.org/abs/2402.08193 |