Inductive Domain Transfer In Misspecified Simulation-Based Inference
Ortal Yona Senouf, Antoine Wehenkel, Cédric Vincent-Cuaz, Emmanuel Abbe, Pascal Frossard
Abstract
Simulation-based inference (SBI) is a statistical inference approach for estimating latent parameters of a physical system when the likelihood is intractable but simulations are available. In practice, SBI is often hindered by model misspecification--the mismatch between simulated and real-world observations caused by inherent modeling simplifications. RoPE, a recent SBI approach, addresses this challenge through a two-stage domain transfer process that combines semi-supervised calibration with optimal transport (OT)-based distribution alignment. However, RoPE operates in a fully transductive setting, requiring access to a batch of test samples at inference time, which limits scalability and generalization. We propose here a fully inductive and amortized SBI framework that integrates calibration and distributional alignment into a single, end-to-end trainable model. Our method leverages mini-batch OT with a closed-form coupling to align real and simulated observations that correspond to the same latent parameters, using both paired calibration data and unpaired samples. A conditional normalizing flow is then trained to approximate the OT-induced posterior, enabling efficient inference without simulation access at test time. Across a range of synthetic and real-world benchmarks--including complex medical biomarker estimation--our approach matches or surpasses the performance of RoPE, as well as other standard SBI and non-SBI estimators, while offering improved scalability and applicability in challenging, misspecified environments.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on7
- Unbalanced minibatch Optimal Transport; applications to Domain AdaptationKilian Fatras, Thibault Séjourné, Rémi Flamary, Nicolas CourtyICML 2021 · 183 citations
- Augmenting Physical Models with Deep Networks for Complex Dynamics ForecastingYuan Yin, Vincent Le Guen, Jérémie Donà, Emmanuel de Bézenac et al.ICLR 2021 · 165 citations
- Physics-Integrated Variational Autoencoders for Robust and Interpretable Generative ModelingNaoya Takeishi, Alexandros KalousisNeurIPS 2021 · 88 citations
- Robust Neural Posterior Estimation and Statistical Model CriticismDaniel Ward, Patrick Cannon, Mark Beaumont, Matteo Fasiolo et al.NeurIPS 2022 · 79 citations
- Neural Approximate Sufficient Statistics for Implicit ModelsYanzhi Chen, Dinghuai Zhang, Michael U. Gutmann, Aaron C. Courville et al.ICLR 2021 · 21 citations
Related papers
- Addressing Misspecification in Simulation-based Inference through Data-driven CalibrationAntoine Wehenkel, Juan L. Gamella, Ozan Sener, Jens Behrmann et al.ICML 2025
- Flow Matching for Scalable Simulation-Based InferenceJonas Wildberger, Maximilian Dax, Simon Buchholz, Stephen R. Green et al.NeurIPS 2023 · 153 citations
- The Curse of Conditions: Analyzing and Improving Optimal Transport for Conditional Flow-Based GenerationHo Kei Cheng, Alexander Gerhard SchwingICCV 2025 · 1 citation
- Faster Inference of Flow-Based Generative Models via Improved Data-Noise CouplingAram Davtyan, Leello Tadesse Dadi, Volkan Cevher, Paolo FavaroICLR 2025
- Robust Amortized Bayesian Inference with Self-Consistency Losses on Unlabeled DataAayush Mishra, Daniel Habermann, Marvin Schmitt, Stefan T. Radev et al.ICLR 2026 · 14 citations
