Towards Deep Learning Models Resistant to Adversarial Attacks: Building Neural Networks Robust to Adversarial Examples from the Perspective of Robust Optimization
This paper, 'Towards Deep Learning Models Resistant to Adv…', published in 2018 by Aleksander Madry and others at the Massachusetts Institute of Technology (MIT), explores how robust optimization can enhance neural network security.
This paper, 'Towards Deep Learning Models Resistant to Adversarial Attacks,' published in 2018 by Aleksander Madry and colleagues at the Massachusetts Institute of Technology (MIT), delves into enhancing neural network security through robust optimization. The study highlights that deep neural networks have inherent weaknesses, making them highly susceptible to adversarial attacks—inputs that are nearly indistinguishable from original data but cause the model to misclassify. This vulnerability is particularly dangerous in safety-critical systems such as autonomous driving and facial recognition.
To address this issue, the authors propose a unified framework based on saddle-point (min-max) problems. This framework formalizes adversarial training as an optimization problem: the inner layer maximizes to find the adversarial perturbation that maximizes the loss function, while the outer layer minimizes by adjusting model parameters to reduce risk under this worst-case scenario. This approach not only encompasses previous attack and defense mechanisms but also provides a clear objective for safety guarantees, specifically resisting all attackers bounded by certain norms (such as the ℓ∞ norm).
In terms of implementation, the paper proves that Projected Gradient Descent (PGD) serves as a reliable 'first-order adversary' capable of effectively solving the aforementioned optimization problem. Experimental results show that models trained with PGD achieve over 89% accuracy against the strongest white-box attacks on the MNIST dataset and 46% on CIFAR-10. This indicates that by increasing model capacity and adopting correct optimization strategies, the adversarial robustness of deep learning models can be significantly improved, bringing them closer to the security standards required for practical deployment.