Deep Feature Aggregation Network for Hyperspectral Remote Sensing Image Classification

被引:28
作者
Zhang, Chunju [1 ,2 ]
Li, Guandong [1 ]
Lei, Runmin [1 ]
Du, Shihong [3 ]
Zhang, Xueying [4 ]
Zheng, Hui [5 ]
Wu, Zhaofu [1 ]
机构
[1] Hefei Univ Technol, Sch Civil Engn, Hefei 230009, Peoples R China
[2] Hefei Univ Technol, Intelligent Interconnected Syst Lab Anhui Prov, Hefei 230009, Peoples R China
[3] Peking Univ, Inst Remote Sensing & GIS, Beijing 100871, Peoples R China
[4] Nanjing Normal Univ, Key Lab Virtual Geog Environm, Nanjing 210023, Peoples R China
[5] Henan Univ, Key Lab Geospatial Technol Middle & Yellow River, Minist Educ, Kaifeng 475004, Peoples R China
基金
中国国家自然科学基金;
关键词
Feature extraction; Aggregates; Laboratories; Hyperspectral imaging; Data mining; Convolution; Dense connectivity; feature fusion; hyperspectral image classification; residual learning; 3-D convolutional neural network (3D-CNN); RESNET;
D O I
10.1109/JSTARS.2020.3020733
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Hyperspectral remote sensing images (HSIs) are rich in spectral-spatial information. The deep learning models can help to automatically extract and discover this rich information from HSIs for classifying HSIs. However, the sampling of the models and the design of the hyperparameters depend on the number of samples and the size of each sample's input space. In the case of limited samples, the description dimension of features is also limited and overfitting to other remote sensing image datasets is evident. This study proposes a novel deep feature aggregation network for HSI classification based on a 3-D convolutional neural network from the perspective of feature aggregation patterns. By introducing the residual learning and dense connectivity strategies, we established a deep feature residual network (DFRN) and a deep feature dense network (DFDN) to exploit the low-, middle-, and high-level features in HSIs. For the Indian Pines and Kennedy Space Center datasets, the DFRN model was determined to be more accurate. On the Pavia University dataset, both the DFDN and DFRN have basically the same accuracy, but the DFDN has faster convergence speed and more stable performance on the validation set than the DFRN. Therefore, when faced with different HSI data, the corresponding aggregation method can be chosen more flexibly according to the requirements on number of training samples and the convergence speed. This is beneficial in the HSI classification.
引用
收藏
页码:5314 / 5325
页数:12
相关论文
共 35 条
[1]  
[Anonymous], 2012, COURSERA NEURAL NETW
[2]   Classification and feature extraction for remote sensing images from urban areas based on morphological transformations [J].
Benediktsson, JA ;
Pesaresi, M ;
Arnason, K .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2003, 41 (09) :1940-1949
[3]   Composite kernels for hyperspectral image classification [J].
Camps-Valls, G ;
Gomez-Chova, L ;
Muñoz-Marí, J ;
Vila-Francés, J ;
Calpe-Maravilla, J .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2006, 3 (01) :93-97
[4]  
Chang CheinI, 2003, Hyperspectral Imaging: Techniques for Spectral Detection and Classification, DOI 10.1007/978-1-4419-9170-6
[5]   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
[6]   Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network [J].
Chen, Yushi ;
Zhao, Xing ;
Jia, Xiuping .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) :2381-2392
[7]   A spatial-spectral kernel-based approach for the classification of remote-sensing images [J].
Fauvel, M. ;
Chanussot, J. ;
Benediktsson, J. A. .
PATTERN RECOGNITION, 2012, 45 (01) :381-392
[8]  
Glorot X., 2011, P 14 INT C ART INT S, P315
[9]   The problem of overfitting [J].
Hawkins, DM .
JOURNAL OF CHEMICAL INFORMATION AND COMPUTER SCIENCES, 2004, 44 (01) :1-12
[10]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778