Deep CNN-based hyperspectral image classification using discriminative multiple spatial-spectral feature fusion

ABSTRACT Convolutional Neural Networks (CNNs) are widely used in various fields, and have shown good performance in hyperspectral image (HSI) classification. Recently, utilizing deep networks to learn spatial-spectral features has become of great interest. However, excessively increasing the depth of network may result in overfitting. Moreover, in HSI classification, the existing network models ignore the strong complementary yet correlated spatial-spectral information among different hierarchical layers. In order to address these two problems, a novel CNN-based method for HSI classification is proposed. Firstly, it considers fusing the outputs of recurrent two layers in each large convolutional block and thereby using the fusion result as the input of next layer, which facilitates the extraction of discriminative features. Next, the spectral-spatial features are extracted by cascading spectral features to four-scale spatial features from shallow to deep layers. Finally, a 1 1 convolution layer is used to interact and integrate information across channels. Without increasing the number of training samples and the size of pixel patches at the training stage, the proposed approach achieves the state-of-the-art results in the experiment on three well-known hyperspectral images.

[1]  Bo Du,et al.  Hyperspectral Remote Sensing Image Subpixel Target Detection Based on Supervised Metric Learning , 2014, IEEE Transactions on Geoscience and Remote Sensing.

[2]  Sheng Wan,et al.  Hyperspectral image classification based on robust discriminative extraction of multiple spectral-spatial features , 2019, International Journal of Remote Sensing.

[3]  Weiwei Song,et al.  Deep Hashing Neural Networks for Hyperspectral Image Feature Extraction , 2019, IEEE Geoscience and Remote Sensing Letters.

[4]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[5]  Antonio J. Plaza,et al.  Deep Pyramidal Residual Networks for Spectral–Spatial Hyperspectral Image Classification , 2019, IEEE Transactions on Geoscience and Remote Sensing.

[6]  Jon Atli Benediktsson,et al.  Deep Learning for Hyperspectral Image Classification: An Overview , 2019, IEEE Transactions on Geoscience and Remote Sensing.

[7]  Liang Xiao,et al.  An improved composite kernel framework for hyperspectral image classification using canonical correlation analysis , 2019, Remote Sensing Letters.

[8]  Antonio J. Plaza,et al.  Probabilistic-Kernel Collaborative Representation for Spatial–Spectral Hyperspectral Image Classification , 2016, IEEE Transactions on Geoscience and Remote Sensing.

[9]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[10]  Haokui Zhang,et al.  Spectral-spatial classification of hyperspectral imagery using a dual-channel convolutional neural network , 2017 .

[11]  Jonathan Cheung-Wai Chan,et al.  Learning and Transferring Deep Joint Spectral–Spatial Features for Hyperspectral Classification , 2017, IEEE Transactions on Geoscience and Remote Sensing.

[12]  Zhiming Luo,et al.  Spectral–Spatial Residual Network for Hyperspectral Image Classification: A 3-D Deep Learning Framework , 2018, IEEE Transactions on Geoscience and Remote Sensing.

[13]  Kilian Q. Weinberger,et al.  Densely Connected Convolutional Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).