Boosting the Performance of Plug-and-Play Priors via Denoiser Scaling
arXiv:2002.11546
Abstract
Plug-and-play priors (PnP) is an image reconstruction framework that uses an image denoiser as an imaging prior. Unlike traditional regularized inversion, PnP does not require the prior to be expressible in the form of a regularization function. This flexibility enables PnP algorithms to exploit the most effective image denoisers, leading to their state-of-the-art performance in various imaging tasks. In this paper, we propose a new denoiser scaling technique to explicitly control the amount of PnP regularization. Traditionally, the performance of PnP algorithms is controlled via intrinsic parameters of the denoiser related to the noise variance. However, many powerful denoisers, such as the ones based on convolutional neural networks (CNNs), do not have tunable parameters that would allow controlling their influence within PnP. To address this issue, we introduce a scaling parameter that adjusts the magnitude of the denoiser input and output. We theoretical justify the denoiser scaling from the perspectives of proximal optimization, statistical estimation, and consensus equilibrium. Finally, we provide numerical experiments demonstrating the ability of denoiser scaling to systematically improve the performance of PnP for denoising CNN priors that do not have explicitly tunable parameters.
References in corpus (10)
- Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising
- Trainable Nonlinear Reaction Diffusion: A Flexible Framework for Fast and Effective Image Restoration
- Fast Image Recovery Using Variable Splitting and Constrained Optimization
- fastMRI: An Open Dataset and Benchmarks for Accelerated MRI
- An Online Plug-and-Play Algorithm for Regularized Image Reconstruction
- The Singular Values of Convolutional Layers
- Plug-and-Play Methods Provably Converge with Properly Trained Denoisers
- Compressive Imaging via Approximate Message Passing with Image Denoising
- Deep Mean-Shift Priors for Image Restoration
- Plug-in Estimation in High-Dimensional Linear Inverse Problems: A Rigorous Analysis