Contrastive Mutual Learning with Pseudo-Label Smoothing for Hyperspectral Image Classification

Lizhu Liu,Hui Zhang,Yaonan Wang
DOI: https://doi.org/10.1109/tim.2024.3406811
IF: 5.6
2024-01-01
IEEE Transactions on Instrumentation and Measurement
Abstract:Semi-supervised learning has become an effective paradigm for reducing the reliance on hyperspectral image (HSI) classification on labeled data. State-of-the-art semi-supervised HSI classification methods learn supplementary knowledge from pseudo-labels, which are predicted by a deep learning model on unlabeled data. Nevertheless, these methods usually overlook the impacts of pseudo-label noise, intraclass spectral variability, and interclass spectral similarity, which may fundamentally constrain the model's capability for refining feature representation. To address these prevalent issues, we propose a novel semi-supervised framework-contrastive mutual learning with pseudo-label smoothing (CMLP) to enable the model to learn more refined features. First, we uniquely combine a mutual learning model and a pseudo-label smoothing strategy (PSS) to reduce the noise knowledge learned by the classification model during HSI feature extraction. Second, we incorporate a mutual pseudo-label guided contrastive (MPGC) learning approach, which helps to maximize interclass dispersion and minimize intraclass compactness, thus mitigating the problem of intraclass spectral variability and interclass spectral similarity within HSI data. In addition, we have introduced a dynamic threshold strategy (DTS) that adjusts the quantity of unlabeled samples introduced during the training process dynamically. This strategy mitigates the adverse impact of unstable predictions of unlabeled data in the early stages of training. The extensive experiments on three benchmark HSI datasets demonstrate that the proposed method can achieve competitive performance compared to state-of-the-art methods.
What problem does this paper attempt to address?