WaveMamba: Spatial-Spectral Wavelet Mamba for Hyperspectral Image Classification

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Ahmad, Muhammad, Usama, Muhammad, Mazzara, Manuel, Distefano, Salvatore
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910719432196096
author Ahmad, Muhammad
Usama, Muhammad
Mazzara, Manuel
Distefano, Salvatore
author_facet Ahmad, Muhammad
Usama, Muhammad
Mazzara, Manuel
Distefano, Salvatore
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2408_01231
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle WaveMamba: Spatial-Spectral Wavelet Mamba for Hyperspectral Image Classification
Ahmad, Muhammad
Usama, Muhammad
Mazzara, Manuel
Distefano, Salvatore
Computer Vision and Pattern Recognition
Image and Video Processing
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.
title WaveMamba: Spatial-Spectral Wavelet Mamba for Hyperspectral Image Classification
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2408.01231