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

What Do Flow-Based Inverse Solvers Approximate? A Posterior-Transport View

About

A growing family of training-free solvers -- FlowDPS, FLOWER, PnP-Flow and their diffusion ancestors (DPS, DAPS) -- repurpose a pretrained flow-matching prior to solve imaging inverse problems by adding a measurement-guidance term to the deterministic probability-flow ODE. Despite strong empirical results, what these per-step corrections actually approximate -- and how far the resulting samples are from the true posterior $p(x\mid y)$ -- has not been characterized. We give a posterior-transport account of flow-based inverse problem solving. Our starting point is a simple but consequential fact: for a \emph{deterministic} flow prior, Bayesian conditioning is realized entirely by a \emph{reweighting of the source distribution}, not by a drift correction; pushing the reweighted source through the \emph{unmodified} velocity field yields exact posterior samples. From this we show that trajectory-guidance solvers can be read as the minimum-kinetic-energy \emph{correction} field needed to morph the unconditional source into the posterior, and that FlowDPS / FLOWER / PnP-Flow correspond to distinct zeroth-order / Gaussian / proximal approximations of this single object; we bound the resulting posterior bias in Wasserstein distance. A controlled $2$D study with a closed-form posterior confirms the theory decisively: source reweighting matches the true posterior to the Monte-Carlo floor on every metric, whereas trajectory guidance incurs $200$--$800\times$ larger error and collapses posterior modes, \emph{regardless of guidance strength}. Guided by the analysis we propose a cheap, principled velocity-correction solver that is competitive across two in-domain priors (AFHQ, CelebA) and two out-of-distribution settings while, unlike point-estimate source-space optimizers, producing diverse posterior samples with uncertainty that correlates with reconstruction error.

Jian Xu, Delu Zeng, John Paisley, Qibin Zhao• 2026

Related benchmarks

TaskDatasetResultRank
Super-ResolutionAFHQ Cat (test)
LPIPS0.122
31
Box InpaintingAFHQ Cat (test)
PSNR25.57
26
InpaintingCelebA 20 images 128x128
PSNR33.1
6
Gaussian deblurAFHQ Cat (test)
PSNR25.53
5
Super-ResolutionCelebA 20 images 128x128
PSNR30.9
4
InpaintingAFHQ Dog
LPIPS0.061
2
InpaintingAFHQ-Wild
LPIPS0.058
2
Super-ResolutionAFHQ Dog
LPIPS0.183
2
Super-ResolutionAFHQ-Wild
LPIPS0.195
2
DeblurringCelebA 128x128 (20 images)
PSNR31.5
2
Showing 10 of 10 rows

Other info

Follow for update