Lightweight multi-scale generative adversarial network with attention for image denoising

Xuegang Hu,Wei Zhao
DOI: https://doi.org/10.1007/s00530-024-01508-4
IF: 3.9
2024-09-26
Multimedia Systems
Abstract:Most existing image denoising methods focus on improving denoising effect while neglecting computational cost. In this paper, a lightweight multi-scale generative adversarial network with attention is proposed to achieve superior denoising performance while reducing computational redundancy. Specifically, A novel generator with two sub-networks is designed for end-to-end image denoising. In the upper branch network, an attention-guided multi-scale convolution (AMC) block and a U-shaped dilated convolution (UDC) block are utilized to filter and extract noise features at multiple scales, enhancing the model's feature representation capability and generalization ability while reducing computational complexity. The lower branch network directly predicts denoised images, consisting mainly of a dilated convolution residual (DCR) block and a residual attention (RA) block. DCR is composed of residual network and dilated convolution, which can capture multi-scale features stably and effectively without increasing the number of parameters. RA can be regarded as a novel attention mechanism to further guide the network for image denoising and image recovery. Experimental results show that compared with some state-of-the-art methods, the proposed method has better performance in terms of parameter overhead, visual effect and objective metrics such as PSNR and SSIM.
computer science, information systems, theory & methods
What problem does this paper attempt to address?