Temporal Ensembling for Semi-Supervised Learning
Samuli LaineTimo Aila
Introduces temporal ensembling, an efficient semi-supervised learning technique that forms consensus targets from previous training epochs under varying augmentations, substantially reducing classification error rates when labeled data is scarce.
The paper introduces two closely related training techniques, called the Π-model and temporal ensembling, that allow deep neural networks to learn from a mixture of a small number of labeled images and a much larger number of unlabeled images. The central idea is to form an ensemble-style target prediction for each unlabeled input by averaging the network’s own outputs across different training epochs, different dropout masks, and different random augmentations; this averaged target then serves as a stable training signal that encourages consistent predictions. Both methods were evaluated on standard image-classification benchmarks using only a few hundred or a few thousand labeled examples per dataset, with the remaining examples treated as unlabeled.
On the SVHN dataset with 500 labels the non-augmented error rate fell from the previous best of 18.44 % to 7.05 %; with standard translations the figure reached 5.12 %. On CIFAR-10 with 4 000 labels the corresponding numbers dropped from 18.63 % to 16.55 % without augmentation and to 12.16 % with horizontal flips and translations. Adding hundreds of thousands of extra unlabeled images drawn from the Tiny Images collection produced a further 2.7-percentage-point gain on CIFAR-100. The same consistency requirement also improved accuracy when every training example was labeled and conferred substantial robustness when up to 80 % of the supplied labels were deliberately corrupted.
These gains matter because labeling large image collections is expensive, while unlabeled images are abundant. The methods require only modest extra storage and computation, work with existing network architectures, and need no generative model or adversarial training. The main practical limitations are dependence on strong regularization (dropout plus input augmentation) and the need to tune a small number of additional hyperparameters that control the weighting and momentum of the ensemble targets. The authors note that further improvements could be obtained by tracking higher-order statistics of the network outputs to estimate prediction uncertainty, but they leave that extension for future work.
- Paper: Improving neural networks by preventing co-adaptation of feature detectors, Geoffrey E. Hinton et al. (2012). Understanding dropout regularization is a prerequisite because temporal ensembling relies heavily on stochastic regularizations like dropout during training to create consensus targets.
- Paper: Semi-supervised Learning with Deep Generative Models, Diederik P. Kingma et al. (2014). This earlier generative semi-supervised approach provides essential context on how deep neural networks utilize unlabeled data through variational objectives before exploring self-ensembling.
- Paper: Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results, Antti Tarvainen et al. (2017). The Mean Teacher algorithm directly extends temporal ensembling by replacing the ensemble of past epoch predictions with an exponential moving average of model weights.
- Paper: MixMatch: A Holistic Approach to Semi-Supervised Learning, David Berthelot et al. (2019). MixMatch builds upon the consistency-regularization principles introduced in temporal ensembling by combining them with data mixing and entropy minimization for semi-supervised learning.
