An Efficient and Lightweight Spectral-Spatial Feature Graph Contrastive Learning Framework for Hyperspectral Image Clustering

Published: 01 Jan 2024, Last Modified: 11 Apr 2025IEEE Trans. Geosci. Remote. Sens. 2024EveryoneRevisionsBibTeXCC BY-SA 4.0
Abstract: Due to the scarcity of prior information and the high complexity of spectral data, hyperspectral image (HSI) clustering presents a significant challenge. Although recent deep clustering methods have demonstrated remarkable performance, their intricate network structures and poor robustness hinder their practical application. To address this issue, we propose an efficient and lightweight spectral-spatial feature graph contrastive learning (S2GCL) framework for robust HSI clustering. Specifically, we have designed a novel spectral-spatial feature encoder that fully leverages the information in HSI by incorporating both spatial structure and spectral similarity matrices. To establish a lightweight model, we implement several effective designs: First, S2GCL eliminates the commonly used data augmentation and discriminator in GCL during the generation of positive embeddings. Second, we use a multilayer perceptron (MLP) to produce low-dimensional embeddings instead of relying on graph convolutional networks (GCNs). Third, negative embeddings are generated through row-shuffling, avoiding the use of neural networks. Finally, we propose a multiple boundary loss function to extract complementary information from spatial structures and neighboring nodes, while also constraining the interclass differences between positive and negative examples. We conducted extensive experiments on four publicly available datasets and compared S2GCL with state-of-the-art clustering methods. The results indicate that S2GCL achieves satisfactory performance. The code for S2GCL will be released at https://github.com/ahappyyang/S2GCL.
Loading