论文标题
逐步培训多级小波残留网络,用于图像denoising
Progressive Training of Multi-level Wavelet Residual Networks for Image Denoising
论文作者
论文摘要
近年来,深度卷积神经网络(CNN)在图像DeNoising中取得了巨大的成功。尽管更深层的网络和更大的模型容量通常使性能受益,但训练非常深的图像降级网络仍然是一个具有挑战性的实际问题。以多级小波-CNN(MWCNN)为例,我们从经验上发现,通过增加小波分解水平或在每个级别内增加卷积层,无法显着提高降解性能。为了解决这个问题,本文介绍了多级小波残差网络(MWRN)体系结构以及渐进式培训(PTMWRN)方案,以改善图像deo deo deo的性能。与MWCNN相反,我们的MWRN在每个离散小波变换(DWT)和逆离散小波变换(IDWT)之前引入了几个残差块。为了缓解训练难度,通过要求中间输出近似地面真相清洁图像的相应小波子带,将适用于MWRN的每个级别的MWRN应用。为了确保尺度特异性损失的有效性,我们还将嘈杂图像的小波子带作为编码器每个尺度的输入。此外,通过培训MWRN的最低水平,并逐步训练高层以带来更多细节,从而为您带来了更多细节,从而为您提供了逐步训练,从而采用了渐进培训计划来更好地学习MWRN。合成和现实世界嘈杂图像的实验表明,我们的PT-MWRN用定量指标和视觉质量的术语对最先进的去核方法表现出色。
Recent years have witnessed the great success of deep convolutional neural networks (CNNs) in image denoising. Albeit deeper network and larger model capacity generally benefit performance, it remains a challenging practical issue to train a very deep image denoising network. Using multilevel wavelet-CNN (MWCNN) as an example, we empirically find that the denoising performance cannot be significantly improved by either increasing wavelet decomposition levels or increasing convolution layers within each level. To cope with this issue, this paper presents a multi-level wavelet residual network (MWRN) architecture as well as a progressive training (PTMWRN) scheme to improve image denoising performance. In contrast to MWCNN, our MWRN introduces several residual blocks after each level of discrete wavelet transform (DWT) and before inverse discrete wavelet transform (IDWT). For easing the training difficulty, scale-specific loss is applied to each level of MWRN by requiring the intermediate output to approximate the corresponding wavelet subbands of ground-truth clean image. To ensure the effectiveness of scale-specific loss, we also take the wavelet subbands of noisy image as the input to each scale of the encoder. Furthermore, progressive training scheme is adopted for better learning of MWRN by beigining with training the lowest level of MWRN and progressively training the upper levels to bring more fine details to denoising results. Experiments on both synthetic and real-world noisy images show that our PT-MWRN performs favorably against the state-of-the-art denoising methods in terms both quantitative metrics and visual quality.