HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Do Adversarially Robust ImageNet Models Transfer Better?

Hadi Salman Andrew Ilyas Logan Engstrom Ashish Kapoor Aleksander Madry

Do Adversarially Robust ImageNet Models Transfer Better?

Abstract

Transfer learning is a widely-used paradigm in deep learning, where models pre-trained on standard datasets can be efficiently adapted to downstream tasks. Typically, better pre-trained models yield better transfer results, suggesting that initial accuracy is a key aspect of transfer learning performance. In this work, we identify another such aspect: we find that adversarially robust models, while less accurate, often perform better than their standard-trained counterparts when used for transfer learning. Specifically, we focus on adversarially robust ImageNet classifiers, and show that they yield improved accuracy on a standard suite of downstream classification tasks. Further analysis uncovers more differences between robust and standard models in the context of transfer learning. Our results are consistent with (and in fact, add to) recent hypotheses stating that robustness leads to improved feature representations. Our code and models are available at https://github.com/Microsoft/robust-models-transfer .

Code Repositories

MadryLab/robustness
Official
pytorch
Mentioned in GitHub
lengstrom/gitlinks
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
object-recognition-on-shape-biasResNet-50 (L2 eps 5.0 adv trained)
shape bias: 69.5

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp