Skip to main navigation Skip to search Skip to main content

Enhancing Hyperspectral Image Classification Through Reinforcement Learning Guided Active Learning

  • University of Limerick
  • University of the West of Scotland

Research output: Contribution to journalArticlepeer-review

Abstract

Hyperspectral image (HSI) classification is vital for environmental monitoring, land cover mapping, and precision agriculture, but its effectiveness is often constrained by the scarcity of labeled samples and high spectral similarity among classes. To address these challenges, we propose CrossCapsViT, a hybrid classification framework that integrates capsule networks (CapsNets) and vision transformers (ViTs) through a cross-attention fusion mechanism and a cross-layer adaptive fusion module, enabling richer and more discriminative spectral-spatial feature learning. To further improve efficiency in data scarce scenarios, we embed an actor-critic reinforcement learning-based active learning (RAL) strategy that jointly leverages accuracy, uncertainty, and diversity in the reward structure, guiding the selection of the most informative samples while reducing labeling effort. Experiments conducted on four benchmark datasets (Kennedy Space Center, Pavia University, Houston University 2013, and Salinas) and a custom UAV-based saltmarsh dataset (Derrymore, collected with a Pika-L sensor) demonstrate that CrossCapsViT with RAL consistently outperforms CapsViT and other baseline models in terms of classification accuracy, robustness, and generalizability. The proposed framework achieves up to 25% improvement in class-level accuracy on challenging vegetation classes, while reducing dependence on large annotated datasets, highlighting its potential for practical deployment in real-world ecological monitoring and remote sensing applications.

Original languageEnglish
Pages (from-to)16314-16332
Number of pages19
JournalIEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Volume19
DOIs
Publication statusPublished - 2026

Keywords

  • Active learning
  • actor-critic model
  • capsule networks (CapsNets)
  • hyperspectral imaging
  • photogrammetry
  • reinforcement learning
  • remote sensing
  • vision transformers (ViTs)

Fingerprint

Dive into the research topics of 'Enhancing Hyperspectral Image Classification Through Reinforcement Learning Guided Active Learning'. Together they form a unique fingerprint.

Cite this