Domain-Invariant Few-Shot Contrastive Learning for Hyperspectral Image Classification

Wenchen Chen, Yanmei Zhang*, Jianping Chu, Xingbo Wang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

In Hyperspectral Image (HSI) classification, acquiring large quantities of high-quality labeled samples is typically costly and impractical. Traditional deep learning methods are limited in such scenarios due to their dependence on sample quantities. To address this challenge, researchers have turned to Few-Shot Learning (FSL). Although existing FSL methods improve classification performance by enhancing domain invariance through domain adaptation, they often overlook the critical issue of high inter-class similarity and large intra-class variability. Moreover, during domain alignment, features of different categories within the same domain may become confused. To address these issues, this paper proposes a novel Domain-Invariant Few-Shot Contrastive Learning (DIFSCL) method, which combines domain adaptation and contrastive learning strategies to not only learn domain-invariant features but also significantly enhance inter-class discriminability. Based on this, we further design a multi-scale adaptive attention mechanism for a hyperspectral feature extraction network to more effectively extract and optimize generalized features within the DIFSCL framework, significantly improving intra-class consistency and inter-class discriminative capability of features. Experimental results on three widely used HSI datasets demonstrate that our method significantly outperforms existing techniques in few-shot classification tasks.

Original languageEnglish
Article number11053
JournalApplied Sciences (Switzerland)
Volume14
Issue number23
DOIs
Publication statusPublished - Dec 2024

Keywords

  • attention mechanism
  • contrastive learning
  • domain adaptation
  • few-shot learning
  • hyperspectral image classification

Fingerprint

Dive into the research topics of 'Domain-Invariant Few-Shot Contrastive Learning for Hyperspectral Image Classification'. Together they form a unique fingerprint.

Cite this