A Generative Adversarial Network-Based Image Denoiser Controlling Heterogeneous Losses

被引:8
作者
Cho, Sung In [1 ]
Park, Jae Hyeon [1 ]
Kang, Suk-Ju [2 ]
机构
[1] Dongguk Univ, Dept Multimedia Engn, Seoul 04620, South Korea
[2] Sogang Univ, Dept Elect Engn, Seoul 121742, South Korea
基金
新加坡国家研究基金会;
关键词
image denoising; convolutional neural network; generative adversarial network; image restoration; structural loss; OPTIMIZATION; REMOVAL; CNN;
D O I
10.3390/s21041191
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
We propose a novel generative adversarial network (GAN)-based image denoising method that utilizes heterogeneous losses. In order to improve the restoration quality of the structural information of the generator, the heterogeneous losses, including the structural loss in addition to the conventional mean squared error (MSE)-based loss, are used to train the generator. To maximize the improvements brought on by the heterogeneous losses, the strength of the structural loss is adaptively adjusted by the discriminator for each input patch. In addition, a depth wise separable convolution-based module that utilizes the dilated convolution and symmetric skip connection is used for the proposed GAN so as to reduce the computational complexity while providing improved denoising quality compared to the convolutional neural network (CNN) denoiser. The experiments showed that the proposed method improved visual information fidelity and feature similarity index values by up to 0.027 and 0.008, respectively, compared to the existing CNN denoiser.
引用
收藏
页码:1 / 17
页数:16
相关论文
共 42 条
[1]  
Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
[2]  
[Anonymous], 2016, ICLR
[3]   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
[4]  
Burger HC, 2012, PROC CVPR IEEE, P2392, DOI 10.1109/CVPR.2012.6247952
[5]   Image Blind Denoising With Generative Adversarial Network Based Noise Modeling [J].
Chen, Jingwen ;
Chen, Jiawei ;
Chao, Hongyang ;
Yang, Ming .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :3155-3164
[6]   Gradient Prior-Aided CNN Denoiser With Separable Convolution-Based Optimization of Feature Dimension [J].
Cho, Sung In ;
Kang, Suk-Ju .
IEEE TRANSACTIONS ON MULTIMEDIA, 2019, 21 (02) :484-493
[7]   Geodesic Path-Based Diffusion Acceleration for Image Denoising [J].
Cho, Sung In ;
Kang, Suk-Ju .
IEEE TRANSACTIONS ON MULTIMEDIA, 2018, 20 (07) :1738-1750
[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]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[10]   PLTD: Patch-Based Low-Rank Tensor Decomposition for Hyperspectral Images [J].
Du, Bo ;
Zhang, Mengfei ;
Zhang, Lefei ;
Hu, Ruimin ;
Tao, Dacheng .
IEEE TRANSACTIONS ON MULTIMEDIA, 2017, 19 (01) :67-79