An End-to-End Hyperspectral Image Classification Method Using Deep Convolutional Neural Network with Spatial Constraint

Zhuang Jia,Wenkai Lu
DOI: https://doi.org/10.1109/lgrs.2020.3008051
IF: 5.343
2021-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:Hyperspectral image (HSI) classification is of vital importance in remote sensing-related applications. Various approaches, including the recently popular convolutional neural network (CNN)-based models, are proposed to tackle the problem of exploitation of the spatial and spectral features in the HSIs for the use of training classifier. In this letter, we design a simple but innovative end-to-end deep U-net-based model for HSI classification task. Unlike the previous CNN based models that mainly use CNN for spatial feature extraction and process the HSI data locally in small patches, our model takes the whole HSI as network input directly and outputs the predicted classes corresponding to each pixel location. Classification loss in the train data set and spatial constraint loss for the predicted result are combined as the loss function in the training stage to learn the mapping from HSI data to classification map and enhance the spatial continuity and consistency of the predicted result. Benchmark HSI data sets are used to evaluate the performance of the proposed method. Experimental results show that our model can achieve promising results comparing with the existing CNN-based methods.
What problem does this paper attempt to address?