PGSMamba: Prompt-Guided Shuffle State Space Model for Hyperspectral Image Classification

Read full text
Authors: Qianlong Yang, Xiaozhen Qiao, Yonggang Du, Mingyu Dou, Xing Xia, Yulei Jia
Cited by: 1
References: 56

Abstract

Hyperspectral image classification demands models capable of efficiently capturing complex spectral–spatial relationships and long-range dependencies. Despite significant advances in CNNs and Transformers, balancing modeling capacity with computational efficiency remains challenging. Recently, Mamba has emerged as a promising alternative with linear-complexity global modeling capabilities. However, current Mamba-based methods suffer from two critical limitations: artifact tokens that compromise foreground-background distinction, and overfitting tendencies exacerbated by their non-hierarchical, homogeneous layer architecture. To address these limitations, we propose PGSMamba, a prompt-guided shuffle state space model for hyperspectral image classification. Our approach incorporates three core components: a patch-prompt sequence construction module for artifact filtering through learnable prompts, a shuffle scanning module providing layer- wise regularization via random perturbations, and a prompt-to-patch aggregation strategy enabling robust global-local feature fusion and effectively suppressing artifact tokens within patch tokens. Comprehensive experiments on five benchmark datasets, including Indian Pines, Pavia University, Houston 2013, WHU-Hi-LongKou, and Salinas, demonstrate that PGSMamba achieves state-of-the-art overall accuracies of 97.88%, 98.09%, 94.02%, 98.92%, and 95.11%, respectively. These results confirm the effectiveness and generalization ability of our approach, highlighting PGSMamba as a promising solution for efficient and robust hyperspectral image classification.

Fields of Study & Subjects

Remote-Sensing Image Classification Advanced Neural Network Applications Domain Adaptation and Few-Shot Learning

Publication details

Journal / source
IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Publisher
Institute of Electrical and Electronics Engineers (IEEE)
Volume
19
Pages
5460–5475
Publication date
1 January 2026
Language
EN