Tuning-free Plug-and-Play Proximal Algorithm for Inverse Imaging Problems
arXiv:2002.09611
Abstract
Plug-and-play (PnP) is a non-convex framework that combines ADMM or other proximal algorithms with advanced denoiser priors. Recently, PnP has achieved great empirical success, especially with the integration of deep learning-based denoisers. However, a key problem of PnP based approaches is that they require manual parameter tweaking. It is necessary to obtain high-quality results across the high discrepancy in terms of imaging conditions and varying scene content. In this work, we present a tuning-free PnP proximal algorithm, which can automatically determine the internal parameters including the penalty parameter, the denoising strength and the terminal time. A key part of our approach is to develop a policy network for automatic search of parameters, which can be effectively learned via mixed model-free and model-based deep reinforcement learning. We demonstrate, through numerical and visual experiments, that the learned policy can customize different parameters for different states, and often more efficient and effective than existing handcrafted criteria. Moreover, we discuss the practical considerations of the plugged denoisers, which together with our learned policy yield state-of-the-art results. This is prevalent on both linear and nonlinear exemplary inverse imaging problems, and in particular, we show promising results on Compressed Sensing MRI and phase retrieval.
References in corpus (21)
- Adam: A Method for Stochastic Optimization
- Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising
- Playing Atari with Deep Reinforcement Learning
- Trust Region Policy Optimization
- Wide-field, high-resolution Fourier ptychographic microscopy
- Non-invasive real-time imaging through scattering layers and around corners via speckle correlations
- Denoising Prior Driven Deep Neural Network for Image Restoration
- Tensor-based formulation and nuclear norm regularization for multi-energy computed tomography
- Generalized SURE for Exponential Families: Applications to Regularization
- Deep Unfolding: Model-Based Inspiration of Novel Deep Architectures
- An Online Plug-and-Play Algorithm for Regularized Image Reconstruction
- Unrolled Optimization with Deep Priors
- Path-Restore: Learning Network Path Selection for Image Restoration
- Plug-and-Play Methods Provably Converge with Properly Trained Denoisers
- DOLPHIn - Dictionary Learning for Phase Retrieval
- One Network to Solve Them All --- Solving Linear Inverse Problems using Deep Projection Models
- Learning to Paint With Model-based Deep Reinforcement Learning
- Regularized Fourier Ptychography using an Online Plug-and-Play Algorithm
- Phase retrieval from noisy data based on sparse approximation of object phase and amplitude
- Differentiable Linearized ADMM
- Dynamically Unfolding Recurrent Restorer: A Moving Endpoint Control Method for Image Restoration