如何训练自定义chatgpt模型 轻量版ChatGPT训练方法开源!3天围绕LLaMA打造,号称训练快15倍

默认分类1年前 (2023)发布 admin
1,040 0
ChatGPT国内版

萧箫 发自 凹非寺

量子位 | 公众号

基于Meta模型打造的轻量版,这就来啦?

Meta宣布推出LLaMA才三天,业界就出现了把它打造成的开源训练方法,号称比训练速度最高快15倍。

LLaMA是Meta推出的超快超小型GPT-3,参数量只有后者的10%,只需要单张GPU就能运行。

把它变成的方法名叫,基于RLHF(基于人类反馈的强化学习)进行训练,在网上很快掀起了一阵热度。

所以,Meta的开源版真的要来了?

先等等,事情倒也没那么简单。

把LLaMA训练成的“开源方法”

点进项目主页来看,会发现它实际上集成了四个部分——

、RLHF方法、LLaMA和基于 agent生成的数据集。

如何训练自定义chatgpt模型 轻量版ChatGPT训练方法开源!3天围绕LLaMA打造,号称训练快15倍

其中,是一个开源深度学习训练优化库,包含名叫Zero的现存优化技术,用于提升大模型训练能力,具体指帮模型提升训练速度、降低成本、提升模型可用性等。

RLHF则会采用奖励模型来对预训练模型进行微调。奖励模型即先用多个模型生成问题问答,再依靠人工对问答进行排序,让它学会打分;随后,基于奖励学习给模型生成的回答进行打分,通过强化学习的方式增强模型能力。

是一个大语言模型应用开发库,希望将各种大语言模型整合起来,结合其他知识来源或计算能力创建一个实用的应用程序。 agent则会像思维链一样放出GPT-3思考的全过程,将操作记录下来。

这时候你会发现,最关键的依旧是LLaMA的模型权重。它从哪里来?

嘿嘿,自己去找Meta申请吧,并不提供。(虽然Meta声称开源LLaMA,但依旧需要申请)

所以本质上来说,并不是一个开源项目,而只是一种基于LLaMA的训练方法,其库内集成的几个项目原本也都是开源的。

实际上,也并非由Meta打造,而是来自一个叫做 AI的初创AI企业。

AI做了一个叫做的开源库,里面集成了一系列即插即用的优化模块,用于提升AI系统性能。

例如这是目前包含的一些模块,包括基于开源的算法打造的、自动感知硬件并对其进行加速的优化模块……

也在这一系列模块中,但要注意的是它的开源也是不可商用的。

如何训练自定义chatgpt模型 轻量版ChatGPT训练方法开源!3天围绕LLaMA打造,号称训练快15倍

所以“国产自研”想要直接拿去用,可能还没那么简单(doge)。

看完这个项目后,有网友表示,要是有人真搞到LLaMA的模型权重(代码)就好了……

但也有网友指出,“比训练方法快15倍”这种说法是一个纯纯的误导:

所谓的快15倍只是因为LLaMA模型本身很小,甚至能在单个GPU上运行,但应该不是因为这个项目所做的任何事情吧?

这位网友还推荐了一个比库中效果更好的RLHF训练方法,名叫trlx,训练速度要比通常的RLHF方法快上3~4倍:

你拿到LLaMA的代码了吗?觉得这个训练方法如何?

地址:

参考链接:

© 版权声明
广告也精彩

相关文章

暂无评论

暂无评论...