Domain-Collaborative Contrastive Learning for Hyperspectral Image Classification

Haiyang Luo,Xueyi Qiao,Yongming Xu,Shengwei Zhong,Chen Gong
DOI: https://doi.org/10.1109/lgrs.2024.3425482
IF: 5.343
2024-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:Variations in atmosphere, lighting, and imaging systems result in diverse category distributions in hyperspectral imagery, impacting the accuracy of cross-domain hyperspectral image classification (HSIC). Unsupervised domain adaptation (UDA) aims to address this issue by learning a model that generalizes effectively across domains, leveraging labels only from source domain (SD). Most existing UDA methods focus on aligning distributions between domains without fully considering the valuable information within individual domains. To fill this gap, this letter proposes a domain-collaborative contrastive learning (DCCL) method. DCCL integrates a novel pseudo-labeling strategy with a cross-domain contrastive learning (CL) framework. Specifically, in the pseudo-labeling phase, the confident examples in target domain (TD) are collaboratively labeled according to the labeled examples in SD and the class centers in TD. Then, the CL phase simultaneously minimizes in-domain and cross-domain contrastive loss to promote the aggregation of examples from the same category in both domains. Experimental results demonstrate that the DCCL achieves the accuracy rates of 93.47% and 54.59% on Pavia and Indiana datasets, respectively, surpassing the performance of other state-of-the-art UDA methods. Our source code is available at https://github.com/Leap-luohaiyang/DCCL-2024.
What problem does this paper attempt to address?