A Direct Approach to Robust Deep Learning Using Adversarial Networks

Huaxia Wang, Chun-Nam Yu

Abstract: Deep neural networks have been shown to perform well in many classical machine learning problems, especially in image classification tasks. However, researchers have found that neural networks can be easily fooled, and they are surprisingly sensitive to small perturbations imperceptible to humans. Carefully crafted input images (adversarial examples) can force a well-trained neural network to provide arbitrary outputs. Including adversarial examples during training is a popular defense mechanism against adversarial attacks. In this paper we propose a new defensive mechanism under the generative adversarial network~(GAN) framework. We model the adversarial noise using a generative network, trained jointly with a classification discriminative network as a minimax game. We show empirically that our adversarial network approach works well against black box attacks, with performance on par with state-of-art methods such as ensemble adversarial training and adversarial training with projected gradient descent.

Keywords: deep learning, adversarial learning, generative adversarial networks

TL;DR: Jointly train an adversarial noise generating network with a classification network to provide better robustness to adversarial attacks.

Code: [ whxbergkamp/RobustDL_GAN](https://github.com/whxbergkamp/RobustDL_GAN)

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/a-direct-approach-to-robust-deep-learning/code)