Search papers, labs, and topics across Lattice.
This paper introduces ENFR (Evidence Networks with Flow-matched References), a novel amortized approach for estimating the absolute model evidence ($\mathcal{Z}$) in simulation-based inference. By learning Bayes factors with respect to a flow-matched reference, ENFR recovers $\mathcal{Z}$ accurately and efficiently, outperforming existing flow-based baselines and other density estimation methods. This enables robust, high-dimensional goodness-of-fit and model misspecification tests crucial for validating simulation-based parameter inference.
Amortized estimation of absolute model evidence is now achievable with ENFR, enabling direct, high-dimensional model misspecification tests without relying on Bayes factors alone.
The evidence, $\mathcal{Z} = p(x | \mathcal{M})$, of a data-generating model $\mathcal{M}$ is a powerful tool in Bayesian frameworks: it enables model inference, model misspecification tests, and goodness-of-fit assessment as an alternative to standard discrepancy statistics, e.g. $\chi^2$. While versatile, the evidence is notoriously difficult to compute, even when the likelihood is available. In both likelihood-based and simulation-based inference, estimating $\mathcal{Z}$ provides an immediate solution to high-dimensional goodness-of-fit or model misspecification tests: using $\mathcal{Z}(x)$ as the test statistic induces a minimum-volume level-$\alpha$ test. Such tests are essential to assess model fidelity, which is necessary to trust simulation-based parameter inference. Existing learned approaches largely target Bayes factors, which only measure evidence ratios. Instead, we provide accurate, fast, and amortized estimates of the absolute evidence. We present ENFR: Evidence Networks with Flow-matched References. Building on the Evidence Network framework of Jeffrey&Wandelt (2024), the proposed method first learns Bayes factors with respect to a flow-matched reference, from which the absolute evidence is recovered. We show improvement in estimation of $\mathcal{Z}$ over the flow-based baseline in all applications considered and the best performance of all density estimation methods considered in our benchmarks. We furthermore introduce and provide the proof for an architecture that jointly learns the individual evidences of a set of models $\{\mathcal{M}_i\}$. We demonstrate the performance and scaling behavior of these approaches on data from a variety of scientific applications.