HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Self-supervised Pretraining of Visual Features in the Wild

Self-supervised Pretraining of Visual Features in the Wild

Abstract

Recently, self-supervised learning methods like MoCo, SimCLR, BYOL and SwAV have reduced the gap with supervised methods. These results have been achieved in a control environment, that is the highly curated ImageNet dataset. However, the premise of self-supervised learning is that it can learn from any random image and from any unbounded dataset. In this work, we explore if self-supervision lives to its expectation by training large models on random, uncurated images with no supervision. Our final SElf-supERvised (SEER) model, a RegNetY with 1.3B parameters trained on 1B random images with 512 GPUs achieves 84.2% top-1 accuracy, surpassing the best self-supervised pretrained model by 1% and confirming that self-supervised learning works in a real world setting. Interestingly, we also observe that self-supervised models are good few-shot learners achieving 77.9% top-1 with access to only 10% of ImageNet. Code: https://github.com/facebookresearch/vissl

Code Repositories

facebookresearch/vissl
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
image-classification-on-places205SEER
Top 1 Accuracy: 66.0
image-classification-on-places205RegNetY-128GF (Supervised)
Top 1 Accuracy: 62.7
self-supervised-image-classification-on-1SEER (RegNetY-256GF)
Number of Params: 1.3B
Top 1 Accuracy: 84.2%
self-supervised-image-classification-on-1SEER (RegNetY-128GF)
Number of Params: 693M
Top 1 Accuracy: 83.8%
semi-supervised-image-classification-on-1SEER Large (RegNetY-256GF)
Top 1 Accuracy: 60.5%
semi-supervised-image-classification-on-1SEER Small (RegNetY-128GF)
Top 1 Accuracy: 57.5%
semi-supervised-image-classification-on-2SEER Large (RegNetY-256GF)
Top 1 Accuracy: 77.9%
semi-supervised-image-classification-on-2SEER Small (RegNetY-128GF)
Top 1 Accuracy: 76.7%

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