Drawing Robust Scratch Tickets: Subnetworks with Inborn Robustness Are Found within Randomly Initialized Networks
Authors
Authors
- Yonggan Fu
- Qixuan Yu
- Yang Zhang
- Shang Wu
- Xu Ouyang
- David Cox
- Yingyan Lin
Authors
- Yonggan Fu
- Qixuan Yu
- Yang Zhang
- Shang Wu
- Xu Ouyang
- David Cox
- Yingyan Lin
Deep Neural Networks (DNNs) are known to be vulnerable to adversarial attacks, i.e., an imperceptible perturbation to the input can mislead DNNs trained on clean images into making erroneous predictions. To tackle this, adversarial training is currently the most effective defense method, by augmenting the training set with adversarial samples generated on the fly. Interestingly, we discover for the first time that there exist subnetworks with inborn robustness, matching or surpassing the robust accuracy of the adversarially trained networks with comparable model sizes, within randomly initialized networks without any model training, indicating that adversarial training on model weights is not indispensable towards adversarial robustness. We name such subnetworks Robust Scratch Tickets (RSTs), which are also by nature efficient. Distinct from the popular lottery ticket hypothesis, neither the original dense networks nor the identified RSTs need to be trained. To validate and understand this fascinating finding, we further conduct extensive experiments to study the existence and properties of RSTs under different models, datasets, sparsity patterns, and attacks, drawing insights regarding the relationship between DNNs’ robustness and their initialization/overparameterization. Furthermore, we identify the poor adversarial transferability between RSTs of different sparsity ratios drawn from the same randomly initialized dense network, and propose a Random RST Switch (R2S) technique, which randomly switches between different RSTs, as a novel defense method built on top of RSTs. We believe our findings about RSTs have opened up a new perspective to study model robustness and extend the lottery ticket hypothesis. Our codes are available at: https://github.com/RICE-EIC/Robust-Scratch-Ticket.
This paper has been published at NeurIPS 2021
Please cite our work using the BibTeX below.
@misc{fu2021drawing,
title={Drawing Robust Scratch Tickets: Subnetworks with Inborn Robustness Are Found within Randomly Initialized Networks},
author={Yonggan Fu and Qixuan Yu and Yang Zhang and Shang Wu and Xu Ouyang and David Cox and Yingyan Lin},
year={2021},
eprint={2110.14068},
archivePrefix={arXiv},
primaryClass={cs.LG}
}