ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
3.1k
Citations
374
Influential Citations
IEEE Transactions on Pattern Analysis and Machine Intelligence
Venue
2018
Year
We propose a new regularization method based on virtual adversarial loss: a new measure of local smoothness of the conditional label distribution given input. Virtual adversarial loss is defined as the robustness of the conditional label distribution around each input data point against local perturbation. Unlike adversarial training, our method defines the adversarial direction without label information and is hence applicable to semi-supervised learning. Because the directions in which we smooth the model are only "virtually" adversarial, we call our method virtual adversarial training (VAT). The computational cost of VAT is relatively low. For neural networks, the approximated gradient of virtual adversarial loss can be computed with no more than two pairs of forward- and back-propagations. In our experiments, we applied VAT to supervised and semi-supervised learning tasks on multiple benchmark datasets. With a simple enhancement of the algorithm based on the entropy minimization principle, our VAT achieves state-of-the-art performance for semi-supervised learning tasks on SVHN and CIFAR-10.
Virtual Adversarial Training (VAT) addresses a fundamental challenge in machine learning: how to make models robust to small input perturbations without requiring labeled data. Prior adversarial training methods needed true labels to define adversarial examples, limiting their use in semi-supervised settings. VAT removes this dependency by defining a "virtual" adversarial direction—the direction that maximally changes the model's output distribution—making it applicable to unlabeled data. This innovation bridges the gap between adversarial robustness and semi-supervised learning, two areas that were previously largely separate.
The paper's practical impact is amplified by its computational efficiency. The authors show that the virtual adversarial gradient can be approximated with just two forward-backward passes, making VAT feasible for large-scale neural networks. This efficiency, combined with strong empirical results, has made VAT a standard baseline in semi-supervised learning research. The method's simplicity and effectiveness have inspired numerous follow-up works, including consistency regularization approaches like Pi-Model and Mean Teacher.
On semi-supervised learning benchmarks, VAT with entropy minimization achieves state-of-the-art results: on SVHN with 250 labeled examples, it achieves 3.86% error rate, outperforming prior methods like Ladder Networks (4.23%) and CatGAN (5.67%). On CIFAR-10 with 4000 labeled examples, VAT achieves 14.82% error, surpassing the previous best of 16.61% from Ladder Networks. In supervised settings, VAT also improves test accuracy on MNIST, CIFAR-10, and CIFAR-100, demonstrating its broad applicability.
VAT has become a cornerstone of semi-supervised learning, influencing both theory and practice. Its core idea—enforcing local smoothness without labels—has been adopted in many subsequent methods, including consistency regularization and contrastive learning. The paper's clear exposition and efficient implementation have made it accessible to practitioners, leading to widespread adoption in domains like computer vision, natural language processing, and speech recognition. VAT also provides a bridge between adversarial robustness and semi-supervised learning, opening new directions for research on robust and data-efficient learning.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba