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 条
[1]   K-SVD: An algorithm for designing overcomplete dictionaries for sparse representation [J].
Aharon, Michal ;
Elad, Michael ;
Bruckstein, Alfred .
IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2006, 54 (11) :4311-4322
[2]   An iterative thresholding algorithm for linear inverse problems with a sparsity constraint [J].
Daubechies, I ;
Defrise, M ;
De Mol, C .
COMMUNICATIONS ON PURE AND APPLIED MATHEMATICS, 2004, 57 (11) :1413-1457
[3]  
Elad M, 2010, SPARSE AND REDUNDANT REPRESENTATIONS, P1, DOI 10.1007/978-1-4419-7011-4
[4]   Image denoising via sparse and redundant representations over learned dictionaries [J].
Elad, Michael ;
Aharon, Michal .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2006, 15 (12) :3736-3745
[5]  
Engan K, 1999, INT CONF ACOUST SPEE, P2443, DOI 10.1109/ICASSP.1999.760624
[6]  
Huang H., 2023, IEEE J. Biomed. Health Inform.
[7]   WINNet: Wavelet-Inspired Invertible Network for Image Denoising [J].
Huang, Jun-Jie ;
Dragotti, Pier Luigi .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 :4377-4392
[8]   Trainable ISTA for Sparse Signal Recovery [J].
Ito, Daisuke ;
Takabe, Satoshi ;
Wadayama, Tadashi .
IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2019, 67 (12) :3113-3125
[9]   NLRNet: An Efficient Nonlocal Attention ResNet for Pansharpening [J].
Lei, Dajiang ;
Chen, Hao ;
Zhang, Liping ;
Li, Weisheng .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
[10]   Multi-level Wavelet-CNN for Image Restoration [J].
Liu, Pengju ;
Zhang, Hongzhi ;
Zhang, Kai ;
Lin, Liang ;
Zuo, Wangmeng .
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, :886-895