Mathematics, Vol. 12, Pages 3572: Enhanced Semi-Supervised Medical Image Classification Based on Dynamic Sample Reweighting and Pseudo-Label Guided Contrastive Learning (DSRPGC)

Mathematics, Vol. 12, Pages 3572: Enhanced Semi-Supervised Medical Image Classification Based on Dynamic Sample Reweighting and Pseudo-Label Guided Contrastive Learning (DSRPGC)

Mathematics doi: 10.3390/math12223572

Authors: Kun Liu Ji Liu Sidong Liu

In semi-supervised learning (SSL) for medical image classification, model performance is often hindered by the scarcity of labeled data and the complexity of unlabeled data. This paper proposes an enhanced SSL approach to address these challenges by effectively utilizing unlabeled data through a combination of pseudo-labeling and contrastive learning. The key contribution of our method is the introduction of a Dynamic Sample Reweighting strategy to select reliable unlabeled samples, thereby improving the model’s utilization of unlabeled data. Additionally, we incorporate multiple data augmentation strategies based on the Mean Teacher (MT) model to ensure consistent outputs across different perturbations. To better capture and integrate multi-scale features, we propose a novel feature fusion network, the Medical Multi-scale Feature Fusion Network (MedFuseNet), which enhances the model’s ability to classify complex medical images. Finally, we introduce a pseudo-label guided contrastive learning (PGC) loss function that improves intra-class compactness and inter-class separability of the model’s feature representations. Extensive experiments on three public medical image datasets demonstrate that our method outperforms existing SSL approaches, achieving 93.16% accuracy on the ISIC2018 dataset using only 20% labeled data, highlighting the potential of our approach to advance medical image classification under limited supervision.