消费级GPU即可微调Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
“只需”10万美元,训练Llama-2级别的大模型。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
尺寸更小但性能不减的MoE模型来了:Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
它叫JetMoE,来自MIT、普林斯顿等研究机构。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
性能妥妥超过同等规模的Llama-2。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM△贾扬清转发Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
要知道,后者可是数十亿美元级别的投入成本。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

JetMoE发布即完全开源,且学术界友好:仅使用公开数据集和开源代码,用消费级GPU就能进行微调。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
不得说,大模型的打造成本,真的比人们想的要便宜更多了。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
Ps. Stable Diffusion前老板Emad也点了赞:Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

10万美刀实现Llama-2性能
JetMoE启发于ModuleFormer的稀疏激活架构。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
(ModuleFormer,一种基于稀疏专家混合(SMoE)的模块化架构,可提高大模型效率和灵活性,去年6月提出)Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
它的注意力层中仍然使用了MoE:Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
80亿参数的JetMoE一共有24个区块,每块包含2个MoE层,分别是注意力头混合 (MoA) 和MLP专家混合 (MoE)。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
每个MoA和MoE层又有8个专家,每次输入token激活2个。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

JetMoE-8B使用公开数据集中的1.25T token进行训练,学习率5.0 x 10-4,全局batch size为4M token。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
具体训练方案遵循MiniCPM(来自面壁智能,2B模型就能赶超Mistral-7B)的思路,共包含两阶段:Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
第一阶段使用线性预热的恒定学习率,用来自大规模开源预训练数据集的1万亿个token进行训练,这些数据集包括RefinedWeb、Pile、Github data等等。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

第二阶段则使用指数学习率衰减,用2500亿个token训练来自第一阶段数据集和超高质量开源数据集的token。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

最终,团队使用96×H100的GPU集群,花费2周时间、约8万美元搞定JetMoE-8B。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
更多技术细节将在不久后发布的技术报告上揭露。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
而在推理过程中,由于JetMoE-8B仅具有22亿个激活参数,因此计算成本大大降低——Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
同时,它还收获了不错的性能表现。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
如下图所示:Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
JetMoE-8B在8个评测基准上获得了5个sota(包括大模型竞技场Open LLM Leaderboard),超过LLaMA-13B、LLaMA2-7B和DeepseekMoE-16B。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

在MT-Bench基准上得分6.681,也超过了130亿参数的LLaMA2、Vicuna等模型。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

作者介绍
JetMoE一共4位作者,分别是:Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
MIT-IBM Watson Lab研究员,研究方向NLP。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
本硕毕业于北航,博士经历于Yoshua Bengio创办的Mila研究机构。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
MIT博士在读, 研究方向为3D成像的数据高效机器学习。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
UC伯克利本科毕业,去年夏天作为学生研究员加入MIT-IBM Watson Lab,导师为Yikang Shen等人。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
普林斯顿博士在读生,本科毕业于北大应用数学和计算机科学,目前也是Together.ai 的兼职研究员,与Tri Dao合作。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
MIT博士在读,同时在创业,MyShell的AI研发主管。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
这家公司刚刚融资了1100万美元,投资者包括Transformer的作者。Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM

传送门:
https://github.com/myshell-ai/JetMoEFei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
参考链接:
https://twitter.com/jiayq/status/1775935845205463292Fei速刷资讯——每天刷点最新资讯,了解这个世界多一点SUSHUAPOS.COM
记者3月20日从中国科学院地质与地球物理研究所获悉,该所科研人员在东北黑土区开展了大范围的野外调查和样品采集工作,通过分析采集的黑土样品发现,黑土物质最初都是由风力搬运而来。相关研究成
3月21日记者从中国科学技术大学获悉,该校物理学院张斗国教授课题组,提出并实现了一种基于矢量光场调控原理的动量空间偏振滤波器件。科研人员将该滤波器件安装于传统无标记光学显微镜的出射端,
3月24日消息,今天数码博主“厂长是关同学”曝光了huaweiMate 70系列电话的部分配置信息。该博主表示,huawei全新的Mate 70系列首发会搭载新的芯片,芯片的性能差不多可以比肩5.5nm,还是值得期待的。同
3月22日,中国互联网络信息中心(CNNIC)发布第53次《中国互联网络发展状况统计报告》(以下简称《报告》)。《报告》显示,截至2023年12月,我国网民规模达10.92亿人,较2022年12月新增网民2480万人,互联网
今天(23日)是第64个世界气象日,今年世界气象日的主题是“气候行动最前线”。地球是人类赖以生存和发展的家园,气候变化给人类造成的影响广泛而深远,关乎当下,更关乎未来。世界气象组织确认2023年全
作为功能机时代的霸主,诺基亚手机曾经创造了巨大的辉煌,但也因错失智能手机发展机遇而“掉队”。近期,外媒Visual Capitalist统计了有史以来最畅销的15款手机型号,其中诺基亚和苹果包揽前十。今
。
本文链接:10万美元训出Llama-2级大模型!全华人打造新型MoE,贾扬清SD前CEO围观http://www.sushuapos.com/show-2-4591-0.html
声明:本网站为非营利性网站,本网页内容由互联网博主自发贡献,不代表本站观点,本站不承担任何法律责任。天上不会到馅饼,请大家谨防诈骗!若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。
上一篇: 超巧 全智贤 宋慧乔在机场撞型,她眼睛都会笑(超巧全智贤宋慧乔在机场撞型)
下一篇: 最懂AI的诺奖经济学得主去世,大模型关键技术受他研究启发