English

ImageNet Pre-training also Transfers Non-Robustness

Computer Vision and Pattern Recognition 2022-12-07 v4 Artificial Intelligence

Abstract

ImageNet pre-training has enabled state-of-the-art results on many tasks. In spite of its recognized contribution to generalization, we observed in this study that ImageNet pre-training also transfers adversarial non-robustness from pre-trained model into fine-tuned model in the downstream classification tasks. We first conducted experiments on various datasets and network backbones to uncover the adversarial non-robustness in fine-tuned model. Further analysis was conducted on examining the learned knowledge of fine-tuned model and standard model, and revealed that the reason leading to the non-robustness is the non-robust features transferred from ImageNet pre-trained model. Finally, we analyzed the preference for feature learning of the pre-trained model, explored the factors influencing robustness, and introduced a simple robust ImageNet pre-training solution. Our code is available at \url{https://github.com/jiamingzhang94/ImageNet-Pretraining-transfers-non-robustness}.

Keywords

Cite

@article{arxiv.2106.10989,
  title  = {ImageNet Pre-training also Transfers Non-Robustness},
  author = {Jiaming Zhang and Jitao Sang and Qi Yi and Yunfan Yang and Huiwen Dong and Jian Yu},
  journal= {arXiv preprint arXiv:2106.10989},
  year   = {2022}
}

Comments

Accepted by AAAI2023

R2 v1 2026-06-24T03:25:07.785Z