Command Palette
Search for a command to run...
FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence
{Kihyuk Sohn∗ David Berthelot∗ Chun-Liang Li Zizhao Zhang Nicholas Carlini Ekin D. Cubuk Alex Kurakin Han Zhang Colin Raffel}

Abstract
Semi-supervised learning (SSL) provides an effectivemeans of leveraging unlabeled data to improve a model’sperformance. In this paper, we demonstrate the power of asimple combination of two common SSL methods: consistency regularization and pseudo-labeling. Our algorithm,FixMatch, first generates pseudo-labels using the model’spredictions on weakly-augmented unlabeled images. For agiven image, the pseudo-label is only retained if the modelproduces a high-confidence prediction. The model is thentrained to predict the pseudo-label when fed a stronglyaugmented version of the same image. Despite its simplicity, we show that FixMatch achieves state-of-the-art performance across a variety of standard semi-supervised learning benchmarks, including 94.93% accuracy on CIFAR-10with 250 labels and 88.61% accuracy with 40 – just 4 labels per class. Since FixMatch bears many similaritiesto existing SSL methods that achieve worse performance,we carry out an extensive ablation study to tease apartthe experimental factors that are most important to FixMatch’s success. We make our code available at https://github.com/google-research/fixmatch.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-classification-on-stl-10 | ReMixMatch | Percentage correct: 94.77 |
| image-classification-on-stl-10 | FixMatch (CTA) | Percentage correct: 94.83 |
| image-classification-on-stl-10 | FixMatch (RA) | Percentage correct: 92.02 |
| image-classification-on-stl-10 | UDA | Percentage correct: 92.34 |
| image-classification-on-stl-10 | Π-Model | Percentage correct: 73.77 |
| image-classification-on-stl-10 | Pseudo-Labeling | Percentage correct: 72.01 |
| image-classification-on-stl-10 | MixMatch | Percentage correct: 89.59 |
| image-classification-on-stl-10 | Mean Teacher | Percentage correct: 78.57 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.