通过规模定律比较视觉语言模型,发现MaMMUT更高效、更易扩展。
Scaling Laws for Robust Comparison of Open Foundation Language-Vision Models and Datasets
- 基于大规模数据测量,推导出CLIP与MaMMUT的完整规模定律。
- MaMMUT在更大规模下提升更强,样本效率优于标准CLIP。
- 适用于不同任务与数据集,支持低成本对比实验。
在迁移学习研究中,规模定律可用于预测基础模型在更大规模下的性能。本文首次为两种重要多模态学习方法——仅使用对比损失的CLIP和同时采用对比与文本生成损失的MaMMUT——构建了覆盖广泛模型规模与训练样本量的完整规模定律。在保证外推点预测精度的前提下,利用这些定律对两者进行比较,发现MaMMUT在规模扩大时表现更优,且样本效率更高。为增强比较可靠性,我们在分类、检索与分割等下游任务,以及DataComp、DFN和Re-LAION等多个开源数据集上均观察到一致趋势。此外,即使采用固定学习率调度也能有效推导规模定律,降低计算成本。准确的规模定律可实现跨尺度的模型与数据集系统性比较,避免仅依赖单一参考尺度导致的误判,为开放基础模型与数据集的改进提供新路径。我们发布了所有预训练模型及其中间检查点,包括在128亿样本(DataComp-1.4B)上训练的openMaMMUT-L/14,零样本ImageNet-1k准确率达80.3%。代码与原始实验数据见:https://github.com/LAION-AI/scaling-laws-for-comparison。
原文摘要 · Abstract (English)
In studies of transferable learning, scaling laws are obtained for various important foundation models to predict their properties and performance at larger scales. We show here how scaling law derivation can also be used for model and dataset comparison, allowing to decide which procedure is to be preferred for pre-training. For the first time, full scaling laws based on dense measurements across a wide span of model and samples seen scales are derived for two important language-vision learning procedures, CLIP and MaMMUT, that use either contrastive only or contrastive and captioning text generative loss. Ensuring sufficient prediction accuracy for held out points, we use derived scaling laws to compare both models, obtaining evidence for MaMMUT's stronger improvement with scale and better sample efficiency than standard CLIP. To strengthen validity of the comparison, we show scaling laws for various downstream tasks, classification, retrieval, and segmentation, and for different open datasets, DataComp, DFN and Re-LAION, observing consistently the same trends. We show that comparison can also be performed when deriving scaling laws with a constant learning rate schedule, reducing compute cost. Accurate derivation of scaling laws provides thus means to perform model and dataset comparison across scale spans, avoiding misleading conclusions based on measurements from single reference scales only, paving the road for systematic comparison and improvement of open foundation models and datasets for their creation. We release all the pre-trained models with their intermediate checkpoints, including openMaMMUT-L/14, which achieves $80.3\%$ zero-shot ImageNet-1k accuracy, trained on 12.8B samples from DataComp-1.4B. Code for reproducing experiments in the paper and raw experiments data can be found at https://github.com/LAION-AI/scaling-laws-for-comparison.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。