Enhancing Hyperspectral Image Classification: Leveraging Unsupervised Information With Guided Group Contrastive Learning

Ben Li,Leyuan Fang,Ning Chen,Jitong Kang,Jun Yue
DOI: https://doi.org/10.1109/tgrs.2024.3350700
IF: 8.2
2024-02-02
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Deep learning (DL) has demonstrated remarkable performance in the classification of hyperspectral images (HSIs) by leveraging its powerful ability to automatically learn deep spectral–spatial features over the years. Nevertheless, the limited supervisory signals along with a vast number of parameters in deep models still pose critical challenges when utilizing a restricted number of samples for training deep networks. To better handle this issue, this article proposes an end-to-end framework called guided group contrastive learning (GGCL) that adaptively integrates unsupervised information into a supervised contrastive learning framework. The proposed method employs a similarity-guided module that measures the spectral–spatial similarity of unsupervised samples based on supervised signals and effectively groups them. Then, the similarity signals of both supervised and unsupervised data are combined with contrastive learning to achieve intragroup feature aggregation and intergroup feature separation with guided group contrastive loss (GGCLoss). The pivotal characteristic of the proposed method lies in the end-to-end incorporation of unsupervised information with supervised signals for contrastive learning. Experiments on three public HSI datasets demonstrate that the proposed method can achieve better performance than existing state-of-the-art (SOTA) methods. For ease of reproducibility, the code of the proposed GGCL will be publicly available at https://github.com/fanerlight/GGCL_HSI.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?