arXiv Machine Learning

Preconditioned One-Step Generative Modeling for Bayesian Inverse Problems in Function Spaces

arXiv:2603. 14798v2 Announce Type: replace-cross Abstract: We propose a machine-learning algorithm for Bayesian inverse problems in the function-space regime.

arXiv AI
Jul 20

Energy-based Transport for Amortized Bayesian Inference

arXiv:2605. 15407v3 Announce Type: replace-cross Abstract: We consider amortized Bayesian inference for nonlinear inverse problems using only samples from the joint distribution of parameters and observations, including problems with unknown functions in a Banach space.

By Ricardo Baptista, Hojjat Kaveh, Andrew M. Stuart
arXiv Machine Learning
Sep 4

Learning Informative Prior with Infinite-Dimensional Continuous Normalizing Flow for Bayesian Inverse Problem

The paper introduces a continuous normalizing flow model for infinite-dimensional Bayesian inference in inverse problems governed by partial differential equations. By defining a neural ordinary differential equation in an infinite-dimensional Hilbert space, a simple reference measure is transformed into a complex prior that captures prior information. The authors establish a theoretical framework for well-posedness, present training methods for two data settings, and provide sampling algorithms, applying the approach to smooth, scattering, and heat conduction inverse problems with supporting numerical experiments.

By Yang Zhao, Junxiong Jia, Tao Zhou
arXiv Machine Learning
Sep 18

PosteriorBench: From Point Estimates to Posterior Matching in Evaluating Generative Inverse Solvers

PosteriorBench is a new benchmark that evaluates how well generative inverse solvers recover full posterior distributions rather than just a single reconstruction. It tests four physics-based inverse problems—Darcy flow inversion, Poisson source recovery, carbon capture and storage, and light transport material inference—using high-fidelity reference posteriors generated by rejection sampling and MCMC. The benchmark employs five metrics (posterior-mean error, posterior-standard-deviation error, maximum mean discrepancy, sliced Wasserstein distance, and radially averaged power-spectrum error) to assess pointwise accuracy, uncertainty, distributional alignment, and global frequency fidelity, revealing significant distribution-matching gaps in current solvers and highlighting the importance of neural operators, guidance weights, and generation noise for posterior-variance calibration.

By Jiachen Yao, Zi-Siang Hsu, Xi Deng, Aditi Gupta, Xin Ju, Sally M Benson, Gege Wen, Anima Anandkumar
arXiv AI
Aug 28

Active Diffusion-Based Inference for Ill-Posed Inverse Problems under Incomplete Priors

The paper introduces an active diffusion-based inverse problem solver that trains a diffusion model to map between parameter and observable spaces. By iteratively detecting and correcting model misspecification through posterior uncertainty, the method can discover and learn the correct parameter region even when initial training bounds exclude the true parameters. The authors demonstrate the solver on a toy inverse problem with infinite solutions and on parameterizing quantum correlation functions for a Quantum Chromodynamics analysis of nucleon structure.

By Jitao Xu, Nobuo Sato, Yaohang Li
arXiv Computer Vision
Sep 18

FlowSGS: Improving Flow Matching Priors for Inverse Imaging with Stochastic Interpolants

FlowSGS introduces a flow-based posterior sampling method that combines Split Gibbs Sampling (SGS) with Langevin dynamics for the likelihood step and Stochastic Interpolants (SI) for the prior step. By integrating a pretrained flow model into the prior step via SI's reverse-time SDE and a novel timestep correction, FlowSGS reduces the number of network evaluations compared to plug‑and‑play diffusion samplers. Experiments demonstrate state‑of‑the‑art performance on various inverse problems, including the first flow‑based solution to a nonlinear inverse problem (Fourier phase retrieval).

By Tianao Li, Xinhui Qian, Emma Alexander