用确定性重加噪机制提升流模型的图像逆问题求解性能
FlowADMM: Plug-and-play ADMM with Flow-based Renoise-Denoise Priors

- 基于流模型构建确定性重加噪-去噪算子,替代随机过程
- 在弱利普希茨条件下证明算法收敛,支持非平稳调度
- 在去噪、超分等任务上超越现有流模型方法,评估次数更少
基于生成扩散和流模型的插件式(PnP)方法在求解逆问题上表现优异,但现有基于扩散与流模型的PnP方法通常依赖随机重加噪-去噪操作,使收敛性分析复杂化。本文揭示并形式化了流模型类PnP方法背后的确定性重加噪-去噪算子,发现其隐含定义为在潜在噪声分布上对去噪器的期望。基于此,提出将该算子嵌入经典交替方向乘子法(ADMM)框架的FlowADMM算法。在流网络满足弱利普希茨条件时建立收敛性保证,并扩展至非平稳时间调度情形。实验表明,FlowADMM在多种逆问题(包括去噪、去模糊、超分辨率、补全)中达到当前最优性能,且所需数据一致性评估次数少于先前方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。