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

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.