Skip to main navigation Skip to search Skip to main content

Few-Shot Learning Based on Embedded Self-Distillation and Adaptive Wasserstein Distance for Hyperspectral Image Classification

  • Wenjie Li
  • , Shizhe Shang
  • , Ronghua Shang
  • , Dongzhu Feng
  • , Weitong Zhang
  • , Chao Wang
  • , Jie Feng
  • , Songhua Xu
  • Xidian University
  • Tianjin Normal University
  • Zhejiang Lab
  • The Second Affiliated Hospital of Xi'an Jiaotong University

Research output: Contribution to journalArticlepeer-review

12 Scopus citations

Abstract

Due to the domain shift, it is challenging to achieve ideal experimental results for cross-domain few-shot learning (FSL) in hyperspectral image (HSI) classification. Most existing FSL algorithms are impacted by the limited samples, and they do not effectively leverage the representations from different layers of the network. Therefore, this article proposes an FSL based on embedded self-distillation and adaptive Wasserstein (ESAW-FSL) distance for HSI classification. First, the embedding self-distillation network is proposed in the feature extraction process of the source domain (SD) and the target domain (TD). The embedding self-distillation network utilizes self-distillation from different perspectives to get discriminative features. In the SD, the mask evaluation of embedded features is employed to guarantee the learning of guiding features. Second, a domain adaptation based on adaptive Wasserstein distance is designed to alleviate the domain shift problem between the domains. A lightweight feature correlation network learns the comprehensive cost matrix in the Wasserstein distance adaptively, and the obtained cost matrix helps achieve domain adaptation by an iterative algorithm. Finally, a focal loss based on double softening is adopted in the process of FSL. The probability is double softened to improve the ratio of correctly classifying hard samples. Experiments are conducted on three widely used hyperspectral datasets and compared with six state-of-the-art algorithms. The overall accuracy (OA) and average accuracy (AA) are achieved in multiple experiments, demonstrating the effectiveness of ESAW-FSL.

Original languageEnglish
Article number5502315
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume63
DOIs
StatePublished - 2025
Externally publishedYes

Keywords

  • Cross-domain
  • deep learning
  • few-shot learning (FSL)
  • hyperspectral image (HSI) classification
  • self-distillation

Fingerprint

Dive into the research topics of 'Few-Shot Learning Based on Embedded Self-Distillation and Adaptive Wasserstein Distance for Hyperspectral Image Classification'. Together they form a unique fingerprint.

Cite this