HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Vision Models Are More Robust And Fair When Pretrained On Uncurated Images Without Supervision

Priya Goyal Quentin Duval Isaac Seessel Mathilde Caron Ishan Misra Levent Sagun Armand Joulin Piotr Bojanowski

Vision Models Are More Robust And Fair When Pretrained On Uncurated Images Without Supervision

Abstract

Discriminative self-supervised learning allows training models on any random group of internet images, and possibly recover salient information that helps differentiate between the images. Applied to ImageNet, this leads to object centric features that perform on par with supervised features on most object-centric downstream tasks. In this work, we question if using this ability, we can learn any salient and more representative information present in diverse unbounded set of images from across the globe. To do so, we train models on billions of random images without any data pre-processing or prior assumptions about what we want the model to learn. We scale our model size to dense 10 billion parameters to avoid underfitting on a large data size. We extensively study and validate our model performance on over 50 benchmarks including fairness, robustness to distribution shift, geographical diversity, fine grained recognition, image copy detection and many image classification datasets. The resulting model, not only captures well semantic information, it also captures information about artistic style and learns salient information such as geolocations and multilingual word embeddings based on visual content only. More importantly, we discover that such model is more robust, more fair, less harmful and less biased than supervised models or models trained on object centric datasets such as ImageNet.

Code Repositories

facebookresearch/vissl
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
action-classification-on-kinetics-700SEER (RegNet10B)
Top-1 Accuracy: 51.9
domain-generalization-on-imagenet-aSEER (RegNet10B)
Top-1 accuracy %: 52.7
domain-generalization-on-imagenet-rSEER (RegNet10B)
Top-1 Error Rate: 43.9
domain-generalization-on-imagenet-sketchSEER (RegNet10B)
Top-1 accuracy: 45.6
fine-grained-image-classification-on-caltechSEER (RegNet10B - linear eval)
Accuracy: 91.0
Top-1 Error Rate: 9.0%
fine-grained-image-classification-on-fgvcSEER (RegNet10B)
Accuracy: 54.82%
fine-grained-image-classification-on-oxford-1SEER (RegNet10B)
Accuracy: 85.3%
fine-grained-image-classification-on-stanfordSEER (RegNet10B)
Accuracy: 68.03%
fine-grained-image-classification-on-sun397SEER (RegNet10B - linear eval)
Accuracy: 80.0
image-classification-on-cifar-10SEER (RegNet10B)
Percentage correct: 90
image-classification-on-cifar-100SEER (RegNet10B)
Percentage correct: 81.53
image-classification-on-clevr-countSEER (RegNet10B)
Top 1 Accuracy: 89.28
image-classification-on-clevr-countSEER (RegNetY-128GF)
Top 1 Accuracy: 87.98
image-classification-on-clevr-distSEER (RegNet10B)
Top 1 Accuracy: 74.98
image-classification-on-clevr-distSEER (RegNetY-128GF)
Top 1 Accuracy: 72.67
image-classification-on-dtdSEER (RegNet10B - linear eval)
Accuracy: 80.5
image-classification-on-eurosatSEER (RegNet10B - linear eval)
Accuracy (%): 97.5
image-classification-on-flowers-102SEER (RegNet10B)
Accuracy: 96.3
image-classification-on-food-101-1SEER (RegNet10B - linear eval)
Accuracy (%): 90.3
image-classification-on-imagenetSEER (RG-10B)
Number of params: 10000M
Top 1 Accuracy: 85.8%
image-classification-on-imagenet-realSEER (RegNet10B)
Accuracy: 89.8%
Params: 10000M
image-classification-on-imagenet-v2SEER (RegNet10B)
Top 1 Accuracy: 76.2
image-classification-on-inaturalist-2018SEER (RegNet10B - finetuned - 384px)
Top-1 Accuracy: 84.7%
image-classification-on-kitti-distSEER (RegNet10B)
Top 1 Accuracy: 78.34
image-classification-on-mnistSEER (RegNet10B)
Accuracy: 99.42
Percentage error: 0.58
image-classification-on-objectnetSEER (RegNet10B)
Top-1 Accuracy: 60.2
image-classification-on-places205SEER (RegNet10B - finetuned - 384px)
Top 1 Accuracy: 69.0
image-classification-on-resisc45ResNet50 (ImageNet-supervised)
Top 1 Accuracy: 88.56
image-classification-on-resisc45DeiT-B/16
Top 1 Accuracy: 92.48
image-classification-on-resisc45SimCLR-v2 (ResNet152-w3 + SK)
Top 1 Accuracy: 89.77
image-classification-on-resisc45MoCo-v3 (ViT-B/16)
Top 1 Accuracy: 93.35
image-classification-on-resisc45SwAV (ResNet50-w5)
Top 1 Accuracy: 94.73
image-classification-on-resisc45MoCo-v2 (ResNet50)
Top 1 Accuracy: 85.4
image-classification-on-resisc45SEER (RegNet10B)
Top 1 Accuracy: 95.61
image-classification-on-resisc45CLIP (ViT-B/16)
Top 1 Accuracy: 92.7
image-classification-on-resisc45DINO (DeiT-B/16)
Top 1 Accuracy: 93.97
image-classification-on-stl-10SEER (RegNet10B)
PARAMS: 10000M
Percentage correct: 97.3
image-classification-on-svhnSEER (RegNet10B)
Percentage error: 13.6
meme-classification-on-hateful-memesSEER (RegNet10B)
ROC-AUC: 0.734
self-supervised-image-classification-on-1SEER (Regnet10B)
Number of Params: 10000M
Top 1 Accuracy: 85.8%
semi-supervised-image-classification-on-1SEER (RegNet10B)
Top 1 Accuracy: 62.4%
semi-supervised-image-classification-on-2SEER (RegNet10B)
Top 1 Accuracy: 78.8%
traffic-sign-recognition-on-gtsrbSEER (RegNet10B)
Accuracy: 90.71%

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