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

FlowADMM: Plug-and-play ADMM with Flow-based Renoise-Denoise Priors

About

Plug-and-play (PnP) methods for solving inverse problems have recently achieved strong performance by leveraging denoising priors based on powerful generative diffusion and flow models. However, existing diffusion- and flow-based PnP methods typically rely on stochastic renoise-denoise operations, which complicate the analysis of their convergence behavior. In this work, we identify and formalize the deterministic renoise-denoise operator underlying flow-based plug-and-play methods. This perspective reveals that these methods implicitly define a deterministic operator given by the expectation of a denoiser over the latent noise distribution. Building on this insight, we propose FlowADMM, a PnP algorithm that integrates the renoise-denoise operator into the classical alternating direction method of multiplier (ADMM) framework. We establish convergence guarantees for FlowADMM under weak Lipschitz conditions on the underlying flow network, and extend the analysis to non-stationary time schedules. Empirically, FlowADMM achieves state-of-the-art performance among flow-based PnP methods on a range of inverse problems, including denoising, deblurring, super-resolution, and inpainting, while requiring fewer data consistency evaluations than prior approaches.

Hendrik Sommerhoff, Michael Moeller• 2026

Related benchmarks

TaskDatasetResultRank
DenoisingAFHQ Cat (test)
PSNR32.9
30
Super-ResolutionCelebA (test)
PSNR33.94
30
Super-ResolutionAFHQ Cat (test)
LPIPS0.199
26
Image DeblurringCelebA (test)
PSNR35.53
25
Random InpaintingAFHQ Cat (test)
PSNR34.82
24
Box InpaintingAFHQ Cat (test)
PSNR27.39
20
Super-ResolutionCelebA 100 images (test)
PSNR21.28
13
DenoisingCelebA (test)
PSNR33.19
11
Random InpaintingCelebA (test)
PSNR35.23
11
Box InpaintingCelebA (test)
PSNR31.72
9
Showing 10 of 11 rows

Other info

Follow for update