Interpolated Adversarial Training: Achieving robust neural networks without sacrificing too much accuracy.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Adversarial robustness has become a central goal in deep learning, both in the theory and the practice. However, successful methods to improve the adversarial robustness (such as adversarial training) greatly hurt generalization performance on the unperturbed data. This could have a major impact on how the adversarial robustness affects real world systems (i.e. many may opt to forego robustness if it can improve accuracy on the unperturbed data). We propose Interpolated Adversarial Training, which employs recently proposed interpolation based training methods in the framework of adversarial training. On CIFAR-10, adversarial training increases the standard test error ( when there is no adversary) from 4.43% to 12.32%, whereas with our Interpolated adversarial training we retain the adversarial robustness while achieving a standard test error of only 6.45%. With our technique, the relative increase in the standard error for the robust model is reduced from 178.1% to just 45.5%. Moreover, we provide mathematical analysis of Interpolated Adversarial Training to confirm its efficiencies and demonstrate its advantages in terms of robustness and generalization.

Authors

  • Alex Lamb
    Montreal Institute for Learning Algorithms (MILA), Canada. Electronic address: lambalex@iro.umontreal.ca.
  • Vikas Verma
    Montreal Institute for Learning Algorithms (MILA), Canada; Aalto University, Finland. Electronic address: vikasverma.iitm@gmail.com.
  • Kenji Kawaguchi
    MIT, Cambridge, MA 02139, U.S.A. kawaguch@mit.edu.
  • Alexander Matyasko
    National University of Singapore, Singapore. Electronic address: alex.m@nus.edu.sg.
  • Savya Khosla
    Google, India.
  • Juho Kannala
    Aalto University, Finland. Electronic address: juho.kaanala@aalto.fi.
  • Yoshua Bengio
    Université de Montréal, Montréal QC H3T 1N8, Canada.