Skip to Main content Skip to Navigation
Conference papers

Unsupervised Pre-Training of Image Features on Non-Curated Data

Abstract : Pre-training general-purpose visual features with convolutional neural networks without relying on annotations is a challenging and important task. Most recent efforts in unsupervised feature learning have focused on either small or highly curated datasets like ImageNet, whereas using non-curated raw datasets was found to decrease the feature quality when evaluated on a transfer task. Our goal is to bridge the performance gap between unsupervised methods trained on curated data, which are costly to obtain, and massive raw datasets that are easily available. To that effect, we propose a new unsupervised approach which leverages self-supervision and clustering to capture complementary statistics from large-scale data. We validate our approach on 96 million images from YFCC100M [42], achieving state-of-the-art results among unsupervised methods on standard benchmarks, which confirms the potential of unsupervised learning when only non-curated raw data are available. We also show that pre-training a supervised VGG-16 with our method achieves 74.9% top-1 classification accuracy on the validation set of ImageNet, which is an improvement of +0.8% over the same network trained from scratch. Our code is available at https://github.com/facebookresearch/DeeperCluster.
Complete list of metadatas

Cited literature [57 references]  Display  Hide  Download

https://hal.archives-ouvertes.fr/hal-02119564
Contributor : Mathilde Caron <>
Submitted on : Monday, September 9, 2019 - 11:52:36 AM
Last modification on : Wednesday, October 14, 2020 - 4:17:11 AM

File

main.pdf
Files produced by the author(s)

Identifiers

Collections

Citation

Mathilde Caron, Piotr Bojanowski, Julien Mairal, Armand Joulin. Unsupervised Pre-Training of Image Features on Non-Curated Data. ICCV 2019 - International Conference on Computer Vision, Oct 2019, Seoul, South Korea. pp.2959-2968, ⟨10.1109/ICCV.2019.00305⟩. ⟨hal-02119564v2⟩

Share

Metrics

Record views

214

Files downloads

1112