Skip to main navigation Skip to search Skip to main content

Self-Supervised Learning with Prediction of Image Scale and Spectral Order for Hyperspectral Image Classification

  • Xiaofei Yang
  • , Weijia Cao*
  • , Yao Lu
  • , Yicong Zhou*
  • *Corresponding author for this work
  • University of Macau
  • CAS - Aerospace Information Research Institute
  • Yangtze Three Gorges Technology and Economy Development Co Ltd.
  • Harbin Institute of Technology Shenzhen

Research output: Contribution to journalArticlepeer-review

Abstract

In recent years, convolutional neural networks (CNNs) have achieved great success in hyperspectral image (HSI) classification attributed to their unparalleled capacity to extract the local information. However, to successfully learn the high-level semantic image features, they always require massive amounts of manually labeled data during the training process, which is expensive, scarce, and impractical, and severely hinders the improvement of supervised deep learning methods. To alleviate these burdens, we present self-supervised learning (SSL) methods for HSI classification by a pretraining model using extensive unlabeled data and fine-tuning the HSI target classification. In this article, we propose a new method for learning image characteristics by training a CNN to recognize the image scale (IS) that is applied to the HSIs. In addition, we propose a multipretext task (MT) method to learn stable and good feature representations combing two different pretext task methods and contrastive loss function. We evaluate the proposed methods in SSL benchmarks on four benchmark HSIs datasets. The experiment results demonstrate that the proposed methods outperform the traditional supervised deep learning methods when large amounts of unlabeled HSIs data are used. Moreover, it demonstrates that the SSL method is promising to alleviate dependence on manually labeled data of HSI classification. Finally, our research contributes to the creation and refinement of SSL methods for pretextual tasks within the HSIs community.

Original languageEnglish
Article number5545715
JournalIEEE Transactions on Geoscience and Remote Sensing
Volume60
DOIs
StatePublished - 2022
Externally publishedYes

Keywords

  • Hyperspectral image (HSI) classification
  • limited labeled samples
  • self-supervised learning (SSL)
  • unsupervised learning

Fingerprint

Dive into the research topics of 'Self-Supervised Learning with Prediction of Image Scale and Spectral Order for Hyperspectral Image Classification'. Together they form a unique fingerprint.

Cite this