IEEE Access (Jan 2021)
Spectral-Spatial Hyperspectral Image Classification Using Robust Dual-Stage Spatial Embedding
Abstract
Recently, many spectral-special classification models have emerged one after another in the remote sensing community. These models aim to introduce the spatial information of the pixel to improve the accuracy of the class attribute of the pixel. However, for the spectral-spatial classification algorithms, not all pixels need to introduce the corresponding spatial information since the use of a large amount of spatial information has a costly time. To solve this problem, this paper proposes a robust dual-stage spatial embedding (RDSSE) model for spectral-spatial classification of hyperspectral image, which is composed of the following several main steps: First, an over-segmentation algorithm is employed to cluster original hyperspectral image into many superpixel blocks with shape adaptive characteristics. Then, we design a $k$ -peak criterion to fuse the spectral feature of pixels within and between superpixels. Next, a low time-consumption spectral classifier is introduced to conduct primary classification for a testing pixel to achieve the corresponding probability distribution. Specifically, the difference between the probability of the largest class and that of the second-largest class is served as class confidence. Finally, the predicted label of the low-confidence testing pixels is reclassified based on a high-accuracy spectral-spatial classification method. Experimental results on several real images illustrated that the proposed RDSSE method can obtain superior performance with respect to several competitive methods.
Keywords