Hyperspectral Image Classification via a Novel Spectral-Spatial 3D ConvLSTM-CNN

被引:24
作者
Farooque, Ghulam [1 ]
Xiao, Liang [1 ]
Yang, Jingxiang [1 ]
Sargano, Allah Bux [2 ]
机构
[1] Nanjing Univ Sci & Technol, Sch Comp Sci & Engn, Nanjing 210094, Peoples R China
[2] COMSATS Univ Islamabad, Dept Comp Sci, Lahore 54000, Pakistan
基金
中国国家自然科学基金;
关键词
3D ConvLSTM; hyperspectral image classification; 3D CNN; spectral-spatial feature extraction; residual network; deep learning; FEATURE-EXTRACTION; NEURAL-NETWORKS;
D O I
10.3390/rs13214348
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
In recent years, deep learning-based models have produced encouraging results for hyperspectral image (HSI) classification. Specifically, Convolutional Long Short-Term Memory (ConvLSTM) has shown good performance for learning valuable features and modeling long-term dependencies in spectral data. However, it is less effective for learning spatial features, which is an integral part of hyperspectral images. Alternatively, convolutional neural networks (CNNs) can learn spatial features, but they possess limitations in handling long-term dependencies due to the local feature extraction in these networks. Considering these factors, this paper proposes an end-to-end Spectral-Spatial 3D ConvLSTM-CNN based Residual Network (SSCRN), which combines 3D ConvLSTM and 3D CNN for handling both spectral and spatial information, respectively. The contribution of the proposed network is twofold. Firstly, it addresses the long-term dependencies of spectral dimension using 3D ConvLSTM to capture the information related to various ground materials effectively. Secondly, it learns the discriminative spatial features using 3D CNN by employing the concept of the residual blocks to accelerate the training process and alleviate the overfitting. In addition, SSCRN uses batch normalization and dropout to regularize the network for smooth learning. The proposed framework is evaluated on three benchmark datasets widely used by the research community. The results confirm that SSCRN outperforms state-of-the-art methods with an overall accuracy of 99.17%, 99.67%, and 99.31% over Indian Pines, Salinas, and Pavia University datasets, respectively. Moreover, it is worth mentioning that these excellent results were achieved with comparatively fewer epochs, which also confirms the fast learning capabilities of the SSCRN.
引用
收藏
页数:21
相关论文
共 67 条
  • [1] A Globally Regularized Joint Neural Architecture for Music Classification
    Ashraf, Mohsin
    Geng, Guohua
    Wang, Xiaofeng
    Ahmad, Farooq
    Abid, Fazeel
    [J]. IEEE ACCESS, 2020, 8 : 220980 - 220989
  • [2] Hyperspectral Region Classification Using a Three-Dimensional Gabor Filterbank
    Bau, Tien C.
    Sarkar, Subhadip
    Healey, Glenn
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2010, 48 (09): : 3457 - 3464
  • [3] Random forest in remote sensing: A review of applications and future directions
    Belgiu, Mariana
    Dragut, Lucian
    [J]. ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2016, 114 : 24 - 31
  • [4] Hyperspectral Remote Sensing Data Analysis and Future Challenges
    Bioucas-Dias, Jose M.
    Plaza, Antonio
    Camps-Valls, Gustavo
    Scheunders, Paul
    Nasrabadi, Nasser M.
    Chanussot, Jocelyn
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING MAGAZINE, 2013, 1 (02) : 6 - 36
  • [5] Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks
    Chen, Yushi
    Jiang, Hanlu
    Li, Chunyang
    Jia, Xiuping
    Ghamisi, Pedram
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10): : 6232 - 6251
  • [6] Spectral-Spatial Classification of Hyperspectral Data Based on Deep Belief Network
    Chen, Yushi
    Zhao, Xing
    Jia, Xiuping
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2015, 8 (06) : 2381 - 2392
  • [7] Deep Learning-Based Classification of Hyperspectral Data
    Chen, Yushi
    Lin, Zhouhan
    Zhao, Xing
    Wang, Gang
    Gu, Yanfeng
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2014, 7 (06) : 2094 - 2107
  • [8] Hyperspectral Images Classification Based on Dense Convolutional Networks with Spectral-Wise Attention Mechanism
    Fang, Bei
    Li, Ying
    Zhang, Haokui
    Chan, Jonathan Cheung-Wai
    [J]. REMOTE SENSING, 2019, 11 (02)
  • [9] Farooque G, 2016, INT CONF FRONT INFO, P93, DOI [10.1109/FIT.2016.025, 10.1109/FIT.2016.23]
  • [10] Advances in Spectral-Spatial Classification of Hyperspectral Images
    Fauvel, Mathieu
    Tarabalka, Yuliya
    Benediktsson, Jon Atli
    Chanussot, Jocelyn
    Tilton, James C.
    [J]. PROCEEDINGS OF THE IEEE, 2013, 101 (03) : 652 - 675