Learned harmonic mean estimation of the marginal likelihood for multimodal posteriors with flow matching

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Polanska, Alicja, McEwen, Jason D.
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914281329524736
author Polanska, Alicja
McEwen, Jason D.
author_facet Polanska, Alicja
McEwen, Jason D.
contents The marginal likelihood, or Bayesian evidence, is a crucial quantity for Bayesian model comparison but its computation can be challenging for complex models, even in parameters space of moderate dimension. The learned harmonic mean estimator has been shown to provide accurate and robust estimates of the marginal likelihood simply using posterior samples. It is agnostic to the sampling strategy, meaning that the samples can be obtained using any method. This enables marginal likelihood calculation and model comparison with whatever sampling is most suitable for the task. However, the internal density estimators considered previously for the learned harmonic mean can struggle with highly multimodal posteriors. In this work we introduce flow matching-based continuous normalizing flows as a powerful architecture for the internal density estimation of the learned harmonic mean. We demonstrate the ability to handle challenging multimodal posteriors, including an example in 20 parameter dimensions, showcasing the method's ability to handle complex posteriors without the need for fine-tuning or heuristic modifications to the base distribution.
format Preprint
id arxiv_https___arxiv_org_abs_2601_18683
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learned harmonic mean estimation of the marginal likelihood for multimodal posteriors with flow matching
Polanska, Alicja
McEwen, Jason D.
Methodology
Instrumentation and Methods for Astrophysics
Machine Learning
The marginal likelihood, or Bayesian evidence, is a crucial quantity for Bayesian model comparison but its computation can be challenging for complex models, even in parameters space of moderate dimension. The learned harmonic mean estimator has been shown to provide accurate and robust estimates of the marginal likelihood simply using posterior samples. It is agnostic to the sampling strategy, meaning that the samples can be obtained using any method. This enables marginal likelihood calculation and model comparison with whatever sampling is most suitable for the task. However, the internal density estimators considered previously for the learned harmonic mean can struggle with highly multimodal posteriors. In this work we introduce flow matching-based continuous normalizing flows as a powerful architecture for the internal density estimation of the learned harmonic mean. We demonstrate the ability to handle challenging multimodal posteriors, including an example in 20 parameter dimensions, showcasing the method's ability to handle complex posteriors without the need for fine-tuning or heuristic modifications to the base distribution.
title Learned harmonic mean estimation of the marginal likelihood for multimodal posteriors with flow matching
topic Methodology
Instrumentation and Methods for Astrophysics
Machine Learning
url https://arxiv.org/abs/2601.18683