arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.07487cs.CVcs.LG

DreamTeacher:使用深度生成模型预训练图像骨干网络

DreamTeacher: Pretraining Image Backbones with Deep Generative Models

NVIDIA(英伟达);University of Toronto(多伦多大学);Vector Institute(向量学院);MIT(麻省理工学院)

Daiqing Li, Huan Ling, Amlan Kar, David Acuna, Seung Wook Kim, Karsten Kreis, Antonio Torralba, Sanja Fidler

AI总结:

提出 DreamTeacher 自监督框架,通过将深度生成模型的特征和任务标签蒸馏到下游图像骨干网络,实现优于现有自监督方法及 ImageNet 分类预训练的表示学习效果。

摘要:

在本工作中,我们提出了一个自监督特征表示学习框架 DreamTeacher,该框架利用生成网络为下游图像骨干网络进行预训练。我们建议将来自训练好的生成模型的知识蒸馏到标准图像骨干网络中,这些骨干网络已针对特定感知任务进行了精心设计。我们研究了两类知识蒸馏:1)将学习到的生成特征蒸馏到目标图像骨干网络上,以替代在 ImageNet 等大型有标签数据集上预训练这些骨干网络;2)将生成网络借助任务头得到的标签蒸馏到目标骨干网络的 logits 上。我们在多个生成模型、密集预测基准和若干预训练机制上进行了广泛分析。我们通过实验发现,DreamTeacher 在所有方面均显著优于现有的自监督表示学习方法。使用 DreamTeacher 进行无监督 ImageNet 预训练,相比在下游数据集上进行 ImageNet 分类预训练带来了显著提升,这表明生成模型,尤其是扩散生成模型,是在无需人工标注的情况下于大型、多样化数据集上进行表示学习的一种有前景的方法。

英文摘要:

In this work, we introduce a self-supervised feature representation learning framework DreamTeacher that utilizes generative networks for pre-training downstream image backbones. We propose to distill knowledge from a trained generative model into standard image backbones that have been well engineered for specific perception tasks. We investigate two types of knowledge distillation: 1) distilling learned generative features onto target image backbones as an alternative to pretraining these backbones on large labeled datasets such as ImageNet, and 2) distilling labels obtained from generative networks with task heads onto logits of target backbones. We perform extensive analyses on multiple generative models, dense prediction benchmarks, and several pre-training regimes. We empirically find that our DreamTeacher significantly outperforms existing self-supervised representation learning approaches across the board. Unsupervised ImageNet pre-training with DreamTeacher leads to significant improvements over ImageNet classification pre-training on downstream datasets, showcasing generative models, and diffusion generative models specifically, as a promising approach to representation learning on large, diverse datasets without requiring manual annotation.

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

↑