A Novel Supervised Distributed Dictionary Learning based on Learned K-SVD for Image Denoising

被引:0
作者
Zhang, Chaoran [1 ]
Huang, Huakun [2 ]
Zhao, Lingjun [3 ]
Xu, Chenkai [2 ]
Zhao, Rui [2 ]
机构
[1] Guangzhou Univ, Cyber Space Inst Adv Technol, Guangzhou, Guangdong, Peoples R China
[2] Guangzhou Univ, Comp Sci & Cyber Engn, Guangzhou, Guangdong, Peoples R China
[3] Guangdong Polytech Normal Univ, Elect & Informat, Guangzhou, Guangdong, Peoples R China
来源
2023 IEEE 16TH INTERNATIONAL SYMPOSIUM ON EMBEDDED MULTICORE/MANY-CORE SYSTEMS-ON-CHIP, MCSOC | 2023年
基金
中国国家自然科学基金;
关键词
SoCs; image denoising; sparse coding; distributed learning; dictionary learning; K-SVD; SPARSE; ALGORITHM;
D O I
10.1109/MCSoC60832.2023.00052
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
With limited computational resources and storage space on SoCs, deploying large deep-learning networks (DNNs) is challenging. However, dictionary learning (DicL) has lower complexity and storage space requirements while improving the interpretability of the model. In addition, deep unfolding techniques can construct high-performance end-to-end networks based on DicL. Therefore, in this paper, we apply Deep KSVD (LKSVD), a deep unfolding network based on the classical KSVD algorithm, with a distributed framework and propose a distributed dictionary learning (DDL) method called DDL-LKSVD. We experimentally validate DDL-LKSVD on the classical and fundamental image denoising problem, and the experimental results show that the average PSNR values we achieve in the DDL-LKSVD proposed in this paper on the Set12 dataset with the noise level 25 are 2.58 dB, 2.74 dB, 0.29 dB, and 0.03 dB higher than those of OMP, ISTA, K-SVD, and LKSVD, respectively.
引用
收藏
页码:306 / 311
页数:6
相关论文
共 21 条
[11]  
Martin D, 2001, EIGHTH IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION, VOL II, PROCEEDINGS, P416, DOI 10.1109/ICCV.2001.937655
[12]  
McMahan HB, 2017, PR MACH LEARN RES, V54, P1273
[13]   An Intelligent Recurrent Neural Network with Long Short-Term Memory (LSTM) BASED Batch Normalization for Medical Image Denoising [J].
Rajeev, R. ;
Samath, J. Abdul ;
Karthikeyan, N. K. .
JOURNAL OF MEDICAL SYSTEMS, 2019, 43 (08)
[14]   Deep K-SVD Denoising [J].
Scetbon, Meyer ;
Elad, Michael ;
Milanfar, Peyman .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2021, 30 :5944-5955
[15]   Regression shrinkage and selection via the lasso: a retrospective [J].
Tibshirani, Robert .
JOURNAL OF THE ROYAL STATISTICAL SOCIETY SERIES B-STATISTICAL METHODOLOGY, 2011, 73 :273-282
[16]   SMDS-Net: Model Guided Spectral-Spatial Network for Hyperspectral Image Denoising [J].
Xiong, Fengchao ;
Zhou, Jun ;
Tao, Shuyin ;
Lu, Jianfeng ;
Zhou, Jiantao ;
Qian, Yuntao .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 :5469-5483
[17]   ISTA-Net: Interpretable Optimization-Inspired Deep Network for Image Compressive Sensing [J].
Zhang, Jian ;
Ghanem, Bernard .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1828-1837
[18]   Plug-and-Play Image Restoration With Deep Denoiser Prior [J].
Zhang, Kai ;
Li, Yawei ;
Zuo, Wangmeng ;
Zhang, Lei ;
Van Gool, Luc ;
Timofte, Radu .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (10) :6360-6376
[19]   FFDNet: Toward a Fast and Flexible Solution for CNN-Based Image Denoising [J].
Zhang, Kai ;
Zuo, Wangmeng ;
Zhang, Lei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2018, 27 (09) :4608-4622
[20]   Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising [J].
Zhang, Kai ;
Zuo, Wangmeng ;
Chen, Yunjin ;
Meng, Deyu ;
Zhang, Lei .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (07) :3142-3155