An improved hybrid CNN for hyperspectral image classification

Abstract

The application of convolutional neural network (CNN) in hyperspectral image (HSI) classification has aroused widespread concern, especially spectral 1D CNN and spatial 2D CNN. Due to intense requirements of calculations and memories, 3D CNN, which is able to process jointly spectral and spatial features, has not yet been widely adopted. Recently, researchers have proposed a hybrid CNN for HSI classification, which obtained better performance than 3D CNN alone. Nevertheless, such a hybrid network has excessive parameters and limited capacity for feature utilization, where smaller training samples are prone to lower accuracy. This paper proposes an improved hybrid CNN to enhance the classification performance, which involves global average pooling, skip connection and appropriate adjustments of the convolution kernels and overall structure. Experimental results from benchmark HSI datasets suggest the effectiveness of our CNN for HSI classification in the situation of limited training set.