DINOv2:无需监督学习鲁棒视觉特征

DINOv2: Learning Robust Visual Features without Supervision

马克西姆·奥卡布 Maxime Oquab · Meta AI · 2023-04-14 · arXiv:2304.07193 ↗ · 被引 9029

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

自然语言处理中基于大量数据预训练模型的近期突破,为计算机视觉领域类似的基础模型开辟了道路。这类模型通过生成通用视觉特征(即无需微调即可跨图像分布和任务工作的特征),能极大简化图像在任何系统中的使用。本研究表明,现有的预训练方法,特别是自监督方法,如果在来自不同来源的足够精选数据上进行训练,就能产生此类特征。我们重新审视现有方法,并结合不同技术来扩展数据和模型规模的预训练。大部分技术贡献旨在加速和稳定大规模训练。在数据方面,我们提出了一种自动流水线,用于构建专用、多样且精选的图像数据集,而非自监督文献中通常使用的未精选数据。在模型方面,我们训练了一个具有 10 亿参数的 ViT 模型(Dosovitskiy 等人,2020),并将其蒸馏成一系列较小的模型,这些模型在图像和像素级别的大多数基准测试中超越了最佳可用的通用特征 OpenCLIP(Ilharco 等人,2021)。

The recent breakthroughs in natural language processing for model pretraining on large quantities of data have opened the way for similar foundation models in computer vision. These models could greatly simplify the use of images in any system by producing all-purpose visual features, i.e., features that work across image distributions and tasks without finetuning. This work shows that existing pretraining methods, especially self-supervised methods, can produce such features if trained on enough curated data from diverse sources. We revisit existing approaches and combine different techniques to scale our pretraining in terms of data and model size. Most of the technical contributions aim at accelerating and stabilizing the training at scale. In terms of data, we propose an automatic pipeline to build a dedicated, diverse, and curated image dataset instead of uncurated data, as typically done in the self-supervised literature. In terms of models, we train a ViT model (Dosovitskiy et al., 2020) with 1B parameters and distill it into a series of smaller models that surpass the best available all-purpose features, OpenCLIP (Ilharco et al., 2021) on most of the benchmarks at image and pixel levels.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 24)

阅读逐段中英对照全文 →