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

重新思考基于Transformer的盲点网络用于自监督图像去噪

Rethinking Transformer-Based Blind-Spot Network for Self-Supervised Image Denoising

Junyi Li, Zhilu Zhang, and Wangmeng Zuo(李俊毅、张志鲁和祖旺萌)

Junyi Li, Zhilu Zhang, Wangmeng Zuo

AI总结:

本文针对自监督图像去噪中Transformer注意力违反盲点要求的问题,重新设计通道分组注意力与空间掩码注意力,提出TBSN,并通过知识蒸馏提升效率,在真实数据集上取得领先性能。

摘要:

盲点网络(BSN)已成为自监督图像去噪(SSID)中主流的神经架构。然而,现有的大多数BSN都采用卷积层实现。尽管Transformer已在许多图像复原任务中展现出克服卷积局限性的潜力,但其注意力机制可能违反盲点要求,从而限制了其在BSN中的适用性。为此,我们提出对通道注意力和空间注意力进行分析与重新设计,以满足盲点要求。具体而言,在多尺度架构中,通道自注意力可能会泄漏盲点信息,因为下采样会将空间特征混洗到通道维度中。为缓解这一问题,我们将通道划分为若干组,并分别执行通道注意力。对于空间自注意力,我们对注意力矩阵施加精心设计的掩码,以限制并模拟扩张卷积的感受野。基于重新设计的通道注意力和窗口注意力,我们构建了一个基于Transformer的盲点网络(TBSN),该网络展现出强大的局部拟合能力和全局视角能力。此外,我们引入了一种知识蒸馏策略,将TBSN蒸馏为更小的去噪器,以在保持性能的同时提高计算效率。在真实世界图像去噪数据集上的大量实验表明,TBSN大幅扩展了感受野,并展现出优于最先进SSID方法的性能。

英文摘要:

Blind-spot networks (BSN) have been prevalent neural architectures in self-supervised image denoising (SSID). However, most existing BSNs are conducted with convolution layers. Although transformers have shown the potential to overcome the limitations of convolutions in many image restoration tasks, the attention mechanisms may violate the blind-spot requirement, thereby restricting their applicability in BSN. To this end, we propose to analyze and redesign the channel and spatial attentions to meet the blind-spot requirement. Specifically, channel self-attention may leak the blind-spot information in multi-scale architectures, since the downsampling shuffles the spatial feature into channel dimensions. To alleviate this problem, we divide the channel into several groups and perform channel attention separately. For spatial selfattention, we apply an elaborate mask to the attention matrix to restrict and mimic the receptive field of dilated convolution. Based on the redesigned channel and window attentions, we build a Transformer-based Blind-Spot Network (TBSN), which shows strong local fitting and global perspective abilities. Furthermore, we introduce a knowledge distillation strategy that distills TBSN into smaller denoisers to improve computational efficiency while maintaining performance. Extensive experiments on real-world image denoising datasets show that TBSN largely extends the receptive field and exhibits favorable performance against state-of-theart SSID methods.

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

↑