A Fast Dense Spectral-Spatial Convolution Network Framework for Hyperspectral Images Classification

被引:300
作者
Wang, Wenju [1 ]
Dou, Shuguang [1 ]
Jiang, Zhongmin [1 ]
Sun, Liujie [1 ]
机构
[1] Univ Shanghai Sci & Technol, Coll Commun & Art Design, SH 021, Shanghai, Peoples R China
关键词
hyperspectral image classification; densely connected convolutional neural network; deep learning; parametric rectified linear unit;
D O I
10.3390/rs10071068
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Recent research shows that deep-learning-derived methods based on a deep convolutional neural network have high accuracy when applied to hyperspectral image (HSI) classification, but long training times. To reduce the training time and improve accuracy, in this paper we propose an end-to-end fast dense spectral-spatial convolution (FDSSC) framework for HSI classification. The FDSSC framework uses different convolutional kernel sizes to extract spectral and spatial features separately, and the valid convolution method to reduce the high dimensions. Densely-connected structures-the input of each convolution consisting of the output of all previous convolution layers-was used for deep learning of features, leading to extremely accurate classification. To increase speed and prevent overfitting, the FDSSC framework uses a dynamic learning rate, parametric rectified linear units, batch normalization, and dropout layers. These attributes enable the FDSSC framework to achieve accuracy within as few as 80 epochs. The experimental results show that with the Indian Pines, Kennedy Space Center, and University of Pavia datasets, the proposed FDSSC framework achieved state-of-the-art performance compared with existing deep-learning-based methods while significantly reducing the training time.
引用
收藏
页数:19
相关论文
共 30 条
[1]  
Abadi M., 2015, PREPRINT
[2]  
[Anonymous], GITHUB REPOSITORY
[3]  
[Anonymous], 2015, ARXIV PREPRINT ARXIV
[4]  
[Anonymous], PROC CVPR IEEE
[5]  
[Anonymous], 2013, INT C MACHINE LEARNI
[6]  
[Anonymous], 2016, ARXIV E PRINTS
[7]  
[Anonymous], 2015, IEEE I CONF COMP VIS, DOI DOI 10.1109/ICCV.2015.123
[8]  
[Anonymous], 2017, P 2017 IEEE C PATT R
[9]  
[Anonymous], 2015, PROCIEEE CONFCOMPUT
[10]   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