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

DIME-FM:蒸馏多模态且高效的基础模型

DIME-FM: DIstilling Multimodal and Efficient Foundation Models

Boston University(波士顿大学);Meta AI

Ximeng Sun, Pengchuan Zhang, Peizhao Zhang, Hardik Shah, Kate Saenko, Xide Xia

AI总结:

本文提出 DIME-FM 蒸馏机制,仅用少量未配对公开图像和句子,将 CLIP-ViTL/14 的知识迁移到小型 ViT-B/32,并在零样本、线性探测及分布偏移鲁棒性上媲美大规模私有数据训练模型。

摘要:

大型视觉-语言基础模型(VLFM),例如 CLIP、ALIGN 和 Florence,是在图像-字幕对的大规模数据集上训练的,并在下游任务上取得了卓越的可迁移性和鲁棒性,但由于其体积庞大、延迟较高且架构固定,它们难以用于许多实际应用。遗憾的是,近期研究表明,目前使用公开的较小规模数据为资源受限的应用训练一个小型定制 VLFM 非常困难。本文提出一种新的蒸馏机制(DIME-FM),使我们能够利用相对少量、廉价且未配对的图像和句子,将大型 VLFM 中包含的知识迁移到更小的定制基础模型中。我们仅使用 4000 万张公开图像和 2840 万条未配对的公开句子,将预训练 CLIP-ViTL/14 模型的知识迁移到 ViT-B/32 模型。所得模型“Distill-ViT-B/32”可与在其私有 WiT 数据集(4 亿个图像-文本对)上预训练的 CLIP-ViT-B/32 模型相媲美:在 ImageNet 和 ELEVATER(20 个图像分类任务)基准上,Distill-ViT-B/32 在零样本和线性探测性能方面取得了相近结果。在来自 ImageNet 的五个具有自然分布偏移的数据集上评估时,它也表现出相当的鲁棒性。

英文摘要:

Large Vision-Language Foundation Models (VLFM), such as CLIP, ALIGN and Florence, are trained on large-scale datasets of image-caption pairs and achieve superior transferability and robustness on downstream tasks, but they are difficult to use in many practical applications due to their large size, high latency and fixed architectures. Unfortunately, recent work shows training a small custom VLFM for resource-limited applications is currently very difficult using public and smaller-scale data. In this paper, we introduce a new distillation mechanism (DIME-FM) that allows us to transfer the knowledge contained in large VLFMs to smaller, customized foundation models using a relatively small amount of inexpensive, unpaired images and sentences. We transfer the knowledge from the pre-trained CLIP-ViTL/14 model to a ViT-B/32 model, with only 40M public images and 28.4M unpaired public sentences. The resulting model "Distill-ViT-B/32" rivals the CLIP-ViT-B/32 model pre-trained on its private WiT dataset (400M image-text pairs): Distill-ViT-B/32 achieves similar results in terms of zero-shot and linear-probing performance on both ImageNet and the ELEVATER (20 image classification tasks) benchmarks. It also displays comparable robustness when evaluated on five datasets with natural distribution shifts from ImageNet.

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

↑