SDAUNet: A simple dual attention mechanism UNet for mixed noise removal
Jielin Jiang,Xiangming Hong,Yingnan Zhao,Xiaonglong Xu,Yan Cui
DOI: https://doi.org/10.1049/ipr2.12905
IF: 2.3
2023-08-12
IET Image Processing
Abstract:In this paper, we propose a simple dual attention mechanism UNet (SDAUNet) for mixed noise removal. In SDAUNet, a simple dual attention module (SDAM) is presented to expand the CNN s receptive field while maintaining low computational cost. With the proposed SDAM, we further propose a simple dual attention convolutional block (SDAConvB) and introduce it to UNet to improve the network s multi‐scale feature extraction ability. Experiments show that the proposed SDAUNet can effectively remove different types of mixed noise and preserve fine‐grained image details. Convolutional neural networks (CNNs) have demonstrated impressive results in additive white Gaussian noise removal due to their strong fitting ability. However, their performance in mixed noise removal remains unsatisfactory, primarily due to their limited receptive field that focuses only on the local features of images and disregards global information. To ameliorate this issue, recent state‐of‐the‐art approaches employ attention mechanism (AM) to capture the global information. However, most AM based methods still suffer from low computational efficiency. In this paper, a novel model named simple dual attention mechanism UNet (SDAUNet) for mixed noise removal is proposed. In SDAUNet, the UNet architecture is used to gradually acquire multi‐scale image features and provide a more comprehensive and accurate representation of the image features than other CNNs. A simple dual attention convolutional block is presented to acquire the global image features that can successfully capture image details with a low burden. The experimental results demonstrate that the SDAUNet model can achieve better measurement metrics and visual performance than other state‐of‐the‐art methods.
computer science, artificial intelligence,engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?