Skip to main navigation Skip to search Skip to main content

Dual-Phase Framework for Few-Shot Hyperspectral Image Classification With Spatiospectral Masked Autoencoder and Episode Training

  • Wijayanti Nurul Khotimah*
  • , Mohammed Bennamoun
  • , Farid Boussaid
  • , Lian Xu
  • , Ferdous Sohel
  • *Corresponding author for this work
  • University of Western Australia
  • Murdoch University

Research output: Contribution to journalArticlepeer-review

Abstract

This article introduces a two-phase learning approach for hyperspectral image (HSI) classification using few-shot learning (FSL). For the first phase, we present a novel spatiospectral masked autoencoder (ssMAE)-an advanced self-supervised learner. For the ssMAE backbone network, we designed a transformer encoder- decoder network, where we replaced the linear layer that is used as the initial feature embedding witha 3-D convolutional layer to better extract local spectral–spatial features from 3-D visible sub-patches. By tapping intovast unlabeled data, the ssMAE learns general HSI features. In the second phase, the ssMAE encoder is fine-tuned to extract discriminative features for classification using the few-shot labeled training samples. This is achieved through a unique hybrid episode learning method that integrates the ssMAE encoder in a prototypical network (PN). We innovate with a mix of global and local prototypes (combined global–local (CGL) prototype) torefine label predictions. This technique maximizes data usage, focuses on specific samples,and mitigates issues from subpar episodes. Tested on three HSI data sets, our approach outperforms alternative few-shot methods. The code will be made publicly available at https://github.com/Weejaa04/SSMAE.

Original languageEnglish
Article number4404616
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume63
DOIs
Publication statusPublished - 2025

Keywords

  • Combination of global and local prototypes
  • few-shot learning (FSL)
  • hybrid episode learning
  • hyperspectral image (HSI) classification
  • prototypical network (PN)
  • spatiospectral masked autoencoder (ssMAE)

Fingerprint

Dive into the research topics of 'Dual-Phase Framework for Few-Shot Hyperspectral Image Classification With Spatiospectral Masked Autoencoder and Episode Training'. Together they form a unique fingerprint.

Cite this