Abstract:Recent development in the field of Deep Learning have exposed the underlying vulnerability of Deep Neural Network (DNN) against adversarial examples. In image classification, an adversarial example is a carefully modified image that is visually imperceptible to the original image but can cause DNN model to misclassify it. Training the network with Gaussian noise is an effective technique to perform model regularization, thus improving model robustness against input variation. Inspired by this classical method, we explore to utilize the regularization characteristic of noise injection to improve DNN's robustness against adversarial attack. In this work, we propose Parametric-Noise-Injection (PNI) which involves trainable Gaussian noise injection at each layer on either activation or weights through solving the min-max optimization problem, embedded with adversarial training. These parameters are trained explicitly to achieve improved robustness. To the best of our knowledge, this is the first work that uses trainable noise injection to improve network robustness against adversarial attacks, rather than manually configuring the injected noise level through cross-validation. The extensive results show that our proposed PNI technique effectively improves the robustness against a variety of powerful white-box and black-box attacks such as PGD, C & W, FGSM, transferable attack and ZOO attack. Last but not the least, PNI method improves both clean- and perturbed-data accuracy in comparison to the state-of-the-art defense methods, which outperforms current unbroken PGD defense by 1.1 % and 6.8 % on clean test data and perturbed test data respectively using Resnet-20 architecture.

Multi-Class Triplet Loss with Gaussian Noise for Adversarial Robustness

Improving Adversarial Robustness of 3D Point Cloud Classification Models

Improving the Robustness of Deep Neural Networks via Adversarial Training with Triplet Loss

Parametric Noise Injection: Trainable Randomness to Improve Deep Neural Network Robustness against Adversarial Attack

Improving Model Robustness Against Adversarial Examples with Redundant Fully Connected Layer.

Enhancing adversarial robustness for deep metric learning via neural discrete adversarial training

Feature Augmentation for Adversarial Robustness

Colored Noise Injection for Training Adversarially Robust Neural Networks

Analytical Moment Regularizer for Gaussian Robust Networks

DeepDefense: Training Deep Neural Networks with Improved Robustness.

Training Robust Deep Neural Networks via Adversarial Noise Propagation

Not So Robust After All: Evaluating the Robustness of Deep Neural Networks to Unseen Adversarial Attacks

Deep Defense: Training DNNs with Improved Adversarial Robustness

A Direct Approach to Robust Deep Learning Using Adversarial Networks

Enhance DNN Adversarial Robustness and Efficiency via Injecting Noise to Non-Essential Neurons

Global Adversarial Attacks for Assessing Deep Learning Robustness

Analyzing the Noise Robustness of Deep Neural Networks

Feature Denoising for Improving Adversarial Robustness

Understanding the Robustness of Randomized Feature Defense Against Query-Based Adversarial Attacks

Improving Machine Learning Robustness via Adversarial Training

Improving Adversarial Robustness Requires Revisiting Misclassified Examples.