FUSE: FK-Steered Multi-Modal Flow Matching for Efficient Simulation-Based Posterior Estimation
摘要
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 state-of-the-art baselines on standard SBI benchmarks, producing posteriors that closely match ground-truth MCMC. Furthermore, in a real-world 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.
引用
@article{arxiv.2607.05252,
title = {FUSE: FK-Steered Multi-Modal Flow Matching for Efficient Simulation-Based Posterior Estimation},
author = {Weichen Qin and Yufan Xie and Peihao Wang and Chia-Jui Chou and Minghui Du and Peng Xu and Ziren Luo and Yi Yang and Jingyi Yu and Bo Liang and Jiakai Zhang},
journal= {arXiv preprint arXiv:2607.05252},
year = {2026}
}
备注
Accepted to the 43rd International Conference on Machine Learning (ICML 2026). 22 pages, 5 figures