Our new X account is live! Follow @wizwand_team for updates
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
InpaintingDIV2K 0.8k
PSNR28.81
14
InpaintingFFHQ 1k
PSNR31.37
14
DeblurringAFHQ-Cat 100 images (test)
PSNR28.97
10
DeblurringCelebA 100 images (test)
PSNR35.67
10
DenoisingCelebA 100 images (test)
PSNR33.14
10
Super-ResolutionCelebA 100 images (test)
PSNR33.09
10
DenoisingAFHQ-Cat 100 images (test)
PSNR32.35
10
Random InpaintingAFHQ-Cat 100 images (test)
PSNR33.7
10
Random InpaintingCelebA 100 images (test)
PSNR33.95
10
Super-ResolutionAFHQ-Cat 100 images (test)
PSNR26.57
10
Showing 10 of 23 rows

Other info

Follow for update