Conference Paper
Machine Learning

Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learning

Takeru Miyato(Kyoto University), Shin‐ichi Maeda(Kyoto University), Masanori Koyama(Kyoto University), Shin Ishii(Kyoto University)
July 23, 2018IEEE Transactions on Pattern Analysis and Machine Intelligence3,089 citations

3.1k

Citations

374

Influential Citations

IEEE Transactions on Pattern Analysis and Machine Intelligence

Venue

2018

Year

Abstract

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.

Analysis

Why This Paper Matters

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.

Technical Contributions

  • Virtual Adversarial Loss: A new regularization term that measures the local smoothness of the conditional label distribution. Unlike standard adversarial training, it does not require label information.
  • Efficient Gradient Approximation: The virtual adversarial direction is computed via a power iteration method that converges in a single step, requiring only two forward-backward passes per input.
  • Entropy Minimization Enhancement: VAT is combined with an entropy minimization term on unlabeled data, further encouraging the model to produce confident predictions on low-density regions.
  • Theoretical Justification: The paper provides a connection between virtual adversarial loss and the local Lipschitz constant of the model, offering a principled regularization perspective.

Results

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.

Significance

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.