Spectral-Spatial Classification of Hyperspectral Images Using CNNs and Approximate Sparse Multinomial Logistic Regression

Abstract

We propose a technique for training convolutional neural networks (CNNs) in which the convolutional layers are trained using a gradient descent based method and the classification layer is trained using a second order method called approximate sparse multinomial logistic regression (ASMLR) which also provides a spatial smoothing procedure that increases the classification accuracy for hyperspectral images. ASMLR performs well on hyperspectral images, and CNNs are known to give good results in many applications such as image classification and object recognition. Thus, the proposed technique allows us to improve the performance of CNNs by training the whole network with an end-to-end framework. This approach takes advantage of convolutional layers for spectral feature extraction, and of the softmax classification layer for feature selection with sparsity constraints, and an intrinsic learning rate adjustment mechanism. In classification, we also use a spatial smoothing method. The proposed method was evaluated on two hyperspectral images for spectral-spatial land cover classification, and the results have shown that it outperforms the CNN and the ASMLR classifiers when they are used separately.