两只羊驼掐头去尾拼一起,屠榜HuggingFace

人工智能 新闻
相关论文刚刚上传到ArXiv,来自韩国公司Upstage AI,使用了新的大模型扩展方法depth up-scaling(DUS)。

HuggingFace开源大模型排行榜,又被屠榜了。

前排被清一色的SOLAR 10.7B微调版本占据,把几周之前的各种Mixtral 8x7B微调版本挤了下去。

图片

SOLAR大模型什么来头?

相关论文刚刚上传到ArXiv,来自韩国公司Upstage AI,使用了新的大模型扩展方法depth up-scaling(DUS)

图片

简单来说就是两只7B羊驼掐头去尾,一只砍掉前8层,一只砍掉后8层。

剩下两个24层缝合在一起,第一个模型的第24层与第二个模型的第9层拼接,最后变成新的48层10.7B大模型。

图片

论文声称新方法超过传统扩展方法如MoE,而且可以与沿用基础大模型完全相同的基础设施。

不需要门控网络等附加模块,针对MoE优化训练框架了,也不需要自定义CUDA内核来快速推理,可以无缝集成到现有方法中,同时保持高效。

团队选择7B规模最强的单体大模型Mistral 7B作为底材,用新方法拼接起来,再超越原版以及MoE版。

同时,经过对齐的Instruct版本也超越对应的MoE Instruct版本。

图片

将缝合进行到底

为什么是这种拼接方式,论文中介绍来自一种直觉。

从最简单的扩展方式开始,也就是把32层的基础大模型重复两次,变成64层。

这样做的好处是不存在异质性,所有层都来自基础大模型,但第32层和第33层(与第1层相同)的接缝处有较大的“层距离”(layer distance)

之前有研究表明,Transformer不同层做不同的事,如越深的层擅长处理越抽象的概念。

团队认为层距离过大可能妨碍模型有效利用预训练权重的能力。

一个潜在的解决方案是牺牲中间层,从而减少接缝处的差异,DUS方法就从这里诞生。

根据性能与模型尺寸的权衡,团队选择从每个模型中删除8层,接缝处从32层连第1层,变成了24层连第9层。

简单拼接后的模型,性能一开始还是会低于原版基础模型,但经过继续预训练可以迅速恢复。

在指令微调阶段,除了使用开源数据集,还制作了数学强化数据集,对齐阶段使用DPO。

最后一步,把使用不同数据集训练的模型版本加权平均,也是把缝合进行到底了。

图片

有网友质疑测试数据泄露的可能性。

图片

团队也考虑到这一点,在论文附录中专门报告了数据污染测试结果,显示出低水平。

图片

最后,SOLAR 10.7B基础模型和微调模型都以Apache 2.0协议开源。

试用过的网友反馈,从JSON格式数据中提取数据表现不错。

图片

论文地址:https://arxiv.org/abs/2312.15166

责任编辑:张燕妮 来源: 量子位
相关推荐

2022-09-27 16:03:51

羊了个羊鸿蒙

2013-09-03 13:14:15

诺基亚Google

2016-03-25 16:02:49

中国企业新闻网

2023-05-22 09:28:30

模型AI

2022-02-17 20:18:27

JS鸿蒙操作系统

2020-11-04 09:43:09

helloworld

2009-04-14 15:30:25

2010-04-30 14:33:56

HTML 5FlashH.264

2011-09-21 11:28:03

HTCWindows 8平板电脑

2015-04-15 10:29:38

Linux 4.0

2022-11-29 16:35:02

Tetris鸿蒙

2022-12-02 14:20:09

Tetris鸿蒙

2017-03-03 18:02:32

华为

2023-03-30 09:32:27

2022-11-14 17:01:34

游戏开发画布功能

2014-10-21 15:07:04

2015-08-19 09:53:17

技术电商创业

2009-08-10 18:16:20

C#面试题

2022-03-31 18:59:43

数据库InnoDBMySQL

2021-08-27 07:06:09

DubboDocker技术
点赞
收藏

51CTO技术栈公众号