A Fast Stochastic Plug-and-Play ADMM for Imaging Inverse Problems
arXiv:2006.11630
Abstract
In this work we propose an efficient stochastic plug-and-play (PnP) algorithm for imaging inverse problems. The PnP stochastic gradient descent methods have been recently proposed and shown improved performance in some imaging applications over standard deterministic PnP methods. However, current stochastic PnP methods need to frequently compute the image denoisers which can be computationally expensive. To overcome this limitation, we propose a new stochastic PnP-ADMM method which is based on introducing stochastic gradient descent inner-loops within an inexact ADMM framework. We provide the theoretical guarantee on the fixed-point convergence for our algorithm under standard assumptions. Our numerical results demonstrate the effectiveness of our approach compared with state-of-the-art PnP methods.
References in corpus (6)
- 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
- An Online Plug-and-Play Algorithm for Regularized Image Reconstruction
- The Practicality of Stochastic Optimization in Imaging Inverse Problems
- Gradient Projection Iterative Sketch for Large-Scale Constrained Least-Squares
- Rest-Katyusha: Exploiting the Solution's Structure via Scheduled Restart Schemes