arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.02831cs.CVeess.IV

用于图像去噪的两阶段渐进式残差密集注意力网络

Two-stage Progressive Residual Dense Attention Network for Image Denoising

Yunnan Normal University(云南师范大学)

Wencong Wu, An Ge, Guannan Lv, Yuelong Xia, Yungang Zhang, Wen Xiong

AI总结:

针对现有去噪模型同等利用层次特征导致性能受限的问题,提出两阶段渐进式残差密集注意力网络TSP-RDANet,通过RDAM和HDRDAM模块逐步去噪并保留浅层特征,在七个基准数据集上取得优异效果。

摘要:

用于图像去噪的深度卷积神经网络(CNNs)能够有效利用丰富的层次特征,并取得了巨大成功。然而,许多基于深度CNN的去噪模型同等利用含噪图像的层次特征,未关注更重要且有用的特征,导致性能相对较低。为解决该问题,我们设计了一种用于图像去噪的新型两阶段渐进式残差密集注意力网络(TSP-RDANet),将整个去噪过程划分为两个子任务以逐步去除噪声。针对这两个连续子任务,设计了两种不同的基于注意力机制的去噪网络:第一阶段设计了残差密集注意力模块(RDAM),第二阶段提出了混合膨胀残差密集注意力模块(HDRDAM)。所提注意力模块能通过不同卷积层间的密集连接学习合适的局部特征,且无关特征也能被抑制。随后,两个子网络通过长跳跃连接相连以保留浅层特征,从而增强去噪性能。在七个基准数据集上的实验验证了,与许多最先进方法相比,所提TSP-RDANet在合成和真实含噪图像去噪上均能获得良好结果。TSP-RDANet代码可在https://github.com/WenCongWu/TSP-RDANet获取。

英文摘要:

Deep convolutional neural networks (CNNs) for image denoising can effectively exploit rich hierarchical features and have achieved great success. However, many deep CNN-based denoising models equally utilize the hierarchical features of noisy images without paying attention to the more important and useful features, leading to relatively low performance. To address the issue, we design a new Two-stage Progressive Residual Dense Attention Network (TSP-RDANet) for image denoising, which divides the whole process of denoising into two sub-tasks to remove noise progressively. Two different attention mechanism-based denoising networks are designed for the two sequential sub-tasks: the residual dense attention module (RDAM) is designed for the first stage, and the hybrid dilated residual dense attention module (HDRDAM) is proposed for the second stage. The proposed attention modules are able to learn appropriate local features through dense connection between different convolutional layers, and the irrelevant features can also be suppressed. The two sub-networks are then connected by a long skip connection to retain the shallow feature to enhance the denoising performance. The experiments on seven benchmark datasets have verified that compared with many state-of-the-art methods, the proposed TSP-RDANet can obtain favorable results both on synthetic and real noisy image denoising. The code of our TSP-RDANet is available at https://github.com/WenCongWu/TSP-RDANet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑