Deep K-SVD Denoising

被引:87
作者
Scetbon, Meyer [1 ]
Elad, Michael [2 ]
Milanfar, Peyman [2 ]
机构
[1] ENSAE, CREST, F-91120 Palaiseau, France
[2] Google Res, Mountain View, CA 94043 USA
关键词
Noise reduction; Dictionaries; Image denoising; Task analysis; Noise measurement; Training; Bridges; K-SVD denoising algorithm; network unfolding; iterative shrinkage algorithms; INVERSE PROBLEMS; SPARSE REPRESENTATIONS; NEURAL-NETWORKS; IMAGE; DICTIONARIES; ALGORITHM; RECOVERY;
D O I
10.1109/TIP.2021.3090531
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This work considers noise removal from images, focusing on the well-known K-SVD denoising algorithm. This sparsity-based method was proposed in 2006, and for a short while it was considered as state-of-the-art. However, over the years it has been surpassed by other methods, including the recent deep-learning-based newcomers. The question we address in this paper is whether K-SVD was brought to its peak in its original conception, or whether it can be made competitive again. The approach we take in answering this question is to redesign the algorithm to operate in a supervised manner. More specifically, we propose an end-to-end deep architecture with the exact K-SVD computational path, and train it for optimized denoising. Our work shows how to overcome difficulties arising in turning the K-SVD scheme into a differentiable, and thus learnable, machine. With a small number of parameters to learn and while preserving the original K-SVD essence, the proposed architecture is shown to outperform the classical K-SVD algorithm substantially, and getting closer to recent state-of-the-art learning-based denoising methods. Adopting a broader context, this work touches on themes around the design of deep-learning solutions for image processing tasks, while paving a bridge between classic methods and novel deep-learning-based ones.
引用
收藏
页码:5944 / 5955
页数:12
相关论文
共 50 条
[31]   Group K-SVD for the classification of gene expression data [J].
He, Ping ;
Fan, Baichuan ;
Xu, Xiaohua ;
Ding, Jie ;
Liang, Yali ;
Lou, Yuan ;
Zhang, Zhijun ;
Chang, Xincheng .
COMPUTERS & ELECTRICAL ENGINEERING, 2019, 76 (143-153) :143-153
[32]   An improvement of the K-SVD algorithm with applications on face recognition [J].
Malkomes, Gustavo ;
Pordeus, Joao Paulo ;
Brito, Carlos Fisch .
2014 BRAZILIAN CONFERENCE ON INTELLIGENT SYSTEMS (BRACIS), 2014, :241-246
[33]   K-SVD with reference: an initialization method for dictionary learning [J].
Cai, Pingmei ;
Wang, Guinan ;
Zhang, Hongjuan .
NEURAL COMPUTING & APPLICATIONS, 2014, 25 (06) :1263-1274
[34]   Discriminative K-SVD for Dictionary Learning in Face Recognition [J].
Zhang, Qiang ;
Li, Baoxin .
2010 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2010, :2691-2698
[35]   K-SVD with reference: an initialization method for dictionary learning [J].
Pingmei Cai ;
Guinan Wang ;
Hongjuan Zhang .
Neural Computing and Applications, 2014, 25 :1263-1274
[36]   Onset Detection Using Leared Dictionary By K-SVD [J].
Gui, Wenming ;
Shao, Xi .
PROCEEDINGS OF 2014 IEEE WORKSHOP ON ADVANCED RESEARCH AND TECHNOLOGY IN INDUSTRY APPLICATIONS (WARTIA), 2014, :406-409
[37]   Robust K-SVD: A Novel Approach for Dictionary Learning [J].
Loza, Carlos A. .
PROGRESS IN ARTIFICIAL INTELLIGENCE AND PATTERN RECOGNITION, IWAIPR 2018, 2018, 11047 :185-192
[38]   Compression of facial images using the K-SVD algorithm [J].
Bryt, Ori ;
Elad, Michael .
JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2008, 19 (04) :270-282
[39]   Application of regression analysis in K-SVD dictionary learning [J].
Jose, Jerin ;
Patel, Jigisha N. ;
Patnaik, Suprava .
OPTIK, 2015, 126 (20) :2295-2299
[40]   Compressed Sensing Based on K-SVD for Brillouin Optical Fiber Distributed Sensors [J].
Dong, Yong ;
Yang, Ya-Nan ;
Azad, Abul Kalam ;
Yang, Zengsen ;
Yu, Kuanglu ;
Zhao, Shuang .
IEEE SENSORS JOURNAL, 2022, 22 (16) :16414-16421