IEEE Access (Jan 2021)
Hyperspectral Image Classification Based on Stacked Contractive Autoencoder Combined With Adaptive Spectral-Spatial Information
Abstract
Hyperspectral image (HSI) contain abundant spectral and spatial information, enabling the accurate classification of ground objects. However, many existing machine learning methods have poor performance, and some existing CNN-based methods require high computational power, which considerably limits their real-world applications. To address these issues, in this paper, we propose an alternative HSI classification method based on the stacked contractive autoencoder (SCAE) and adaptive spectral-spatial information to improve the accuracy of HSI classification. Specifically, the non-subsampled shearlet transform (NSST) with the guided filtering (NG) enhances spatial structure information. Subsequently, we present an adaptive spatial information extraction method to extract the spatial information of pixels. Furthermore, we propose an HSI classification network, called SCAE-LR, for feature extraction and classification. The SCAE is implemented to extract the adaptive spectral-spatial feature, and a logistic regression (LR) layer is employed for classification. Extensive experiments on the Indian Pines data set and the Pavia University data set demonstrate the superior performance of our method.
Keywords