Imbalanced Semi-Supervised Learning via Label Refinement and Threshold Adjustment
Not on the map. Why?
Abstract (arXiv)
Semi-supervised learning (SSL) algorithms often struggle to perform well when trained on imbalanced data. In such scenarios, the generated pseudo-labels tend to exhibit a bias toward the majority class, and models relying on these pseudo-labels can further amplify this bias. Existing imbalanced SSL algorithms explore pseudo-labeling strategies based on either pseudo-label refinement (PLR) or threshold adjustment (THA), aiming to mitigate the bias through heuristic-driven designs. However, through a careful statistical analysis, we find that existing strategies are suboptimal: most PLR algorithms are either overly empirical or rely on the unrealistic assumption that models remain well-calibrated throughout training, while most THA algorithms depend on flawed metrics for pseudo-label selection. To address these shortcomings, we first derive the theoretically optimal form of pseudo-labels under class imbalance. This foundation leads to our key contribution: SEmi-supervised learning with pseudo-label optimization based on VALidation data (SEVAL), a unified framework that learns both PLR and THA parameters from a class-balanced subset of training data. By jointly optimizing these components, SEVAL adapts to specific task requirements while ensuring per-class pseudo-label reliability. Our experiments demonstrate that SEVAL outperforms state-of-the-art SSL methods, producing more accurate and effective pseudo-labels across various imbalanced SSL scenarios while remaining compatible with diverse SSL algorithms. The code is publicly available (https://github.com/ZerojumpLine/SEVAL).
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 48 of its 83 references that are in the map. First posted on arXiv 7 Jul 2024.
Builds on
48 of 83 references.
References · 48
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2016Wide Residual Networkscited by 3,057
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,080
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
- 2019MixMatch: A Holistic Approach to Semi-Supervised Learningcited by 1,390
- 2019Unsupervised Data Augmentation for Consistency Trainingcited by 1,057
- 2016Temporal Ensembling for Semi-Supervised Learningcited by 1,054
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 987
- 2017Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learningcited by 918
- 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 806
36 more
- 2019Large-Scale Long-Tailed Recognition in an Open Worldcited by 704
- 2020Long-tail learning via logit adjustmentcited by 605
- 2021FlexMatch: Boosting Semi-Supervised Learning with Curriculum Pseudo Labelingcited by 537
- 2018Detecting and Correcting for Label Shift with Black Box Predictorscited by 364
- 2019ReMixMatch: Semi-Supervised Learning with Distribution Alignment and Augmentation Anchoringcited by 349
- 2021In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learningcited by 301
- 2019Label Propagation for Deep Semi-supervised Learningcited by 287
- 2022FreeMatch: Self-adaptive Thresholding for Semi-supervised Learningcited by 232
- 2021CReST: A Class-Rebalancing Self-Training Framework for Imbalanced Semi-Supervised Learningcited by 174
- 2019Population Based Augmentation: Efficient Learning of Augmentation Policy Schedulescited by 153
- 2021Dash: Semi-Supervised Learning with Dynamic Thresholdingcited by 147
- 2022Leveraging Unlabeled Data to Predict Out-of-Distribution Performancecited by 141
- 2023SoftMatch: Addressing the Quantity-Quality Trade-off in Semi-supervised Learningcited by 133
- 2020Distribution Aligning Refinery of Pseudo-label for Imbalanced Semi-supervised Learningcited by 105
- 2022USB: A Unified Semi-supervised Learning Benchmark for Classificationcited by 95
- 2022Debiased Learning from Naturally Imbalanced Pseudo-Labelscited by 85
- 2021ABC: Auxiliary Balanced Classifier for Class-imbalanced Semi-supervised Learningcited by 65
- 2017Triple Generative Adversarial Netscited by 59
- 2020Posterior Re-calibration for Imbalanced Datasetscited by 53
- 2021DASO: Distribution-Aware Semantics-Oriented Pseudo-label for Imbalanced Semi-Supervised Learningcited by 51
- 2024Probabilistic Contrastive Learning for Long-Tailed Visual Recognitioncited by 42
- 2021CoSSL: Co-Learning of Representation and Classifier for Imbalanced Semi-Supervised Learningcited by 41
- 2021A Realistic Evaluation of Semi-Supervised Learning for Fine-Grained Classificationcited by 36
- 2021Analyzing Overfitting under Class Imbalance in Neural Networks for Image Segmentationcited by 34
- 2021The Semi-Supervised iNaturalist-Aves Challenge at FGVC7 Workshopcited by 21
- 2024Robust Semi-supervised Learning by Wisely Leveraging Open-set Datacited by 20
- 2022Imbalanced Semi-supervised Learning with Bias Adaptive Classifiercited by 18
- 2021Rethinking Re-Sampling in Imbalanced Semi-Supervised Learningcited by 16
- 2018Semi-Supervised Learning via Compact Latent Space Clusteringcited by 15
- 2024SimPro: A Simple Probabilistic Framework Towards Realistic Long-Tailed Semi-Supervised Learningcited by 14
- 2023InstanT: Semi-supervised Learning with Instance-dependent Thresholdscited by 13
- 2022An Embarrassingly Simple Baseline for Imbalanced Semi-Supervised Learningcited by 12
- 2023InPL: Pseudo-labeling the Inliers First for Imbalanced Semi-supervised Learningcited by 12
- 2022Estimating Model Performance under Domain Shifts with Class-Specific Confidence Scorescited by 10
- 2022On the Importance of Calibration in Semi-supervised Learningcited by 6
- 2023Diffusion Model based Semi-supervised Learning on Brain Hemorrhage Images for Efficient Midline Shift Quantificationcited by 3
Built on by
Not cited yet.