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

SS-BSN:基于非局部自相似性的自监督去噪注意力盲点网络

SS-BSN: Attentive Blind-Spot Network for Self-Supervised Denoising with Nonlocal Self-Similarity

University of Seoul(首尔大学);Vieworks(Vieworks公司)

Young-Joo Han, Ha-Jin Yu

AI总结:

本文提出轻量级自相似性注意力模块(SS-Attention)以捕获非局部自相似性,并将其集成至盲点网络构建SS-BSN,在SIDD和DND基准上实现了更优的自监督图像去噪性能。

摘要:

最近,基于监督学习的图像去噪方法已有大量研究。然而,这些方法依赖大规模的噪声-干净图像对,这在实践中难以获取。为解决该局限,已提出仅需噪声图像即可训练的自监督训练去噪方法。这些方法基于卷积神经网络(CNN),并展现出良好的性能。但是,基于CNN的方法未考虑利用传统方法中至关重要的非局部自相似性,这可能导致性能受限。本文提出了自相似性注意力(SS-Attention),这是一种能够捕获非局部自相似性以解决该问题的新型自注意力模块。我们专注于以逐像素方式设计轻量级自注意力模块,由于经典自注意力模块的复杂度随空间分辨率呈二次增长,这几乎无法实现。此外,我们将SS-Attention集成到盲点网络中,称为基于自相似性的盲点网络(SS-BSN)。我们在真实世界图像去噪任务上进行了实验。所提方法在Smartphone Image Denoising Dataset (SIDD)和Darmstadt Noise Dataset (DND)基准数据集上的自监督去噪中,定量和定性均优于最先进的方法。

英文摘要:

Recently, numerous studies have been conducted on supervised learning-based image denoising methods. However, these methods rely on large-scale noisy-clean image pairs, which are difficult to obtain in practice. Denoising methods with self-supervised training that can be trained with only noisy images have been proposed to address the limitation. These methods are based on the convolutional neural network (CNN) and have shown promising performance. However, CNN-based methods do not consider using nonlocal self-similarities essential in the traditional method, which can cause performance limitations. This paper presents self-similarity attention (SS-Attention), a novel self-attention module that can capture nonlocal self-similarities to solve the problem. We focus on designing a lightweight self-attention module in a pixel-wise manner, which is nearly impossible to implement using the classic self-attention module due to the quadratically increasing complexity with spatial resolution. Furthermore, we integrate SS-Attention into the blind-spot network called self-similarity-based blind-spot network (SS-BSN). We conduct the experiments on real-world image denoising tasks. The proposed method quantitatively and qualitatively outperforms state-of-the-art methods in self-supervised denoising on the Smartphone Image Denoising Dataset (SIDD) and Darmstadt Noise Dataset (DND) benchmark datasets.

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

↑