提升移动端图文模型的零样本性能,实现低延迟下的高精度。
MobileCLIP2: Improving Multi-Modal Reinforced Training
- 用多模态强化训练优化图文对齐,结合多种生成器与教师模型。
- 在ImageNet-1k上比前代模型提升2.2%零样本准确率,效率更高。
- 适合移动设备部署,支持快速构建新数据集,开源代码可复现。
基础图文模型如CLIP具备零样本能力,广泛应用于各类场景。MobileCLIP是一系列延迟3-15毫秒、参数量50-150M、零样本准确率处于领先水平的轻量级模型。其核心在于低延迟架构与一种高效的多模态强化训练方法,实现了来自多个图像描述生成器和CLIP教师模型的知识蒸馏,具备高效、可扩展、可复现的特点。本文通过改进:1)在DFN数据集上训练更优的CLIP教师集成;2)在DFN上训练并基于高质量图文数据集微调的描述生成器教师模型,进一步优化了多模态强化训练。通过消融实验发现:对比知识蒸馏中温度调节的重要性、描述生成器微调对多样性的作用,以及多模型合成描述的叠加增益。我们训练出新一代模型MobileCLIP2,实现了低延迟条件下的最先进零样本准确率。其中,MobileCLIP2-B相比MobileCLIP-B在ImageNet-1k上提升2.2%准确率。尤为显著的是,MobileCLIP2-S4在零样本准确率上达到SigLIP-SO400M/14水平,但体积小2倍,并在2.5倍更低延迟下优于DFN ViT-L/14。我们已开源预训练模型(https://github.com/apple/ml-mobileclip)及数据生成代码(https://github.com/apple/ml-mobileclip-dr),后者支持使用任意教师模型进行分布式可扩展的数据生成。
原文摘要 · Abstract (English)
Foundation image-text models such as CLIP with zero-shot capabilities enable a wide array of applications. MobileCLIP is a recent family of image-text models at 3-15ms latency and 50-150M parameters with state-of-the-art zero-shot accuracy. The main ingredients in MobileCLIP were its low-latency and light architectures and a novel multi-modal reinforced training that made knowledge distillation from multiple caption-generators and CLIP teachers efficient, scalable, and reproducible. In this paper, we improve the multi-modal reinforced training of MobileCLIP through: 1) better CLIP teacher ensembles trained on the DFN dataset, 2) improved captioner teachers trained on the DFN dataset and fine-tuned on a diverse selection of high-quality image-caption datasets. We discover new insights through ablations such as the importance of temperature tuning in contrastive knowledge distillation, the effectiveness of caption-generator fine-tuning for caption diversity, and the additive improvement from combining synthetic captions generated by multiple models. We train a new family of models called MobileCLIP2 and achieve state-of-the-art ImageNet-1k zero-shot accuracies at low latencies. In particular, we observe 2.2% improvement in ImageNet-1k accuracy for MobileCLIP2-B compared with MobileCLIP-B architecture. Notably, MobileCLIP2-S4 matches the zero-shot accuracy of SigLIP-SO400M/14 on ImageNet-1k while being 2$\times$ smaller and improves on DFN ViT-L/14 at 2.5$\times$ lower latency. We release our pretrained models (https://github.com/apple/ml-mobileclip) and the data generation code (https://github.com/apple/ml-mobileclip-dr). The data generation code makes it easy to create new reinforced datasets with arbitrary teachers using distributed scalable processing.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。