WaveMamba: Spatial-Spectral Wavelet Mamba for Hyperspectral Image Classification
Muhammad Ahmad, Muhammad Usama, Manuel Mazzara, Salvatore Distefano
Abstract
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 letter introduces WaveMamba, a novel approach that integrates wavelet transformation with the spatial-spectral Mamba (SSMamba) 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.