A New Spectral-Spatial Pseudo-3D Dense Network for Hyperspectral Image Classification.

Ailin Li,Zhaowei Shang
DOI: https://doi.org/10.1109/ijcnn.2019.8851917
2019-01-01
Abstract:The recent research of hyperspectral images(HSIs) classification depicts that taking spectral-spatial features into account can considerably improve accuracy. Since HSI is a 3D cube datum, amounts of 3D network structures emerged to extract spectral-spatial features consecutively. In this paper, we present a simple spectral-spatial classification framework(SSP3DNet) based on densely connected structure with Pseudo-3D block for HSIs. Firstly, a data augmentation strategy was implemented to solve the problem of limited and uneven training samples in the data preprocessing step. Secondly, the Pseudo-3D block can capture both spectral and spatial features simultaneously which is more economic by decreasing the number of parameters compared with traditional 3D convolution network. Then DenseNet learning framework is utilized to ease the training of networks as well as improving the classification performance. Especially, to prevent over-fitting, some tricks like early stopping, LeakyReLu, batch normalization, and dropout layers are used in our SSP3DNet which enable the SSP3DNet to obtain accuracy within 80 epochs. Experimental results on two well-known hyperspectral datasets show that the proposed SSP3DNet method achieves the best classification accuracy in comparison with lately traditional and deep-learning-based methods.
What problem does this paper attempt to address?