Three-Dimensional ResNeXt Network Using Feature Fusion and Label Smoothing for Hyperspectral Image Classification

被引:26
作者
Wu, Peida [1 ]
Cui, Ziguan [1 ]
Gan, Zongliang [1 ]
Liu, Feng [1 ]
机构
[1] Nanjing Univ Posts & Telecommun, Coll Telecommun & Informat Engn, Nanjing 210003, Peoples R China
基金
中国国家自然科学基金;
关键词
deep learning; hyperspectral image classification; group convolution; spectral-spatial features; SPECTRAL-SPATIAL CLASSIFICATION;
D O I
10.3390/s20061652
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
In recent years, deep learning methods have been widely used in the hyperspectral image (HSI) classification tasks. Among them, spectral-spatial combined methods based on the three-dimensional (3-D) convolution have shown good performance. However, because of the three-dimensional convolution, increasing network depth will result in a dramatic rise in the number of parameters. In addition, the previous methods do not make full use of spectral information. They mostly use the data after dimensionality reduction directly as the input of networks, which result in poor classification ability in some categories with small numbers of samples. To address the above two issues, in this paper, we designed an end-to-end 3D-ResNeXt network which adopts feature fusion and label smoothing strategy further. On the one hand, the residual connections and split-transform-merge strategy can alleviate the declining-accuracy phenomenon and decrease the number of parameters. We can adjust the hyperparameter cardinality instead of the network depth to extract more discriminative features of HSIs and improve the classification accuracy. On the other hand, in order to improve the classification accuracies of classes with small numbers of samples, we enrich the input of the 3D-ResNeXt spectral-spatial feature learning network by additional spectral feature learning, and finally use a loss function modified by label smoothing strategy to solve the imbalance of classes. The experimental results on three popular HSI datasets demonstrate the superiority of our proposed network and an effective improvement in the accuracies especially for the classes with small numbers of training samples.
引用
收藏
页数:29
相关论文
共 48 条
[1]  
[Anonymous], PROC CVPR IEEE
[2]  
[Anonymous], 2016, P IEEE INT C COMM MA
[3]  
[Anonymous], 2017, COMMUN ACM, DOI DOI 10.1145/3065386
[4]  
[Anonymous], 2003, HYPERSPECTRAL IMAGIN
[5]   SSDC-DenseNet: A Cost-Effective End-to-End Spectral-Spatial Dual-Channel Dense Network for Hyperspectral Image Classification [J].
Bai, Yutong ;
Zhang, Qifan ;
Lu, Zexin ;
Zhang, Yi .
IEEE ACCESS, 2019, 7 :84876-84889
[6]   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
[7]   Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks [J].
Chen, Yushi ;
Jiang, Hanlu ;
Li, Chunyang ;
Jia, Xiuping ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10) :6232-6251
[8]   Deep Learning-Based Classification of Hyperspectral Data [J].
Chen, Yushi ;
Lin, Zhouhan ;
Zhao, Xing ;
Wang, Gang ;
Gu, Yanfeng .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (06) :2094-2107
[9]   Advances in Spectral-Spatial Classification of Hyperspectral Images [J].
Fauvel, Mathieu ;
Tarabalka, Yuliya ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Tilton, James C. .
PROCEEDINGS OF THE IEEE, 2013, 101 (03) :652-675
[10]   Spectral and Spatial Classification of Hyperspectral Data Using SVMs and Morphological Profiles [J].
Fauvel, Mathieu ;
Benediktsson, Jon Atli ;
Chanussot, Jocelyn ;
Sveinsson, Johannes R. .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2008, 46 (11) :3804-3814