A Multi-Scale Feature Extraction-Based Normalized Attention Neural Network for Image Denoising

被引:20
作者
Wang, Yi [1 ]
Song, Xiao [2 ]
Gong, Guanghong [1 ]
Li, Ni [1 ]
机构
[1] Beihang Univ, Sch Automat Sci & Elect Engn, Beijing 100191, Peoples R China
[2] Beihang Univ, Sch Cyber Sci & Technol, Beijing 100191, Peoples R China
关键词
image denoising; attention neural network; multi-scale feature extraction; PSNR; SSIM; TRANSFORM; NOISE;
D O I
10.3390/electronics10030319
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Due to the rapid development of deep learning and artificial intelligence techniques, denoising via neural networks has drawn great attention due to their flexibility and excellent performances. However, for most convolutional network denoising methods, the convolution kernel is only one layer deep, and features of distinct scales are neglected. Moreover, in the convolution operation, all channels are treated equally; the relationships of channels are not considered. In this paper, we propose a multi-scale feature extraction-based normalized attention neural network (MFENANN) for image denoising. In MFENANN, we define a multi-scale feature extraction block to extract and combine features at distinct scales of the noisy image. In addition, we propose a normalized attention network (NAN) to learn the relationships between channels, which smooths the optimization landscape and speeds up the convergence process for training an attention model. Moreover, we introduce the NAN to convolutional network denoising, in which each channel gets gain; channels can play different roles in the subsequent convolution. To testify the effectiveness of the proposed MFENANN, we used both grayscale and color image sets whose noise levels ranged from 0 to 75 to do the experiments. The experimental results show that compared with some state-of-the-art denoising methods, the restored images of MFENANN have larger peak signal-to-noise ratios (PSNR) and structural similarity index measure (SSIM) values and get better overall appearance.
引用
收藏
页码:1 / 19
页数:18
相关论文
共 54 条
[1]  
Abiko R, 2019, INT CONF ACOUST SPEE, P1717, DOI [10.1109/icassp.2019.8683878, 10.1109/ICASSP.2019.8683878]
[2]  
[Anonymous], 2009, J ELECTRON IMAGING, DOI DOI 10.1109/MSP.2008.930649
[3]  
[Anonymous], 1999, Kodak lossless true color image suite
[4]  
[Anonymous], 2020, ELECTRONICS
[5]   A review of image denoising algorithms, with a new one [J].
Buades, A ;
Coll, B ;
Morel, JM .
MULTISCALE MODELING & SIMULATION, 2005, 4 (02) :490-530
[6]  
Burger HC, 2012, PROC CVPR IEEE, P2392, DOI 10.1109/CVPR.2012.6247952
[7]   Single Image Super-ResolutionUsing Deep CNN with Dense Skip Connections and Inception-ResNet [J].
Chen, Chao ;
Qi, Feng .
2018 NINTH INTERNATIONAL CONFERENCE ON INFORMATION TECHNOLOGY IN MEDICINE AND EDUCATION (ITME 2018), 2018, :999-1003
[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]   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
[10]   Weighted Nuclear Norm Minimization with Application to Image Denoising [J].
Gu, Shuhang ;
Zhang, Lei ;
Zuo, Wangmeng ;
Feng, Xiangchu .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :2862-2869