Gradient Step Denoiser for convergent Plug-and-Play

Abstract

Plug-and-Play methods constitute a class of iterative algorithms for imaging problems where regularization is performed by an off-the-shelf denoiser. Although Plug-and-Play methods can lead to tremendous visual performance for various image problems, the few existing convergence guarantees are based on unrealistic (or suboptimal) hypotheses on the denoiser, or limited to strongly convex data terms. In this work, we propose a new type of Plug-and-Play methods, based on half-quadratic splitting, for which the denoiser is realized as a gradient descent step on a functional parameterized by a deep neural network. Exploiting convergence results for proximal gradient descent algorithms in the non-convex setting, we show that the proposed Plug-and-Play algorithm is a convergent iterative scheme that targets stationary points of an explicit global functional. Besides, experiments show that it is possible to learn such a deep denoiser while not compromising the performance in comparison to other state-of-the-art deep denoisers used in Plug-and-Play schemes. We apply our proximal gradient algorithm to various ill-posed inverse problems, e.g. deblurring, super-resolution and inpainting. For all these applications, numerical results empirically confirm the convergence results. Experiments also show that this new algorithm reaches state-of-the-art performance, both quantitatively and qualitatively.

References in corpus (7)

• Fixed-Point and Objective Convergence of Plug-and-Play Algorithms
• Plug-and-play ISTA converges with kernel denoisers
• On Plug-and-Play Regularization using Linear Denoisers
• Image Restoration using Autoencoding Priors
• Solving Inverse Problems by Joint Posterior Maximization with Autoencoding Prior
• Bayesian imaging using Plug & Play priors: when Langevin meets Tweedie
• Recovery Analysis for Plug-and-Play Priors using the Restricted Eigenvalue Condition

Cited by in corpus (4)

• PatchNR: Learning from Very Few Images by Patch Normalizing Flow Regularization
• Fast Diffusion EM: a diffusion model for blind inverse problems with application to deconvolution
• Convergent Bregman Plug-and-Play Image Restoration for Poisson Inverse Problems
• Equivariant plug-and-play image reconstruction