最大CLIP！LAION发布CLIP的扩增定律最新消息

时间：2022-12-29 10:29:03 来源:

(资料图)

点蓝色字关注“机器学习算法工程师”

设为星标，干货直达！

近日，LAION等机构在Reproducible scaling laws for contrastive language-image learning发布了CLIP的扩增定律（scaling law），其中最大的CLIP为ViT-H/14，基于LAION-2B数据集训练，可以在ImageNet1K数据集上到78.0%的zero-shot准确度，性能超过OpenAI目前开源的CLIP L/14，和Meta AI的FLIP Huge模型性能相当，但是模型已经开源在https://github.com/LAION-AI/scaling-laws-openclip。

论文的实验采用开源数据集LAION-400M和LAION-2B数据集，训练框架采用开源的OpenCLIP，所以论文的实验是可以复现的。

训练在1520 NVIDIA A100 GPUs上进行，采用PyTorch DDP分布式训练策略，采用混合精度（但是fp16会不稳定，所以采用bf16，或者基于TF32的float32），训练的batch size在 86-88K之间。

实验的总体结论是：扩增定律也明显适用CLIP，当扩增模型，训练数据和算力时，模型在下游任务上有一致性的提升。但是不同的训练数据集表现出不同的扩增系数：OpenCLIP的模型（基于LAION-2B数据集）在图文检索任务上有较大的扩增系数，而OpenAI CLIP模型（基于私有的WebImageText 400M数据集）在zero-shot分类任务上有较强的扩增系数。

这个结论和FLIP的结论比较吻合，这说明训练数据集对CLIP的性能确实有比较大的影响。用论文结论的一句话来说就是：Scaling behavior depends on task type and pre-training dataset。

更多内容可见论文：https://arxiv.org/abs/2212.07143

标签：

新闻快讯

X 关闭

精品推荐

X 关闭

最大CLIP！LAION发布CLIP的扩增定律最新消息

精彩推送

31省份新增确诊病例43例其中本土病例28例

陕西华北黄淮有较强降水过程热带低压影响海南岛等地

神舟十二号重返地球：“天地大搜救”

藏身网络平台非法引流是如何“浇灌”黑灰产的？

黑龙江省22日12-24时新增新冠肺炎本土确诊病例3例

一大波雨水将至！北京今天白天有分散性阵雨傍晚起降雨增多

久坐、体重大、锻炼少，腰椎间盘突出越来越年轻化

妈妈做手术带着26岁残障儿子住院医院开“特殊病房”

清理入侵物种恢复海岸生态

新闻快讯

精品推荐

新闻快讯

最大CLIP！LAION发布CLIP的扩增定律 最新消息

精彩推送

新闻快讯

精品推荐

新闻快讯

最大CLIP！LAION发布CLIP的扩增定律最新消息