arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.10366cs.CVcs.AIcs.LG

通过ImageNet-1K访问视觉基础模型

Accessing Vision Foundation Models via ImageNet-1K

Northeastern University(东北大学);Khoury College of Computer Science, Northeastern University(东北大学科里计算机科学学院)

Yitian Zhang, Xu Ma, Yue Bai, Huan Wang, Yun Fu

AI总结:

针对视觉基础模型训练资源需求大、数据难获取的问题,提出Proteus方案,通过移除数据集偏差设计、设置三种训练目标,以ImageNet级成本将基础模型蒸馏为小型等效模型,在19个基准测试中性能优异,提升了基础模型训练的可访问性。

摘要:

视觉基础模型因大规模训练数据而具备卓越的泛化能力。然而,它们需要巨大的训练资源,且训练数据往往无法获取(例如CLIP、DINOv2),这对开发能促进研究的衍生模型构成了巨大挑战。在这项工作中,我们提出了一种非常简单且通用的解决方案,名为Proteus,无需访问原始训练数据,即可将基础模型蒸馏为ImageNet-1K上的小型等效模型。具体而言,我们移除了传统知识蒸馏设置中导致数据集偏差的设计,并提出了三种训练目标(即token、patch和特征),以最大化知识迁移的效果。通过这种方式,Proteus以ImageNet级别的成本进行训练,却具备惊人的能力,为更广泛的研究社区提供了训练基础模型的可访问性。当以DINOv2-g/14作为教师模型时,Proteus-L/14在19个基准测试中与使用1.42亿训练数据的Oracle方法DINOv2-L/14性能相当,且以仅120万图像的显著更小训练集,优于其他视觉基础模型(包括CLIP-L/14(4亿)、OpenCLIP-L/14(4亿/20亿)和SynCLR-L/14(6亿))。

英文摘要:

Vision foundation models are renowned for the generalization ability due to massive training data. Nevertheless, they demand tremendous training resources, and the training data is often inaccessible, e.g., CLIP, DINOv2, posing great challenges to developing derivatives that could facilitate the research. In this work, we offer a very simple and general solution, named \textit{Proteus}, to distill foundation models into smaller equivalents on ImageNet-1K without access to the original training data. Specifically, we remove the designs from conventional knowledge distillation settings that result in dataset bias and present three levels of training objectives, i.e., token, patch, and feature, to maximize the efficacy of knowledge transfer. In this manner, Proteus is trained at ImageNet-level costs with surprising ability, facilitating the accessibility of training foundation models for the broader research community. When leveraging DINOv2-g/14 as the teacher, Proteus-L/14 matches the performance of the Oracle method DINOv2-L/14 (142M training data) across 19 benchmarks and outperforms other vision foundation models including CLIP-L/14 (400M), OpenCLIP-L/14 (400M/2B) and SynCLR-L/14 (600M) with a significantly smaller training set of 1.2M images.

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

↑