A PLUG-AND-PLAY DEEP IMAGE PRIOR

被引:26
作者
Sun, Zhaodong [1 ]
Latorre, Fabian [1 ]
Sanchez, Thomas [1 ]
Cevher, Volkan [1 ]
机构
[1] Ecole Polytech Fed Lausanne EPFL, Lausanne, Switzerland
来源
2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021) | 2021年
基金
欧洲研究理事会;
关键词
Deep Image Prior; Plug-and-Play Prior; Alternating Direction Method of Multipliers (ADMM); Inverse Problem; Overfitting; ALGORITHM;
D O I
10.1109/ICASSP39728.2021.9414879
中图分类号
O42 [声学];
学科分类号
070206 ; 082403 ;
摘要
Deep image priors (DIP) offer a novel approach for the regularization that leverages the inductive bias of a deep convolutional architecture in inverse problems. However, the quality of DIP approaches often degrades when the number of iterations exceeds a certain threshold due to overfitting. To mitigate this effect, this work incorporates a plug-and-play prior scheme which can accommodate additional regularization steps within a DIP framework. Our modification is achieved using an augmented Lagrangian formulation of the problem, and is solved using an Alternating Direction Method of Multipliers (ADMM) variant, which can capture existing DIP approaches as a special case. We show experimentally that our ADMM-based DIP pairing outperforms competitive baselines in PSNR while exhibiting less overfitting.
引用
收藏
页码:8103 / 8107
页数:5
相关论文
共 22 条
[1]   Solving ill-posed inverse problems using iterative deep neural networks [J].
Adler, Jonas ;
Oktem, Ozan .
INVERSE PROBLEMS, 2017, 33 (12)
[2]   A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems [J].
Beck, Amir ;
Teboulle, Marc .
SIAM JOURNAL ON IMAGING SCIENCES, 2009, 2 (01) :183-202
[3]  
Bora A, 2017, PR MACH LEARN RES, V70
[4]   Distributed optimization and statistical learning via the alternating direction method of multipliers [J].
Boyd S. ;
Parikh N. ;
Chu E. ;
Peleato B. ;
Eckstein J. .
Foundations and Trends in Machine Learning, 2010, 3 (01) :1-122
[5]  
Boyd Stephen, 2004, Convex Optimization, DOI DOI 10.1017/CBO9780511804441
[6]   A non-local algorithm for image denoising [J].
Buades, A ;
Coll, B ;
Morel, JM .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 2, PROCEEDINGS, 2005, :60-65
[7]   A Bayesian Perspective on the Deep Image Prior [J].
Cheng, Zezhou ;
Gadelha, Matheus ;
Maji, Subhransu ;
Sheldon, Daniel .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :5438-5446
[8]   Image denoising by sparse 3-D transform-domain collaborative filtering [J].
Dabov, Kostadin ;
Foi, Alessandro ;
Katkovnik, Vladimir ;
Egiazarian, Karen .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2007, 16 (08) :2080-2095
[9]   Compressed sensing [J].
Donoho, DL .
IEEE TRANSACTIONS ON INFORMATION THEORY, 2006, 52 (04) :1289-1306
[10]  
Goodfellow IJ, 2014, ADV NEUR IN, V27, P2672