FUSE: FK-Steered Multi-Modal Flow Matching for Efficient Simulation-Based Posterior Estimation
Weichen Qin, Yufan Xie, Peihao Wang, Chia-Jui Chou, Minghui Du, Peng Xu, Ziren Luo, Yi Yang, Jingyi Yu, Bo Liang, Jiakai Zhang
Abstract
Simulation-Based Inference (SBI) is critical for scientific discovery, with generative models offering a promising path toward efficient inference. However, existing methods struggle with effective multimodal modeling. They often rely on brute-force fusion strategies that ignore the structural disparities between parameters and observations, thus limiting estimation fidelity. In this work, we introduce FUSE (Feynman-Kac steered mUlti-modal flow matching for efficient Simulation-based posterior Estimation). Unlike prior work, FUSE employs a dual-track architecture that preserves the distinct features of multimodal inputs while facilitating dynamic interaction. Additionally, we propose an FK-steered sampling strategy that leverages intermediate observation likelihoods to guide the generative trajectories, effectively improving the sample quality during inference. Our approach outperforms stateof-the-art baselines on standard SBI benchmarks, producing posteriors that closely match reference MCMC posterior samples. Furthermore, in a realworld exoplanet orbital estimation task, FUSE successfully resolves complex parameter degeneracies that challenge existing methods, highlighting its potential to accelerate complex scientific discoveries in astrophysics and beyond.
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.
Builds on28
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Adding Conditional Control to Text-to-Image Diffusion ModelsLvmin Zhang, Anyi Rao, Maneesh AgrawalaICCV 2023 · 6,759 citations
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 5,568 citations
Related papers
- Flow Matching Calibration for Simulation-Based Inference under Model MisspecificationPierre-Louis Ruhlmann, Michael Arbel, Florence Forbes, Pedro Luiz Coelho RodriguesICML 2026 · 2 citations
- Flow Matching for Scalable Simulation-Based InferenceJonas Wildberger, Maximilian Dax, Simon Buchholz, Stephen R. Green et al.NeurIPS 2023 · 153 citations
- FNOPE: Simulation-based inference on function spaces with Fourier Neural OperatorsGuy Moss, Leah Sophie Muhle, Reinhard Drews, Jakob H. Macke et al.NeurIPS 2025 · 3 citations
- Tokenised Flow Matching for Hierarchical Simulation Based InferenceGiovanni Charles, Cosmo Santoni, Seth Flaxman, Elizaveta SemenovaICML 2026
- Addressing Misspecification in Simulation-based Inference through Data-driven CalibrationAntoine Wehenkel, Juan L. Gamella, Ozan Sener, Jens Behrmann et al.ICML 2025
