WaveMamba: Spatial-Spectral Wavelet Mamba for Hyperspectral Image Classification
This addresses computational efficiency and data needs in hyperspectral imaging for domain-specific applications, representing an incremental advancement.
The paper tackled hyperspectral image classification by introducing WaveMamba, which integrates wavelet transformation with a spatial-spectral Mamba architecture, resulting in accuracy improvements of 4.5% on the University of Houston dataset and 2.0% on the Pavia University dataset.
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.