Back to articles
Article
Volume: 35 | Article ID: COIMG-170
Image
Automatic parameter tuning for plug-and-play algorithms using generalized cross validation and Stein's unbiased risk estimation for linear inverse problems in computational imaging
  DOI :  10.2352/EI.2023.35.14.COIMG-170  Published OnlineJanuary 2023
Abstract
Abstract

We propose two automatic parameter tuning methods for Plug-and-Play (PnP) algorithms that use CNN denoisers. We focus on linear inverse problems and propose an iterative algorithm to calculate generalized cross-validation (GCV) and Stein’s unbiased risk estimator (SURE) functions for a half-quadratic splitting-based PnP (PnP-HQS) algorithm that uses a state-of- the-art CNN denoiser. The proposed methods leverage forward mode automatic differentiation to calculate the GCV and SURE functions and tune the parameters of a PnP-HQS algorithm automatically by minimizing the GCV and SURE functions using grid search. Because linear inverse problems appear frequently in computational imaging, the proposed methods can be applied in various domains. Furthermore, because the proposed methods rely on GCV and SURE functions, they do not require access to the ground truth image and do not require collecting an additional training dataset, which is highly desirable for imaging applications for which acquiring data is costly and time-consuming. We evaluate the performance of the proposed methods on deblurring and MRI experiments and show that the GCV-based proposed method achieves comparable performance to that of the oracle tuning method that adjusts the parameters by maximizing the structural similarity index between the ground truth image and the output of the PnP algorithm. We also show that the SURE-based proposed method often leads to worse performance compared to the GCV-based proposed method.

Subject Areas :
Views 171
Downloads 41
 articleview.views 171
 articleview.downloads 41
  Cite this article 

Canberk Ekmekci, Mujdat Cetin, "Automatic parameter tuning for plug-and-play algorithms using generalized cross validation and Stein's unbiased risk estimation for linear inverse problems in computational imagingin Electronic Imaging,  2023,  pp 170--1 - 170-6,  https://doi.org/10.2352/EI.2023.35.14.COIMG-170

 Copy citation
  Copyright statement 
Copyright © 2023, Society for Imaging Science and Technology 2023
ei
Electronic Imaging
2470-1173
2470-1173
Society for Imaging Science and Technology
IS&T 7003 Kilworth Lane, Springfield, VA 22151 USA