Efficient denoising method for real-world noise image using Scalable Convolution and Channel Interaction Attention

被引:0
作者
Li, Xiaoxia [1 ,2 ]
Dong, Liugu [1 ]
Wang, Li [3 ]
Zhou, Yingyue [1 ,2 ]
机构
[1] Southwest Univ Sci & Technol, Sch Informat Engn, Mianyang 621010, Sichuan, Peoples R China
[2] Robot Technol Used Special Environm Key Lab Sichua, Mianyang 621010, Sichuan, Peoples R China
[3] Suzhou HYC Technol CO LTD, Suzhou 215000, Jiangsu, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolution neural networks; Real noise image denoising; Attention mechanism; Scalable convolution; NETWORK;
D O I
10.1007/s11554-024-01575-w
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We propose a three-stage image denoising method, called Scalable Convolution and Channel Interaction Attention (SC-CIA), to address the high computational cost, complexity, and suboptimal performance of traditional convolution image denoising networks when dealing with real-world noise. In the first stage, we use a variant of dynamic convolution called Scalable Convolution for shallow feature extraction. This method utilizes a set of adaptive small convolution kernels with spatial variations and applies them to full-resolution feature mapping through slicing operations. By combining bilinear interpolation and Scalable Convolution operations, it minimizes computational resources while enhancing the network's ability to capture position and shape information from images. In addition, our method incorporates an attention mechanism for channel interaction. This mechanism groups every two channels of the input feature map, generating attention maps for each subgroup. The output features are then aggregated and rearranged to achieve channel information interaction and feature enhancement, thereby improving the model's ability to remove real noise. Compared to various mainstream denoising networks, our method achieves excellent PSNR/SSIM performance on SIDD and DND datasets, while significantly reducing computational complexity (measured in MACs). In particular, when processing 512 x\documentclass[12pt]{minimal} \usepackage{amsmath} \usepackage{wasysym} \usepackage{amsfonts} \usepackage{amssymb} \usepackage{amsbsy} \usepackage{mathrsfs} \usepackage{upgreek} \setlength{\oddsidemargin}{-69pt} \begin{document}$$\times$$\end{document} 512 images, our method only uses 10% of the MACs used by MIRNet, and has an inference speed that is 3.73 times faster than MIRNet. These results highlight the potential of our method for fast denoising applications.
引用
收藏
页数:14
相关论文
共 45 条
[1]   A High-Quality Denoising Dataset for Smartphone Cameras [J].
Abdelhamed, Abdelrahman ;
Lin, Stephen ;
Brown, Michael S. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1692-1700
[2]   SRLibrary: Comparing different loss functions for super-resolution over various convolutional architectures [J].
Anagun, Yildiray ;
Isik, Sahin ;
Seke, Erol .
JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2019, 61 :178-187
[3]   Real Image Denoising with Feature Attention [J].
Anwar, Saeed ;
Barnes, Nick .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3155-3164
[4]   A General Survey on Attention Mechanisms in Deep Learning [J].
Brauwers, Gianni ;
Frasincar, Flavius .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2023, 35 (04) :3279-3298
[5]   CA-BSN: Mural Image Denoising Based on Cross-Attention Blind Spot Network [J].
Cai, Xingquan ;
Liu, Yao ;
Liu, Shike ;
Zhang, Haoyu ;
Sun, Haiyan .
APPLIED SCIENCES-BASEL, 2024, 14 (02)
[6]   Masked Image Training for Generalizable Deep Image Denoising [J].
Chen, Haoyu ;
Gu, Jinjin ;
Liu, Yihao ;
Magid, Salma Abdel ;
Dong, Chao ;
Wang, Qiong ;
Pfister, Hanspeter ;
Zhu, Lei .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR, 2023, :1692-1703
[7]   NBNet: Noise Basis Learning for Image Denoising with Subspace Projection [J].
Cheng, Shen ;
Wang, Yuzhi ;
Huang, Haibin ;
Liu, Donghao ;
Fan, Haoqiang ;
Liu, Shuaicheng .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :4894-4904
[8]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[9]  
Chung JY, 2014, Arxiv, DOI arXiv:1412.3555
[10]   ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks [J].
Ding, Xiaohan ;
Guo, Yuchen ;
Ding, Guiguang ;
Han, Jungong .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :1911-1920