Skip to main navigation Skip to search Skip to main content

Soft Augmentation-Based Siamese CNN for Hyperspectral Image Classification with Limited Training Samples

  • School of Electronics and Information Engineering, Harbin Institute of Technology
  • Shenyang Aerospace University

Research output: Contribution to journalArticlepeer-review

Abstract

The lack of training samples remains one of the major obstacles in applying convolutional neural networks (CNNs) to the hyperspectral image (HSI) classification. In this letter, the accurate classification of HSI with limited training samples is investigated. Due to the advantages of minimizing the distance between samples in the same class and maximizing the distance between samples in different classes, siamese CNN is used for HSI classification with limited training samples. After that, to improve the classification performance, data augmentation is investigated for siamese CNN-based HSI classification. Specifically, pair data augmentation based on CutMix is proposed to generate the training pairs of the same or different classes and the new generated training pairs are used to train siamese CNN. Traditional data augmentation methods simply generate new training samples. It is not proper when data augmentation methods keep the loss function and change the content of the input at the same time. Therefore, a soft-loss-based siamese CNN, which changes its loss according to the coupled replacement data augmentation, is proposed to further address the HSI classification with limited training samples. In the experimental part, on two widely used hyperspectral datasets, the influences of different training samples and window sizes are discussed, and the experimental results reveal that the proposed soft-augmentation-based siamese CNN provides competitive results with limited training samples compared with state-of-the-art methods.

Original languageEnglish
JournalIEEE Geoscience and Remote Sensing Letters
Volume19
DOIs
StatePublished - 2022
Externally publishedYes

Keywords

  • Convolutional neural network (CNN)
  • data augmentation
  • hyperspectral image (HSI) classification
  • limited training samples
  • siamese network

Fingerprint

Dive into the research topics of 'Soft Augmentation-Based Siamese CNN for Hyperspectral Image Classification with Limited Training Samples'. Together they form a unique fingerprint.

Cite this