Defending against Backdoor Attack on Deep Neural Networks
Hao Cheng, Kaidi Xu, Sijia Liu, Pin-Yu Chen, Pu Zhao, Xue Lin
TL;DR
This work analyzes how backdoor data-poisoning alters internal DNN responses by leveraging Grad-CAM visualization and activation statistics. It shows that backdoor triggers cause pronounced activation in the trigger region and identifies the $\ell_\infty$-norm of neuron activations as the most discriminative signal between clean and triggered inputs. Based on this, it introduces $\ell_\infty$-based neuron pruning to remove trigger-sensitive neurons, achieving a substantial drop in attack success rate (e.g., from $81.6\%$ to $48.42\%$) with only minor clean-accuracy loss on the GTSRB dataset with AlexNet. The method relies on trigger-reverse-engineered patterns and requires no access to the training data, offering a practical defense against backdoor attacks. It also suggests directions for further defense strategies and more powerful attack analyses.
Abstract
Although deep neural networks (DNNs) have achieved a great success in various computer vision tasks, it is recently found that they are vulnerable to adversarial attacks. In this paper, we focus on the so-called \textit{backdoor attack}, which injects a backdoor trigger to a small portion of training data (also known as data poisoning) such that the trained DNN induces misclassification while facing examples with this trigger. To be specific, we carefully study the effect of both real and synthetic backdoor attacks on the internal response of vanilla and backdoored DNNs through the lens of Gard-CAM. Moreover, we show that the backdoor attack induces a significant bias in neuron activation in terms of the $\ell_\infty$ norm of an activation map compared to its $\ell_1$ and $\ell_2$ norm. Spurred by our results, we propose the \textit{$\ell_\infty$-based neuron pruning} to remove the backdoor from the backdoored DNN. Experiments show that our method could effectively decrease the attack success rate, and also hold a high classification accuracy for clean images.
