Mask-Enhanced Contrastive Learning for Hyperspectral Image Classification

Xianghai Cao,Jiayu Yu,Ruijie Xu,Jiaxuan Wei,Licheng Jiao
DOI: https://doi.org/10.1109/tgrs.2024.3479220
IF: 8.2
2024-11-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Recently, self-supervised learning (SSL) has gained great prominence in hyperspectral image classification (HSIC) due to its powerful capability to alleviate data-hunger problem. The generative-based method and the contrastive-based method have become two main streams in the field of SSL. To fully integrate the merits of both of them, we propose an efficient hybrid SSL method, that is, mask-enhanced contrastive learning (MECL). Essentially, MECL remains a prototypical contrastive-learning (CL) method, but incorporates the masking-and-predicting idea of the generative-based method. When the sample gradually matches the prototypes, feature reconstruction is implicitly performed in MECL step by step. Furthermore, we design a spatial-spectral multimasking mechanism for hyperspectral data and also propose two complementary strategies to prevent MECL from collapsing. To demonstrate the performance of MECL, a wide range of experiments are carried out in our work. On the one hand, the results of ablation experiments prove that HSIC does benefit from the fusion of SSL methods. On the other hand, the classification results on three hyperspectral datasets confirm that our MECL is an effective SSL method with high stability and strong reliability.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?