苹果小模型来了:权重、代码、数据集全开源,性能超越Mistral-7B

点击蓝字

关注我们

关注并星标

从此不迷路

计算机视觉研究院

1c2ce57bca4727010a849966355586a4.gif

9e8d3317a1d7b1de8be3c8c539db9cc6.gif

公众号ID计算机视觉研究院

学习群扫码在主页获取加入方式

b3d5799b4e11b0bef6a08ee6e4f71466.png

00050911b4d5078d98bbf649ae0eabf1.png

  • 论文链接:https://arxiv.org/pdf/2406.11794

  • 项目链接:https://huggingface.co/apple/DCLM-7B

计算机视觉研究院专栏

Column of Computer Vision Institute

小模型成趋势?

bfbfdb8e9ac627b3ef9e58b464d29962.gif

本周,OpenAI 上线小模型 GPT-4o-mini,小模型赛道正式开卷。近期加入这一赛道的还有苹果。

最近,苹果公司作为 DataComp-LM(DCLM)项目的研究机构之一,在 Hugging Face 上发布了 DCLM-7B 开源模型。该模型性能已经超越了 Mistral-7B,并且正在逼近其他领先的开源模型,包括 Llama 3 和 Gemma。

论文作者之一、苹果机器学习团队 Vaishaal Shankar 将 DCLM 模型描述为「真正开源的最佳模型」,因为 DCLM 不仅开源了模型权重,还开源了训练代码和预训练数据集。

e89faaa4227d519243867cc342bfcbc1.png

研究介绍

大型语言模型(LLM)目前面临的一个评估挑战是缺乏受控比较。LLM 研究通常会比较采用不同架构、计算或超参数的模型,因此难以理清影响语言模型质量的因素。

基于此,研究团队提出了语言模型数据比较新基准 ——DCLM,这是语言模型训练数据整编(curation)的第一个基准,旨在让 LLM 通过设计高质量数据集来提高模型性能,特别是在多模态领域。

研究团队发现基于模型的过滤,即由机器学习 (ML) 模型从较大的数据集中自动过滤和选择高质量数据,可能是构建高质量训练集的关键。

DCLM 整体思路很简单:使用一个标准化的框架来进行实验,包括固定的模型架构、训练代码、超参数和评估,最终找出哪种数据整理策略最适合训练出高性能的模型。

1bc29ea76ee05031e43a357cab170dee.png

使用 DCLM,研究团队构建了一个高质量数据集 DCLM-BASELINE,并用该数据集从头开始训练了一个 7B 参数模型 —— DCLM-7B。

50965e18bd88f205d479fdd2563e7c0e.png

7f3aee8c79c2e0e543a46842450a5efb.png

DCLM-7B 模型的细节。

DCLM-7B 使用基于 OpenLM 框架的预训练方案,在 MMLU 基准上 5-shot 准确率达到 64%,可与 Mistral-7B-v0.3(63%)和 Llama 3 8B(66%)相媲美,并且在 53 个自然语言理解任务上的平均表现也可与 Mistral-7B-v0.3、Llama 3 8B 相媲美,而所需计算量仅为 Llama 3 8B 的 1/6。

5e6b06bf17b631f34fffe741471a52fd.png

以下是 DCLM-7B 在各种任务(部分)上的评估结果:

45c83e542a3cf9caddc2b66beec74290.png

DCLM-7B 与其他同等大小模型比较结果如下表所示:

3cefbb5d606ab8527cdfcecc5bb96b9e.png

值得注意的是,大部分其他模型虽然开放权重但封闭数据。这就是 Vaishaal Shankar 将 DCLM 模型描述为「真正开源」的原因。

参考链接:https://venturebeat.com/ai/apple-shows-off-open-ai-prowess-new-models-outperform-mistral-and-hugging-face-offerings/

0c34b8ca7508965df19102bcca4f5f22.gif

END

494f4aeba003534e9fa22d9c34074c5a.gif

33aefe3b45d7c633d8755ba9a8ef6fcd.gif

转载请联系本公众号获得授权

d0adbbd8e807c01f72d6e4890ed87cca.gif

计算机视觉研究院学习群等你加入!

ABOUT

计算机视觉研究院

计算机视觉研究院主要涉及深度学习领域,主要致力于目标检测、目标跟踪、图像分割、OCR、模型量化、模型部署等研究方向。研究院每日分享最新的论文算法新框架,提供论文一键下载,并分享实战项目。研究院主要着重”技术研究“和“实践落地”。研究院会针对不同领域分享实践过程,让大家真正体会摆脱理论的真实场景,培养爱动手编程爱动脑思考的习惯!

8cf9dbd1818112677d6ad1107aaf863b.png

 往期推荐 

🔗

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值