Enhancing hyperspectral image prediction with contrastive learning in low-label regimes
摘要
Labelled data scarcity remains a longstanding challenge in hyperspectral image analysis, primarily due to high spectral dimensionality and the laborious nature of manual annotation. Self-supervised contrastive learning (SSCL) recently emerged as a promising approach to address this challenge due to its ability to learn robust representations by distinguishing similar and dissimilar data samples guided by the inherent properties of data rather than by labels. Our study builds upon a previously established two-stage patch-level, multi-label classification method for hyperspectral imagery using contrastive learning and further examines its performance on single-label and multi-label classification tasks under scenarios of limited training data. The methodology unfolds in two stages. Initially, we train an encoder with a projection network in a contrastive learning approach. Next, we fine-tune the pre-trained encoder with a classifier. Our empirical results on four public hyperspectral datasets demonstrate consistent improvements over fully supervised methods, boosting overall accuracy by up to