AdaptivePnP: Fast Auto-Tuned Plug-and-Play for Deconvolution
Organizations: Department of Mathematics, Purdue University, West Lafayette, IN 47907, USA · National Security Sciences Directorate, Oak Ridge National Laboratory, Oak Ridge, TN 37830, USA · School of Electrical and Computer Engineering, Purdue University, West Lafayette, IN 47907, USA
Abstract
Plug-and-play (PnP) methods reconstruct images by alternating a data-fitting step with an image denoiser. Their performance depends on two parameters: the step size and the measurement noise level . In practice, is often tuned by grid search over validation images and must be retuned whenever the problem setting changes, and is assumed known or estimated separately. We propose AdaptivePnP, a PnP framework that jointly estimates and at every iteration from the data-fitting residual. To do so, we apply the PnP virtual model to the data-fitting proximal map and show that the residual is zero-mean Gaussian with covariance parameterized by and . However, when one parameter is much larger than the other, the residual carries little information about the smaller one and using the maximum likelihood (ML) estimator can degrade the PnP iterations. We therefore introduce the identifiable maximum likelihood (IML) estimator, which restricts the ML estimator to an identifiable window in which the normalized Cramér--Rao bound of every parameter is bounded by a user-specified tolerance. For the virtual model, this window is an interval of ratios , so computing the IML estimate reduces to a one-dimensional search over . We apply AdaptivePnP to two standard PnP methods, yielding algorithms that require neither a tuned nor a known . On image deblurring and super-resolution, AdaptivePnP matches or outperforms PnP baselines tuned with the true . Code is available at https://github.com/hdubasullivan/AdaptivePnP.
Figures & tables
| Isotropic Gaussian | Anisotropic Gaussian | Motion | |||||
| Tuned PnP | AdaptivePnP | Tuned PnP | AdaptivePnP | Tuned PnP | AdaptivePnP | ||
| Deblurring | PGM | 27.83 / 0.768 | 28.39 / 0.807 | 25.49 / 0.701 | 25.83 / 0.719 | 26.64 / 0.719 | 28.01 / 0.777 |
| ADMM | 28.27 / 0.793 | 28.32 / 0.808 | 25.77 / 0.722 | 25.79 / 0.719 | 28.02 / 0.800 | 27.96 / 0.782 | |
| 2 Super-Resolution | PGM | 24.07 / 0.603 | 26.69 / 0.737 | 24.83 / 0.634 | 25.23 / 0.675 | 22.24 / 0.470 | 25.47 / 0.650 |
| ADMM | 23.88 / 0.623 | 26.55 / 0.737 | 25.13 / 0.671 | 25.28 / 0.685 | 23.03 / 0.578 | 25.79 / 0.701 | |