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

被引:24
作者
Wang, Weiquan [1 ]
Chen, Yushi [1 ]
He, Xin [1 ]
Li, Zhaokui [2 ]
机构
[1] Harbin Inst Technol, Sch Elect & Informat Engn, Harbin 150001, Peoples R China
[2] Shenyang Aerosp Univ, Sch Comp Sci, Shenyang 110136, Peoples R China
关键词
Training; Convolutional neural networks; Handheld computers; Hyperspectral imaging; Data models; Feature extraction; Principal component analysis; Convolutional neural network (CNN); data augmentation; hyperspectral image (HSI) classification; limited training samples; siamese network;
D O I
10.1109/LGRS.2021.3103180
中图分类号
P3 [地球物理学]; P59 [地球化学];
学科分类号
0708 ; 070902 ;
摘要
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.
引用
收藏
页数:5
相关论文
共 19 条
[11]   Supervised Deep Feature Extraction for Hyperspectral Image Classification [J].
Liu, Bing ;
Yu, Xuchu ;
Zhang, Pengqiang ;
Yu, Anzhu ;
Fu, Qiongying ;
Wei, Xiangpo .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (04) :1909-1921
[12]   Classification of hyperspectral remote sensing images with support vector machines [J].
Melgani, F ;
Bruzzone, L .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2004, 42 (08) :1778-1790
[13]   A new deep convolutional neural network for fast hyperspectral image classification [J].
Paoletti, M. E. ;
Haut, J. M. ;
Plaza, J. ;
Plaza, A. .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2018, 145 :120-147
[14]   Neighboring Region Dropout for Hyperspectral Image Classification [J].
Paoletti, Mercedes E. ;
Haut, Juan M. ;
Plaza, Javier ;
Plaza, Antonio .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (06) :1032-1036
[15]   Capsule Networks for Hyperspectral Image Classification [J].
Paoletti, Mercedes E. ;
Haut, Juan Mario ;
Fernandez-Beltran, Ruben ;
Plaza, Javier ;
Plaza, Antonio ;
Li, Jun ;
Pla, Filiberto .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (04) :2145-2160
[16]   Deep Pyramidal Residual Networks for Spectral-Spatial Hyperspectral Image Classification [J].
Paoletti, Mercedes E. ;
Mario Haut, Juan ;
Fernandez-Beltran, Ruben ;
Plaza, Javier ;
Plaza, Antonio J. ;
Pla, Filiberto .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (02) :740-754
[17]   HybridSN: Exploring 3-D-2-D CNN Feature Hierarchy for Hyperspectral Image Classification [J].
Roy, Swalpa Kumar ;
Krishna, Gopal ;
Dubey, Shiv Ram ;
Chaudhuri, Bidyut B. .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (02) :277-281
[18]   CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features [J].
Yun, Sangdoo ;
Han, Dongyoon ;
Oh, Seong Joon ;
Chun, Sanghyuk ;
Choe, Junsuk ;
Yoo, Youngjoon .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6022-6031
[19]   Deformable Convolutional Neural Networks for Hyperspectral Image Classification [J].
Zhu, Jian ;
Fang, Leyuan ;
Ghamisi, Pedram .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2018, 15 (08) :1254-1258