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

WaveMamba:用于高光谱图像分类的空谱小波Mamba

WaveMamba: Spatial-Spectral Wavelet Mamba for Hyperspectral Image Classification

University of Messina(墨西拿大学);National University of Computer and Emerging Sciences(国家计算机与新兴科学大学);Innopolis University(因诺波利斯大学)

Muhammad Ahmad, Muhammad Usama, Manuel Mazzara, Salvatore Distefano

AI总结:

针对高光谱图像分类中计算效率低、需大量标记数据的挑战,提出WaveMamba,将小波变换与空谱Mamba架构结合,捕捉局部纹理与全局上下文,在两个数据集上准确率分别提升4.5%和2.0%。

摘要:

高光谱成像(HSI)已被证明是在多种应用中捕捉详细光谱和空间信息的强大工具。尽管深度学习(DL)和Transformer架构在HSI分类方面取得了进展,但计算效率低和需要大量标记数据等挑战仍然存在。本文提出WaveMamba,这是一种将小波变换与空谱Mamba架构相结合以增强HSI分类的新方法。WaveMamba在端到端可训练模型中捕捉局部纹理模式和全局上下文关系。基于小波的增强特征随后通过状态空间架构处理,以建模空谱关系和时间依赖性。实验结果表明,WaveMamba优于现有模型,在休斯顿大学数据集上准确率提高4.5%,在帕维亚大学数据集上提高2.0%。

英文摘要:

Hyperspectral Imaging (HSI) has proven to be a powerful tool for capturing detailed spectral and spatial information across diverse applications. Despite the advancements in Deep Learning (DL) and Transformer architectures for HSI classification, challenges such as computational efficiency and the need for extensive labeled data persist. This paper introduces WaveMamba, a novel approach that integrates wavelet transformation with the spatial-spectral Mamba architecture to enhance HSI classification. WaveMamba captures both local texture patterns and global contextual relationships in an end-to-end trainable model. The Wavelet-based enhanced features are then processed through the state-space architecture to model spatial-spectral relationships and temporal dependencies. The experimental results indicate that WaveMamba surpasses existing models, achieving an accuracy improvement of 4.5\% on the University of Houston dataset and a 2.0\% increase on the Pavia University dataset.

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

↑