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

Flower: A Flow-Matching Solver for Inverse Problems

About

We introduce Flower, a solver for linear inverse problems. It leverages a pre-trained flow model to produce reconstructions that are consistent with the observed measurements. Flower operates through an iterative procedure over three steps: (i) a flow-consistent destination estimation, where the velocity network predicts a denoised target; (ii) a refinement step that projects the estimated destination onto a feasible set defined by the forward operator; and (iii) a time-progression step that re-projects the refined destination along the flow trajectory. We provide a theoretical analysis that demonstrates how Flower approximates Bayesian posterior sampling, thereby unifying perspectives from plug-and-play methods and generative inverse solvers. On the practical side, Flower achieves state-of-the-art reconstruction quality while using nearly identical hyperparameters across various linear inverse problems. Our code is available at https://github.com/mehrsapo/Flower.

Mehrsa Pourya, Bassam El Rawas, Michael Unser• 2025

Related benchmarks

TaskDatasetResultRank
Image InpaintingFFHQ (test)
LPIPS0.23
73
Super-ResolutionFFHQ (test)
SSIM42.3
32
Super-ResolutionCelebA (test)
PSNR33.09
30
DenoisingAFHQ Cat (test)
PSNR32.35
30
Super-ResolutionAFHQ Cat (test)
LPIPS0.249
26
Image DeblurringCelebA (test)
PSNR35.67
25
Random InpaintingAFHQ Cat (test)
PSNR33.7
24
Super-ResolutionFFHQ 1k
FID49.55
23
Box InpaintingAFHQ Cat (test)
PSNR26.88
20
InpaintingDIV2K 0.8k
PSNR28.81
14
Showing 10 of 38 rows

Other info

Follow for update