arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.03885eess.IVcs.CV

基于空间-光谱循环 Transformer 的高光谱图像去噪

Hyperspectral Image Denoising via Spatial-Spectral Recurrent Transformer

Nanjing University of Science and Technology(南京理工大学);University of Macau(澳门大学);Griffith University(格里菲斯大学);Zhejiang University(浙江大学)

Guanyiman Fu, Fengchao Xiong, Jianfeng Lu, Jun Zhou, Jiantao Zhou, Yuntao Qian

AI总结:

针对高光谱图像去噪中空间自相似性与全局光谱相关性被割裂利用的问题,提出 SSRT-UNet,通过双分支共享键值与跨波段循环建模实现二者联合,取得优于现有方法的性能。

摘要:

高光谱图像(HSI)经常受到来自成像内部机制和环境因素的噪声影响。利用 HSI 特有的领域知识,例如全局光谱相关性(GSC)和非局部空间自相似性(NSS),对于有效去噪至关重要。现有方法往往通过多个模块独立利用这些知识成分,忽视了 HSI 固有的三维特性,而在该特性中领域知识紧密关联,从而导致性能欠佳。为应对这一挑战,本文提出一种用于 HSI 去噪的空间-光谱循环 transformer U-Net(SSRT-UNet)。所提出的 SSRT-UNet 在单个 SSRT 模块内整合 NSS 与 GSC 属性。该模块由空间分支和光谱分支组成。光谱分支结合使用 transformer 与循环神经网络,在各波段之间执行循环计算,从而能够超越固定波段数量来利用 GSC。同时,空间分支在 GSC 的引导下,通过与光谱分支共享键和值,对每个波段的 NSS 进行编码。两个分支之间的这种交互实现了 NSS 与 GSC 的联合利用,避免将二者独立处理。实验结果表明,本文方法优于若干替代方法。源代码将在 https://github.com/lronkitty/SSRT 提供。

英文摘要:

Hyperspectral images (HSIs) often suffer from noise arising from both intra-imaging mechanisms and environmental factors. Leveraging domain knowledge specific to HSIs, such as global spectral correlation (GSC) and non-local spatial self-similarity (NSS), is crucial for effective denoising. Existing methods tend to independently utilize each of these knowledge components with multiple blocks, overlooking the inherent 3D nature of HSIs where domain knowledge is strongly interlinked, resulting in suboptimal performance. To address this challenge, this paper introduces a spatial-spectral recurrent transformer U-Net (SSRT-UNet) for HSI denoising. The proposed SSRT-UNet integrates NSS and GSC properties within a single SSRT block. This block consists of a spatial branch and a spectral branch. The spectral branch employs a combination of transformer and recurrent neural network to perform recurrent computations across bands, allowing for GSC exploitation beyond a fixed number of bands. Concurrently, the spatial branch encodes NSS for each band by sharing keys and values with the spectral branch under the guidance of GSC. This interaction between the two branches enables the joint utilization of NSS and GSC, avoiding their independent treatment. Experimental results demonstrate that our method outperforms several alternative approaches. The source code will be available at https://github.com/lronkitty/SSRT.

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

↑