Batch Normalization Assisted Adversarial Pruning: Towards Lightweight, Sparse and Robust Models.

Xiao Wei,Yao Zhu,Shu-Tao Xia
DOI: https://doi.org/10.1109/ijcnn52387.2021.9534077
2021-01-01
Abstract:Although deep neural networks (DNNs) have achieved great success in many areas, they are extremely vulnerable to imperceptible adversarial attacks. DNNs can improve robustness through adversarial training at the expense of massive computing and large network capacity, which makes the model difficult to be lightweight and deployed on edge or resource-constrained devices. To solve this problem, we take both the advantages of structured and unstructured pruning to get a lightweight and sparse model. With the fact that adversarial attacks mainly exist in the high-frequency component of images, we analyze the Batch Normalization (BN) layer to find that the magnitude of its scaling factor can identify those high-frequency components. Therefore, we propose to consider the magnitude of the scaling factor for the pruning criterion. Furthermore, unlike existing methods that prune networks only base on weights, we define “effective weight” which considers both the scaling factor and weight magnitude in unstructured pruning. A novel pruning method is proposed by considering the effective weight. We also verify the Lottery Ticket Hypothesis in adversarial pruning methods yet unknown to the adversarial community.
What problem does this paper attempt to address?