Neural Posterior Estimation

Overview

Neural posterior estimation trains a model to map data to a distribution over the parameters that could have generated it, making Bayesian inference possible when a simulator exists but its likelihood is difficult to compute. Common approaches use normalizing flows, autoregressive models, diffusion or score-based methods, and learned summaries; some compress parameters or data, split large datasets into shards, or combine neural proposals with conventional samplers. These methods are tested on simulation-based inference benchmarks and in settings such as imaging, epidemiology, dynamical systems, and cognitive models, where they can turn repeated inference into fast posterior sampling. A central challenge is ensuring calibration and useful joint uncertainty when simulations are limited or differ from real data, since accurate parameter marginals do not always guarantee accurate predictions or dependencies.

Latest papers

32 papers9 papers in the last four weeks, up 125% on the four weeks before. 0.1% of all new papers.

Jan 12, 2026

Adapting Neural Posterior Estimators for Simulation-Based Inference in Agent-Based Models

Agent-based models (ABMs) are widely used in economics and social sciences. They represent heterogeneous personal behaviors and their interactions by using simulations, whereas their likelihoods are often intractable for parameter estimations. Therefore, most traditional statistical inference techniques are hindered. Driven by recent advances in probabilistic machine learning, recent works have applied neural networks to achieve likelihood-free inference in large-scale ABMs. Specifically, Neural Posterior Estimation using normalizing flows provide amortized Bayesian inference from simulations, yet a pre-trained estimator may become inaccurate when the target task changes. In this work, we investigate the practicality of adapting a pre-trained conditional normalizing flow through a unified sequential neural posterior estimation test-time training (SNPE-TTT) framework. The framework includes four adaptation strategies: full-parameter adaptation, low-rank adaptation, and two update spaces identified from target-task gradients. We evaluate all four methods across economic simulation models with both tractable and intractable likelihoods. These benchmarks demonstrate that the sequential neural posterior estimation test-time training framework can adapt pre-trained posterior estimators to related inference tasks using a limited budget of target simulations. Posterior recovery and computational benefits depend on the task and adaptation strategy, with lower target-stage cost in some settings but no consistently superior posterior accuracy.
May 27, 2025

Causal Posterior Estimation

We present Causal Posterior Estimation (CPE), a novel method for Bayesian inference in simulator models, where evaluating the likelihood function is intractable or computationally expensive, but generating outputs given parameter values is straightforward. CPE approximates the posterior distribution using flow matching while directly incorporating the conditional dependence structure induced by the model's graphical representation into the neural network architecture. Across extensive experiments, we demonstrate that hard-coding these conditional dependencies into the network, rather than requiring them to be learned from data, enables CPE to achieve highly accurate posterior inference that matches or outperforms state-of-the-art baselines.