mlsanityData to 7 Oct 2026

Imbalanced Semi-Supervised Learning via Label Refinement and Threshold Adjustment

arXiv 2407.05370 ↗First posted 7 Jul 2024cs.LGcited by 0

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2017On Calibration of Modern Neural Networkscited by 4,860
  3. 2016Wide Residual Networkscited by 3,057
  4. 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
  5. 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,080
  6. 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
  7. 2019MixMatch: A Holistic Approach to Semi-Supervised Learningcited by 1,390
  8. 2019Unsupervised Data Augmentation for Consistency Trainingcited by 1,057
  9. 2016Temporal Ensembling for Semi-Supervised Learningcited by 1,054
  10. 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 987
  11. 2017Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learningcited by 918
  12. 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 806
36 more
  1. 2019Large-Scale Long-Tailed Recognition in an Open Worldcited by 704
  2. 2020Long-tail learning via logit adjustmentcited by 605
  3. 2021FlexMatch: Boosting Semi-Supervised Learning with Curriculum Pseudo Labelingcited by 537
  4. 2018Detecting and Correcting for Label Shift with Black Box Predictorscited by 364
  5. 2019ReMixMatch: Semi-Supervised Learning with Distribution Alignment and Augmentation Anchoringcited by 349
  6. 2021In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learningcited by 301
  7. 2019Label Propagation for Deep Semi-supervised Learningcited by 287
  8. 2022FreeMatch: Self-adaptive Thresholding for Semi-supervised Learningcited by 232
  9. 2021CReST: A Class-Rebalancing Self-Training Framework for Imbalanced Semi-Supervised Learningcited by 174
  10. 2019Population Based Augmentation: Efficient Learning of Augmentation Policy Schedulescited by 153
  11. 2021Dash: Semi-Supervised Learning with Dynamic Thresholdingcited by 147
  12. 2022Leveraging Unlabeled Data to Predict Out-of-Distribution Performancecited by 141
  13. 2023SoftMatch: Addressing the Quantity-Quality Trade-off in Semi-supervised Learningcited by 133
  14. 2020Distribution Aligning Refinery of Pseudo-label for Imbalanced Semi-supervised Learningcited by 105
  15. 2022USB: A Unified Semi-supervised Learning Benchmark for Classificationcited by 95
  16. 2022Debiased Learning from Naturally Imbalanced Pseudo-Labelscited by 85
  17. 2021ABC: Auxiliary Balanced Classifier for Class-imbalanced Semi-supervised Learningcited by 65
  18. 2017Triple Generative Adversarial Netscited by 59
  19. 2020Posterior Re-calibration for Imbalanced Datasetscited by 53
  20. 2021DASO: Distribution-Aware Semantics-Oriented Pseudo-label for Imbalanced Semi-Supervised Learningcited by 51
  21. 2024Probabilistic Contrastive Learning for Long-Tailed Visual Recognitioncited by 42
  22. 2021CoSSL: Co-Learning of Representation and Classifier for Imbalanced Semi-Supervised Learningcited by 41
  23. 2021A Realistic Evaluation of Semi-Supervised Learning for Fine-Grained Classificationcited by 36
  24. 2021Analyzing Overfitting under Class Imbalance in Neural Networks for Image Segmentationcited by 34
  25. 2021The Semi-Supervised iNaturalist-Aves Challenge at FGVC7 Workshopcited by 21
  26. 2024Robust Semi-supervised Learning by Wisely Leveraging Open-set Datacited by 20
  27. 2022Imbalanced Semi-supervised Learning with Bias Adaptive Classifiercited by 18
  28. 2021Rethinking Re-Sampling in Imbalanced Semi-Supervised Learningcited by 16
  29. 2018Semi-Supervised Learning via Compact Latent Space Clusteringcited by 15
  30. 2024SimPro: A Simple Probabilistic Framework Towards Realistic Long-Tailed Semi-Supervised Learningcited by 14
  31. 2023InstanT: Semi-supervised Learning with Instance-dependent Thresholdscited by 13
  32. 2022An Embarrassingly Simple Baseline for Imbalanced Semi-Supervised Learningcited by 12
  33. 2023InPL: Pseudo-labeling the Inliers First for Imbalanced Semi-supervised Learningcited by 12
  34. 2022Estimating Model Performance under Domain Shifts with Class-Specific Confidence Scorescited by 10
  35. 2022On the Importance of Calibration in Semi-supervised Learningcited by 6
  36. 2023Diffusion Model based Semi-supervised Learning on Brain Hemorrhage Images for Efficient Midline Shift Quantificationcited by 3

Built on by

Not cited yet.