HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

If your data distribution shifts, use self-learning

Evgenia Rusak Steffen Schneider George Pachitariu Luisa Eck Peter Gehler Oliver Bringmann Wieland Brendel Matthias Bethge

If your data distribution shifts, use self-learning

Abstract

We demonstrate that self-learning techniques like entropy minimization and pseudo-labeling are simple and effective at improving performance of a deployed computer vision model under systematic domain shifts. We conduct a wide range of large-scale experiments and show consistent improvements irrespective of the model architecture, the pre-training technique or the type of distribution shift. At the same time, self-learning is simple to use in practice because it does not require knowledge or access to the original training data or scheme, is robust to hyperparameter choices, is straight-forward to implement and requires only a few adaptation epochs. This makes self-learning techniques highly attractive for any practitioner who applies machine learning algorithms in the real world. We present state-of-the-art adaptation results on CIFAR10-C (8.5% error), ImageNet-C (22.0% mCE), ImageNet-R (17.4% error) and ImageNet-A (14.8% error), theoretically study the dynamics of self-supervised adaptation methods and propose a new classification dataset (ImageNet-D) which is challenging even with adaptation.

Code Repositories

bethgelab/robustness
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
unsupervised-domain-adaptation-on-imagenet-aEfficientNet-L2 NoisyStudent + RPL
Top 1 Error: 14.8
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + DeepAug + Augmix + RPL
mean Corruption Error (mCE): 34.8
unsupervised-domain-adaptation-on-imagenet-cEfficientNet-L2+ENT
mean Corruption Error (mCE): 23.0
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + IG-3.5B + RPL
mean Corruption Error (mCE): 40.9
unsupervised-domain-adaptation-on-imagenet-cResNet50 + RPL
mean Corruption Error (mCE): 50.5
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + RPL
mean Corruption Error (mCE): 43.2
unsupervised-domain-adaptation-on-imagenet-cResNet50 + ENT
mean Corruption Error (mCE): 51.6
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + ENT
mean Corruption Error (mCE): 44.3
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + DeepAug + Augmix + ENT
mean Corruption Error (mCE): 35.5
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + IG-3.5B + ENT
mean Corruption Error (mCE): 40.8
unsupervised-domain-adaptation-on-imagenet-cEfficientNet-L2+RPL
mean Corruption Error (mCE): 22.0
unsupervised-domain-adaptation-on-imagenet-rResNet50 + RPL
Top 1 Error: 54.1
unsupervised-domain-adaptation-on-imagenet-rEfficientNet-L2 Noisy Student + ENT
Top 1 Error: 19.7
unsupervised-domain-adaptation-on-imagenet-rEfficientNet-L2 Noisy Student + RPL
Top 1 Error: 17.4
unsupervised-domain-adaptation-on-imagenet-rResNet50 + ENT
Top 1 Error: 56.1

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