Training-Free Robust Neural Network Search Via Pruning

Qiancheng Yang,Yong Luo,Bo Du
DOI: https://doi.org/10.1109/icme57554.2024.10687950
2024-01-01
Abstract:Adversarial examples widely exist in visual tasks, which look almost the same as normal images but can induce neural networks to make completely wrong predictions. This exposes significant risks to deploying deep learning to visual tasks. The robustness of neural networks to adversarial images is shaped by multiple factors entangled together, including adversarial training strategies, model architectures, hyper-parameters, and optimization algorithms. This makes searching for robust architectures a significantly complex task. In this paper, we propose a Robust Training-free Pruning neural architecture search (RTP-NAS) framework that endeavors to disentangle the influence of architectures from other factors. We adopt Universal Adversarial Perturbations (UAPs) to construct a transferable adversarial data space across different network architectures. RTP-NAS further employs two adversarial training-free indicators, the condition number of the neural tangent kernel (NTK) and the number of linear regions calculated on the adversarial data space, to measure the intrinsic adversarial robustness of candidate architectures. Comprehensive experiments show that the proposed method achieves superior performance to other existing approaches, with a significantly lower computational cost. The code will be released publicly.
What problem does this paper attempt to address?