TY - GEN
T1 - Efficient Hyperspectral Imagery Classification Method with Lightweight Structure and Image Transformation-Based Data Augmentation
AU - Ivanitsa, Denis
AU - Wei, Wei
N1 - Publisher Copyright:
© 2022 IEEE.
PY - 2022
Y1 - 2022
N2 - 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.
AB - 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.
KW - Classification
KW - Data Augmentation
KW - Hyperspectral Image (HSI)
KW - Lightweight Design
UR - http://www.scopus.com/inward/record.url?scp=85140412176&partnerID=8YFLogxK
U2 - 10.1109/IGARSS46834.2022.9883408
DO - 10.1109/IGARSS46834.2022.9883408
M3 - 会议稿件
AN - SCOPUS:85140412176
T3 - International Geoscience and Remote Sensing Symposium (IGARSS)
SP - 3560
EP - 3563
BT - IGARSS 2022 - 2022 IEEE International Geoscience and Remote Sensing Symposium
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 2022 IEEE International Geoscience and Remote Sensing Symposium, IGARSS 2022
Y2 - 17 July 2022 through 22 July 2022
ER -