A Multi-Head Convolutional Neural Network With Multi-path Attention improves Image Denoising

04/27/2022
by   Jiahong Zhang, et al.
8

Recently, convolutional neural networks (CNNs) and attention mechanisms have been widely used in image denoising and achieved satisfactory performance. However, the previous works mostly use a single head to receive the noisy image, limiting the richness of extracted features. Therefore, a novel CNN with multiple heads (MH) named MHCNN is proposed in this paper, whose heads will receive the input images rotated by different rotation angles. MH makes MHCNN simultaneously utilize features of rotated images to remove noise. We also present a novel multi-path attention mechanism (MPA) to integrate these features effectively. Unlike previous attention mechanisms that handle pixel-level, channel-level, and patch-level features, MPA focuses on features at the image level. Experiments show MHCNN surpasses other state-of-the-art CNN models on additive white Gaussian noise (AWGN) denoising and real-world image denoising. Its peak signal-to-noise ratio (PSNR) results are higher than other networks, such as DnCNN, BRDNet, RIDNet, PAN-Net, and CSANN. It is also demonstrated that the proposed MH with MPA mechanism can be used as a pluggable component.

READ FULL TEXT

page 1

page 3

page 5

page 6

research
04/04/2023

DCANet: Dual Convolutional Neural Network with Attention for Image Blind Denoising

Noise removal of images is an essential preprocessing procedure for many...
research
09/14/2022

Considering Image Information and Self-similarity: A Compositional Denoising Network

Recently, convolutional neural networks (CNNs) have been widely used in ...
research
08/01/2019

Pyramid Real Image Denoising Network

While deep Convolutional Neural Networks (CNNs) have shown extraordinary...
research
07/27/2021

Adaptive Denoising via GainTuning

Deep convolutional neural networks (CNNs) for image denoising are usuall...
research
07/17/2017

Wide Inference Network for Image Denoising via Learning Pixel-distribution Prior

We explore an innovative strategy for image denoising by using convoluti...
research
09/02/2021

Anatomical-Guided Attention Enhances Unsupervised PET Image Denoising Performance

Although supervised convolutional neural networks (CNNs) often outperfor...
research
07/28/2017

Learning Pixel-Distribution Prior with Wider Convolution for Image Denoising

In this work, we explore an innovative strategy for image denoising by u...

Please sign up or login with your details

Forgot password? Click here to reset