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

基于概念簇复杂度的网络规模数据集有效剪枝

Effective pruning of web-scale datasets based on complexity of concept clusters

University of Tübingen(蒂宾根大学);Meta AI (FAIR)(Meta AI(FAIR));ELLIS Institute Tübingen;Max-Planck Institute for Intelligent Systems Tübingen(蒂宾根马克斯·普朗克智能系统研究所);University of California San Diego(加州大学圣迭戈分校);DatologyAI

Amro Abbas, Evgenia Rusak, Kushal Tirumala, Wieland Brendel, Kamalika Chaudhuri, Ari S. Morcos

AI总结:

针对网络规模多模态数据集训练计算成本高的问题,提出基于概念复杂度的自适应剪枝方法,在仅用27.7%LAION数据与算力的情况下,提升了CLIP风格模型在多项任务上的性能,刷新了DataComp Medium基准的ImageNet零样本准确率纪录。

摘要:

利用海量网络规模数据集为机器学习模型带来了前所未有的性能提升,但也对其训练提出了极高的计算要求。为提升训练效率与数据效率,本文针对用于训练CLIP风格模型的大规模多模态数据集,探索了剪枝的性能极限。当前ImageNet上最高效的剪枝方法会根据嵌入将数据样本聚类为不同概念,并剪去最具原型性的样本。我们将该方法扩展至LAION数据集,并通过提出剪枝率应针对具体概念、且与概念复杂度相适配的思路对其进行改进。借助一种简单直观的复杂度度量方式,我们能够将训练成本降至常规训练的四分之一。通过对LAION数据集进行过滤,我们发现使用规模更小的高质量数据集进行训练,能够以显著更低的训练成本实现更优的性能。具体而言,我们在ImageNet零样本准确率上超过了基于LAION训练的OpenCLIP-ViT-B32模型1.1个百分点,而仅使用了27.7%的数据与训练计算量。尽管训练成本大幅降低,我们在ImageNet分布偏移任务、检索任务以及VTAB上也观察到了性能提升。在DataComp Medium基准上,我们取得了全新的ImageNet零样本准确率最优结果,且在38项评估任务上的平均零样本准确率具备竞争力。

英文摘要:

Utilizing massive web-scale datasets has led to unprecedented performance gains in machine learning models, but also imposes outlandish compute requirements for their training. In order to improve training and data efficiency, we here push the limits of pruning large-scale multimodal datasets for training CLIP-style models. Today's most effective pruning method on ImageNet clusters data samples into separate concepts according to their embedding and prunes away the most prototypical samples. We scale this approach to LAION and improve it by noting that the pruning rate should be concept-specific and adapted to the complexity of the concept. Using a simple and intuitive complexity measure, we are able to reduce the training cost to a quarter of regular training. By filtering from the LAION dataset, we find that training on a smaller set of high-quality data can lead to higher performance with significantly lower training costs. More specifically, we are able to outperform the LAION-trained OpenCLIP-ViT-B32 model on ImageNet zero-shot accuracy by 1.1p.p. while only using 27.7% of the data and training compute. Despite a strong reduction in training cost, we also see improvements on ImageNet dist. shifts, retrieval tasks and VTAB. On the DataComp Medium benchmark, we achieve a new state-of-the-art Imagehttps://info.arxiv.org/help/prep#commentsNet zero-shot accuracy and a competitive average zero-shot accuracy on 38 evaluation tasks.

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

↑