MarginMatch: Improving Semi-Supervised Learning with Pseudo-Margins
Not on the map. Why?
Abstract (arXiv)
We introduce MarginMatch, a new SSL approach combining consistency regularization and pseudo-labeling, with its main novelty arising from the use of unlabeled data training dynamics to measure pseudo-label quality. Instead of using only the model's confidence on an unlabeled example at an arbitrary iteration to decide if the example should be masked or not, MarginMatch also analyzes the behavior of the model on the pseudo-labeled examples as the training progresses, to ensure low quality predictions are masked out. MarginMatch brings substantial improvements on four vision benchmarks in low data regimes and on two large-scale datasets, emphasizing the importance of enforcing high-quality pseudo-labels. Notably, we obtain an improvement in error rate over the state-of-the-art of 3.25% on CIFAR-100 with only 25 labels per class and of 3.78% on STL-10 using as few as 4 labels per class. We make our code available at https://github.com/tsosea2/MarginMatch.
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Builds on 32 of its 51 references that are in the map. First posted on arXiv 17 Aug 2023.
Builds on
32 of 51 references.
References · 32
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2014Going Deeper with Convolutionscited by 4,241
- 2016Wide Residual Networkscited by 3,070
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,269
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,085
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2019MixMatch: A Holistic Approach to Semi-Supervised Learningcited by 1,391
20 more
- 2019Self-training with Noisy Student improves ImageNet classificationcited by 1,311
- 2019Unsupervised Data Augmentation for Consistency Trainingcited by 1,060
- 2016Temporal Ensembling for Semi-Supervised Learningcited by 1,057
- 2017Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learningcited by 920
- 2017Spectrally-normalized margin bounds for neural networkscited by 686
- 2018Exploring the Limits of Weakly Supervised Pretrainingcited by 605
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 582
- 2021FlexMatch: Boosting Semi-Supervised Learning with Curriculum Pseudo Labelingcited by 540
- 2019Pseudo-Labeling and Confirmation Bias in Deep Semi-Supervised Learningcited by 510
- 2016Regularization With Stochastic Transformations and Perturbations for Deep Semi-Supervised Learningcited by 441
- 2019ReMixMatch: Semi-Supervised Learning with Distribution Alignment and Augmentation Anchoringcited by 349
- 2016Adversarial Training Methods for Semi-Supervised Text Classificationcited by 280
- 2017WebVision Database: Visual Learning and Understanding from Web Datacited by 262
- 2020Identifying Mislabeled Data using the Area Under the Margin Rankingcited by 261
- 2019Interpolation Consistency Training for Semi-Supervised Learningcited by 251
- 2014Learning with Pseudo-Ensemblescited by 241
- 2018Predicting the Generalization Gap in Deep Networks with Margin Distributionscited by 113
- 2018Large Margin Deep Networks for Classificationcited by 112
- 2016Mutual Exclusivity Loss for Semi-Supervised Deep Learningcited by 22
- 2021OpenCoS: Contrastive Semi-supervised Learning for Handling Open-set Unlabeled Datacited by 22
Built on by
Citing papers per year: 2024: 4; 2025: 5; 2026: 1.
10 papers.
- 2025CGMatch: A Different Perspective of Semi-supervised Learningcited by 11
- 2024Foster Adaptivity and Balance in Learning with Noisy Labelscited by 9
- 2025Self-Reinforcing Prototype Evolution with Dual-Knowledge Cooperation for Semi-Supervised Lifelong Person Re-Identificationcited by 7
- 2024Learning with Imbalanced Noisy Data by Preventing Bias in Sample Selectioncited by 6
- 2024Reinforcement Learning-Guided Semi-Supervised Learningcited by 2
- 2024Learning from True-False Labels via Multi-modal Prompt Retrievingcited by 1
- 2025SpikeMatch: Semi-Supervised Learning with Temporal Dynamics of Spiking Neural Networkscited by 0
- 2025Human-Corrected Labels Learning: Enhancing Labels Quality via Human Correction of VLMs Discrepanciescited by 0
- 2025Semi-Supervised Learning for Large Language Models Safety and Content Moderationcited by 0
- 2026MTSSL: Meta-Thresholding Semi-Supervised Learningcited by 0
- 2026MTSSL: Meta-Thresholding Semi-Supervised Learning17 Jul 2026
- 2025Semi-Supervised Learning for Large Language Models Safety and Content Moderation24 Dec 2025
- 2025Human-Corrected Labels Learning: Enhancing Labels Quality via Human Correction of VLMs Discrepancies12 Nov 2025
- 2025SpikeMatch: Semi-Supervised Learning with Temporal Dynamics of Spiking Neural Networks26 Sep 2025
- 2025Self-Reinforcing Prototype Evolution with Dual-Knowledge Cooperation for Semi-Supervised Lifelong Person Re-Identification2 Jul 2025
- 2025CGMatch: A Different Perspective of Semi-supervised Learning4 Mar 2025
- 2024Foster Adaptivity and Balance in Learning with Noisy Labels3 Jul 2024
- 2024Learning from True-False Labels via Multi-modal Prompt Retrieving24 May 2024
- 2024Reinforcement Learning-Guided Semi-Supervised Learning2 May 2024
- 2024Learning with Imbalanced Noisy Data by Preventing Bias in Sample Selection17 Feb 2024