Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

We propose a machine-learning algorithm for Bayesian inverse problems in the function-space regime. Based on one-step generative transport, the method learns an amortized neural operator whose pushforward of a Gaussian source approximates the posterior distribution conditioned on each new observation. We show that white-noise sources are incompatible with the function-space limit, and therefore adopt a prior-aligned GRF as the source. We justify this choice through the Lipschitz regularity of the resulting one-step conditional posterior transport and numerical experiments on linear inverse and PDE-based inverse problems. The method is not distilled from MCMC: it is trained only with prior samples and simulated partial noisy observations. Once trained, it generates a $64\times64$ posterior sample in $\sim 10^{-3}$s, avoiding repeated forward-model evaluations in MCMC and repeated network evaluations in multistep generative samplers while matching key posterior summaries.

Zilan Cheng, Li-Lian Wang, Zhongjian Wang• 2026

Related benchmarks

TaskDatasetResultRank
Posterior SamplingAdvection
Time per Sample (s)3.5
6
Posterior SamplingDarcy
Time per Sample (s)3.5
6
Posterior SamplingReaction-diffusion
Time per Sample (s)3.5
6
Posterior SamplingNavier-Stokes
Time per Sample (s)3.5
6
Showing 4 of 4 rows

Other info

Follow for update