Abstract:Plug-and-play (PnP) is a non-convex framework that combines ADMM or otherproximal algorithms with advanced denoiser priors. Recently, PnP has achievedgreat empirical success, especially with the integration of deep learning-baseddenoisers. However, a key problem of PnP based approaches is that they requiremanual parameter tweaking. It is necessary to obtain high-quality resultsacross the high discrepancy in terms of imaging conditions and varying scenecontent. In this work, we present a tuning-free PnP proximal algorithm, whichcan automatically determine the internal parameters including the penaltyparameter, the denoising strength and the terminal time. A key part of ourapproach is to develop a policy network for automatic search of parameters,which can be effectively learned via mixed model-free and model-based deepreinforcement learning. We demonstrate, through numerical and visualexperiments, that the learned policy can customize different parameters fordifferent states, and often more efficient and effective than existinghandcrafted criteria. Moreover, we discuss the practical considerations of theplugged denoisers, which together with our learned policy yieldstate-of-the-art results. This is prevalent on both linear and nonlinearexemplary inverse imaging problems, and in particular, we show promisingresults on Compressed Sensing MRI and phase retrieval.

Tuning-free Plug-and-Play Proximal Algorithm for Inverse Imaging Problems