Deep Spatial-Spectral Subspace Clustering for Hyperspectral Images Based on Contrastive Learning

Xiang Hu,Teng Li,Tong Zhou,Yuanxi Peng
DOI: https://doi.org/10.3390/rs13214418
IF: 5
2021-11-03
Remote Sensing
Abstract:Hyperspectral image (HSI) clustering is a major challenge due to the redundant spectral information in HSIs. In this paper, we propose a novel deep subspace clustering method that extracts spatial–spectral features via contrastive learning. First, we construct positive and negative sample pairs through data augmentation. Then, the data pairs are projected into feature space using a CNN model. Contrastive learning is conducted by minimizing the distances of positive pairs and maximizing those of negative pairs. Finally, based on their features, spectral clustering is employed to obtain the final result. Experimental results gained over three HSI datasets demonstrate that our proposed method is superior to other state-of-the-art methods.
environmental sciences,imaging science & photographic technology,remote sensing,geosciences, multidisciplinary
What problem does this paper attempt to address?