Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Self-supervised Pretraining of Visual Features in the Wild

About

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

Priya Goyal, Mathilde Caron, Benjamin Lefaudeux, Min Xu, Pengchao Wang, Vivek Pai, Mannat Singh, Vitaliy Liptchinsky, Ishan Misra, Armand Joulin, Piotr Bojanowski• 2021

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)--
2643
Image ClassificationImageNet (val)
Top-1 Acc84.2
1206
Instance SegmentationCOCO 2017 (val)--
1201
Object DetectionCOCO (val)
mAP41.6
633
Instance SegmentationCOCO (val)
APmk37.6
475
Instance SegmentationCOCO
APmask43.2
291
Object DetectionCOCO
AP (Box)48.5
144
Image ClassificationImageNet 1% labeled--
118
Image ClassificationPlaces205 (val)
Top-1 Accuracy56
68
Image ClassificationVOC 2007 (test)
mAP89.4
67
Showing 10 of 17 rows

Other info

Code

Follow for update