Efficient Hyperspectral Imagery Classification Method with Lightweight Structure and Image Transformation-Based Data Augmentation

Denis Ivanitsa, Wei Wei

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

5 Scopus citations

Abstract

With the increasing popularity of deep learning models in the hyperspectral image (HSI) classification field, more and more complex methods have been proposed. However, an increase in accuracy was accompanied by an increase in model size and computational complexity. As a result, application of these models will be limited in solutions with strict hardware specifications, as well as solutions with a limited training set. To reduce the required number of parameters and total FLOPs, we design a lightweight model for HSI classification, based on a ghost module. Specifically, we use 3D ghost modules to build an efficient 3D-CNN network termed TinyNet. To further increase the performance of our TinyNet model, a combination of cross-entropy and contrastive center loss is utilized for training. Additionally, we design a novel augmentation technique based on image transformations. The proposed lightweight model, together with our augmentation technique, can lead to satisfactory HSI classification results. Experimental results on two HSI datasets demonstrate the effectiveness of the proposed HSI classification method when compared to the competing techniques.

Original languageEnglish
Title of host publicationIGARSS 2022 - 2022 IEEE International Geoscience and Remote Sensing Symposium
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages3560-3563
Number of pages4
ISBN (Electronic)9781665427920
DOIs
StatePublished - 2022
Event2022 IEEE International Geoscience and Remote Sensing Symposium, IGARSS 2022 - Kuala Lumpur, Malaysia
Duration: 17 Jul 202222 Jul 2022

Publication series

NameInternational Geoscience and Remote Sensing Symposium (IGARSS)
Volume2022-July

Conference

Conference2022 IEEE International Geoscience and Remote Sensing Symposium, IGARSS 2022
Country/TerritoryMalaysia
CityKuala Lumpur
Period17/07/2222/07/22

Keywords

  • Classification
  • Data Augmentation
  • Hyperspectral Image (HSI)
  • Lightweight Design

Fingerprint

Dive into the research topics of 'Efficient Hyperspectral Imagery Classification Method with Lightweight Structure and Image Transformation-Based Data Augmentation'. Together they form a unique fingerprint.

Cite this