arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.09040cs.CVeess.IV

带引导注意力的混合光谱去噪Transformer

Hybrid Spectral Denoising Transformer with Guided Attention

Beijing Institute of Technology(北京理工大学);Hangzhou Dianzi University(杭州电子科技大学)

Zeqiang Lai, Chenggang Yan, Ying Fu

AI总结:

本文提出HSDT,通过S3Conv、引导光谱自注意力和自调制前馈网络,高效捕获高光谱图像的全局与局部空间-光谱相关性,在低计算开销下显著提升去噪性能。

摘要:

本文提出了一种用于高光谱图像去噪的混合光谱去噪Transformer(HSDT)。将Transformer适配到高光谱图像(HSI)所面临的挑战在于,既要解决现有基于CNN的方法在捕获全局和局部空间-光谱相关性方面的局限性,又要保持效率和灵活性。为解决这些问题,我们引入了一种混合方法,结合了两种模型的优势,并包含空间-光谱可分离卷积(S3Conv)、引导光谱自注意力(GSSA)和自调制前馈网络(SM-FFN)。我们的S3Conv作为3D卷积的轻量级替代方案,能够提取更多空间-光谱相关特征,同时保持处理任意波段数HSI的灵活性。这些特征随后由GSSA自适应处理,GSSA在光谱波段上执行3D自注意力,并由一组编码光谱特征的可学习查询进行引导。这不仅使我们的模型具备强大的全局光谱相关性识别能力,而且保持了线性复杂度。此外,我们的SM-FFN提出了自调制机制,增强信息更丰富区域的激活,从而进一步强化聚合特征。在模拟噪声和真实世界噪声下的多个数据集上进行了大量实验,结果表明我们的HSDT在保持低计算开销的同时,显著优于现有的最先进方法。代码位于https://github.com/Zeqiang-Lai/HSDT。

英文摘要:

In this paper, we present a Hybrid Spectral Denoising Transformer (HSDT) for hyperspectral image denoising. Challenges in adapting transformer for HSI arise from the capabilities to tackle existing limitations of CNN-based methods in capturing the global and local spatial-spectral correlations while maintaining efficiency and flexibility. To address these issues, we introduce a hybrid approach that combines the advantages of both models with a Spatial-Spectral Separable Convolution (S3Conv), Guided Spectral Self-Attention (GSSA), and Self-Modulated Feed-Forward Network (SM-FFN). Our S3Conv works as a lightweight alternative to 3D convolution, which extracts more spatial-spectral correlated features while keeping the flexibility to tackle HSIs with an arbitrary number of bands. These features are then adaptively processed by GSSA which per-forms 3D self-attention across the spectral bands, guided by a set of learnable queries that encode the spectral signatures. This not only enriches our model with powerful capabilities for identifying global spectral correlations but also maintains linear complexity. Moreover, our SM-FFN proposes the self-modulation that intensifies the activations of more informative regions, which further strengthens the aggregated features. Extensive experiments are conducted on various datasets under both simulated and real-world noise, and it shows that our HSDT significantly outperforms the existing state-of-the-art methods while maintaining low computational overhead. Code is at https: //github.com/Zeqiang-Lai/HSDT.

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

↑