Search papers, labs, and topics across Lattice.
This paper introduces Distributionally Robust Neural Posterior Estimation (DRO-NPE) to address overconfidence and unreliability in neural posterior estimation (NPE) when simulation budgets are limited. DRO-NPE replaces the standard NPE objective with a worst-case loss over a Wasserstein ambiguity set, effectively controlling overfitting. Experiments on benchmark SBI tasks demonstrate that DRO-NPE improves coverage and calibration, narrowing the gap between empirical and population NPE loss, especially in low-simulation regimes.
Neural posteriors are often overconfident with limited data, but DRO-NPE offers a surprisingly simple fix: train against a worst-case loss within a Wasserstein ball.
Simulation-based inference with neural posterior estimation (NPE) often yields overconfident and unreliable posteriors under limited simulation budgets. To address this, we propose DRO-NPE, a distributionally robust approach that replaces the standard NPE objective with a worst-case loss over a Wasserstein ambiguity set. We introduce KL-based metrics for miscoverage and miscalibration, and use these to show that the DRO-NPE objective controls overfitting and reduces posterior overconfidence. Our method is tractable, parallelisable, and readily integrates with standard normalising flows. Across benchmark SBI tasks, DRO-NPE consistently improves coverage and calibration, while narrowing the gap between empirical and population NPE loss, leading to more reliable inference in low-simulation regimes.
