Abstract:Background: Semi-supervised learning (SSL) can effectively use information from unlabeled data to improve model performance, which has great significance in medical imaging tasks. Pseudo-labeling is a classical SSL method that uses a model to predict unlabeled samples and selects the prediction with the highest confidence level as the pseudo-labels and then uses the generated pseudo-labels to train the model. Most of the current pseudo-label-based SSL algorithms use predefined fixed thresholds for all classes to select unlabeled data. Purpose: However, data imbalance is a common problem in medical image tasks, where the use of fixed threshold to generate pseudo-labels ignores different classes of learning status and learning difficulties. The aim of this study is to develop an algorithm to solve this problem. Methods: In this work, we propose Multi-Curriculum Pseudo-Labeling (MCPL), which evaluates the learning status of the model for each class at each epoch and automatically adjusts the thresholds for each class. We apply MCPL to FixMatch and propose a new SSL framework for medical image classification, which we call the improved algorithm FaxMatch. To mitigate the impact of incorrect pseudo-labels on the model, we use label smoothing (LS) strategy to generate soft labels (SL) for pseudo-labels. Results: We have conducted extensive experiments to evaluate our method on two public benchmark medical image classification datasets: the ISIC 2018 skin lesion analysis and COVID-CT datasets. Experimental results show that our method outperforms fully supervised baseline, which uses only labeled data to train the model. Moreover, our method also outperforms other state-of-the-art methods. Conclusions: We propose MCPL and construct a semi-supervised medical image classification framework to reduce the reliance of the model on a large number of labeled images and reduce the manual workload of labeling medical image data.

ComMatch: A semi-supervised learning classification algorithm based on model calibration

ActiveMatch: End-to-end Semi-supervised Active Representation Learning

HyperMatch: Noise-Tolerant Semi-Supervised Learning Via Relaxed Contrastive Constraint

FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence

CHMATCH: Contrastive Hierarchical Matching and Robust Adaptive Threshold Boosted Semi-Supervised Learning

FreeMatch: Self-adaptive Thresholding for Semi-supervised Learning

Boosting Semi-Supervised Learning by Exploiting All Unlabeled Data

MutexMatch: Semi-Supervised Learning With Mutex-Based Consistency Regularization

FlexMatch: Boosting Semi-Supervised Learning with Curriculum Pseudo Labeling

SimMatch: Semi-supervised Learning with Similarity Matching

ReliaMatch: Semi-Supervised Classification with Reliable Match

AllMatch: Exploiting All Unlabeled Data for Semi-Supervised Learning

RelationMatch: Matching In-batch Relationships for Semi-supervised Learning

Boosting Semi-Supervised Learning with Contrastive Complementary Labeling

Class-Aware Contrastive Semi-Supervised Learning

HIERMATCH: Leveraging Label Hierarchies for Improving Semi-Supervised Learning

FaxMatch: Multi-Curriculum Pseudo-Labeling for semi-supervised medical image classification

OTAMatch: Optimal Transport Assignment with PseudoNCE for Semi-supervised Learning

Interpolation-Based Contrastive Learning for Few-Label Semi-Supervised Learning

SCOMatch: Alleviating Overtrusting in Open-set Semi-supervised Learning

Ensemble Learning with Advanced Fast Image Filtering Features for Semi-Global Matching