Hyperspectral remote sensing image classification based on enhanced pseudo 3D features and salient band selection

Abstract

Hyperspectral classification is a research hotspot in the field of remote sensing. Recently, 3D convolutional neural networks (CNNs) have achieved better classification performances than traditional machine learning algorithms. However, because of the large kernel size and spectral redundancy, the classification accuracy and efficiency of existing CNN-based methods are still restrained. In this paper, a lightweight model based on the enhanced pseudo 3D features and salient band selection is proposed for HSI classification. Specifically, an enhanced pseudo 3D convolution block is constructed to extract spectral-spatial features with less parameters. Then, a salient band selection block without parameters is designed to relieve the spectral redundancy. To obtain the diverse spectral dependency, a local-connected layer is introduced to explore the interactions between adjacent bands. By integrating these blocks, deep spectral-spatial pseudo 3D features can be well prepared for classification. Experiments on three HSI data sets show that the proposed model outperforms the state-of-the-arts. Source code is available at https://github.com/ningyang-li/EP3FEN.