Enhancing Semi-Supervised Few-Shot Hyperspectral Image Classification via Progressive Sample Selection

Jiaguo Zhao, Junjie Zhang*, Huaxi Huang, Jian Zhang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Hyperspectral images (HSIs) provide valuable spatial–spectral information for ground analysis. However, in few-shot (FS) scenarios, the limited availability of training samples poses significant challenges in capturing the sample distribution under diverse environmental conditions. Semi-supervised learning has shown promise in exploring the distribution of unlabeled samples through pseudo-labels. Nonetheless, FS HSI classification encounters the issue of high intra-class spectral variability and inter-class spectral similarity, which often lead to the diffusion of unreliable pseudo-labels during the iterative process. In this paper, we propose a simple yet effective progressive pseudo-label selection strategy that leverages the spatial–spectral consistency of HSI pixel samples. By leveraging spatially aligned ground materials as connected regions with the same semantic and similar spectrum, pseudo-labeled samples were selected based on round-wise confidence scores. Samples within both spatially and semantically connected regions of FS samples were assigned pseudo-labels and joined subsequent training rounds. Moreover, considering the spatial positions of FS samples that may appear in diverse patterns, to fully utilize unlabeled samples that fall outside the neighborhood of FS samples but still belong to certain connected regions, we designed a matching active learning approach for expert annotation based on the temporal confidence difference. We identified samples with the highest training value in specific regions, utilizing the consistency between predictive labels and expert labels to decide whether to include the region or the sample itself in the subsequent semi-supervised iteration. Experiments on both classic and more recent HSI datasets demonstrated that the proposed base model achieved SOTA performance even with extremely rare labeled samples. Moreover, the extended version with active learning further enhances performance by involving limited additional annotation.

Original languageEnglish
Article number1747
JournalRemote Sensing
Volume16
Issue number10
DOIs
Publication statusPublished - May 2024
Externally publishedYes

Keywords

  • active learning
  • hyperspectral image classification
  • pseudo-label selection
  • semi-supervised learning

Fingerprint

Dive into the research topics of 'Enhancing Semi-Supervised Few-Shot Hyperspectral Image Classification via Progressive Sample Selection'. Together they form a unique fingerprint.

Cite this