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 条
[1]   Deep Learning for Classification of Hyperspectral Data [J].
Audebert, Nicolas ;
Le Saux, Bertrand ;
Lefevre, Sebastien .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2019, 7 (02) :159-173
[2]   Hyperspectral Remote Sensing Data Analysis and Future Challenges [J].
Bioucas-Dias, Jose M. ;
Plaza, Antonio ;
Camps-Valls, Gustavo ;
Scheunders, Paul ;
Nasrabadi, Nasser M. ;
Chanussot, Jocelyn .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2013, 1 (02) :6-36
[3]   Classification of Hyperspectral Images by Using Extended Morphological Attribute Profiles and Independent Component Analysis [J].
Dalla Mura, Mauro ;
Villa, Alberto ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Bruzzone, Lorenzo .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2011, 8 (03) :542-546
[4]   Deep Metric Learning-Based Feature Embedding for Hyperspectral Image Classification [J].
Deng, Bin ;
Jia, Sen ;
Shi, Daming .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2020, 58 (02) :1422-1435
[5]   New Frontiers in Spectral-Spatial Hyperspectral Image Classification The latest advances based on mathematical morphology, Markov random fields, segmentation, sparse representation, and deep learning [J].
Ghamisi, Pedram ;
Maggiori, Emmanuel ;
Li, Shutao ;
Souza, Roberto ;
Tarabalka, Yuliya ;
Moser, Gabriele ;
De Giorgi, Andrea ;
Fang, Leyuan ;
Chen, Yushi ;
Chi, Mingmin ;
Serpico, Sebastiano B. ;
Benediktsson, Jon Atli .
IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2018, 6 (03) :10-43
[6]   A CNN-Based Spatial Feature Fusion Algorithm for Hyperspectral Imagery Classification [J].
Guo, Alan J. X. ;
Zhu, Fei .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (09) :7170-7181
[7]   Hyperspectral Image Classification Using Random Occlusion Data Augmentation [J].
Haut, Juan Mario ;
Paoletti, Mercedes E. ;
Plaza, Javier ;
Plaza, Antonio ;
Li, Jun .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2019, 16 (11) :1751-1755
[8]   Object-Oriented Mangrove Species Classification Using Hyperspectral Data and 3-D Siamese Residual Network [J].
He, Zhi ;
Shi, Qian ;
Liu, Kai ;
Cao, Jingjing ;
Zhan, Wen ;
Cao, Beifen .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (12) :2150-2154
[9]  
Koch G, 2015, ICML DEEP LEARNING W, P2
[10]   Hyperspectral Image Classification Using Deep Pixel-Pair Features [J].
Li, Wei ;
Wu, Guodong ;
Zhang, Fan ;
Du, Qian .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2017, 55 (02) :844-853