arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2105.08054cs.CV

Divide and Contrast: Self-supervised Learning from Uncurated Data

MIT(麻省理工学院);DeepMind(深度思维)

Yonglong Tian, Olivier J. Henaff, Aaron van den Oord

英文摘要:

Self-supervised learning holds promise in leveraging large amounts of unlabeled data, however much of its progress has thus far been limited to highly curated pre-training data such as ImageNet. We explore the effects of contrastive learning from larger, less-curated image datasets such as YFCC, and find there is indeed a large difference in the resulting representation quality. We hypothesize that this curation gap is due to a shift in the distribution of image classes -- which is more diverse and heavy-tailed -- resulting in less relevant negative samples to learn from. We test this hypothesis with a new approach, Divide and Contrast (DnC), which alternates between contrastive learning and clustering-based hard negative mining. When pretrained on less curated datasets, DnC greatly improves the performance of self-supervised learning on downstream tasks, while remaining competitive with the current state-of-the-art on curated datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑