字节开源MoE关键优化技术,训练成本再砍40%!内部万卡集群已部署
创始人
2025-03-11 08:11:26
0

新浪科技讯 3月10日晚间消息,字节豆包大模型团队官宣开源一项针对 MoE 架构的关键优化技术,可将大模型训练效率提升1.7倍,成本节省40%。据悉,该技术已实际应用于字节的万卡集群训练,累计帮助节省了数百万 GPU 小时训练算力。

MoE 是当前大模型的主流架构,但其在分布式训练中存在大量跨设备通信开销,严重制约了大模型训练效率和成本。以海外主流模型Mixtral-8x7B为例, 其训练过程中通信时间占比可高达 40%。针对这一难题,字节在内部研发了COMET计算-通信重叠技术,通过多项创新,大幅压缩了MoE专家通信空转时间。

相较DeepSeek近期开源的DualPipe等MoE优化方案,COMET可以像插件一样直接接入已有的MoE训练框架,支持业界绝大部分主流大模型,无需对训练框架进行侵入式改动。因简洁、通用的设计理念,该工作以5/5/5/4 的高分入选全球机器学习系统顶级会议 MLSys 2025 ,被认为“在大规模生产环境中极具应用潜力”。

具体而言, COMET 从系统层面建立了面向 MoE 的细粒度流水线编程方式,通过引入共享张量依赖解析、自适应负载分配两项关键机制,来解决通信与计算之间的粒度错配问题,并精准平衡通信与计算负载,最终大幅提升MoE流水线整体效率。 引入COMET后,单个 MoE 层上可实现 1.96 倍加速、端到端平均 1.71 倍效率提升,且在不同并行策略、输入规模及硬件环境下均表现稳定。

值得一提的是,COMET 与Deepseek 研发的DualPipe方案还可以联合使用。在降低MoE通信开销上,COMET 采用了计算-通信融合算子的优化方式, DualPipe则通过排布算子来掩盖通信,两种方案并不冲突,结合使用或将更大幅度压缩模型训练成本。

目前,COMET支持多种MoE并行模式,部署灵活、方便。同时,COMET核心代码已开源,并向开发者提供了一套友好的 Python API,计划兼容 Triton 等编译生态。(罗宁)

【免责声明】本文仅代表作者本人观点,与和讯网无关。和讯网站对文中陈述、观点判断保持中立,不对所包含内容的准确性、可靠性或完整性提供任何明示或暗示的保证。请读者仅作参考,并请自行承担全部责任。邮箱:news_center@staff.hexun.com

相关内容

AI赋能创新 激发消费新动...
2024年中央经济工作会议提到“开展‘人工智能+’行动,培育未来产...
2025-03-12 16:40:12
浙商证券:AI医疗进入多模...
智通财经APP获悉,浙商证券发布研报称,大模型的推理能力和多模态能...
2025-03-12 16:14:02
东华软件:公司未运用Man...
人民财讯3月12日电,东华软件(002065)3月12日在互动平台...
2025-03-12 13:42:47
5G+ AI解锁远程医疗的...
2月28日,新疆喀什地区第二人民医院和上海复旦大学附属眼耳鼻喉科医...
2025-03-12 12:14:58
解密广东隐形武器!杜锋魔鬼...
在过去的一周里,广东东阳光男篮如同一阵飓风,先是以惊人的44分优势...
2025-03-12 12:14:11
君逸数码:AI智能体产品可...
金融界3月12日消息,有投资者在互动平台向君逸数码提问:请问董秘,...
2025-03-12 12:13:38

热门资讯

王家大院现在的所有者还是王家的... 王家大院现在的所有者还是王家的后人吗?我说的是山西灵石的王家大院...现在不知道是被收为国,还是仍为...
写字好看的女生有什么优势 写字好看的女生有什么优势见字如见人,字好看 很加分的。学校有书法比赛时很吃香哦感觉没有打字快的女生有...
杨大勇的妻子是谁 杨大勇的妻子是谁杨大勇的妻子是一位名叫王小丽的女性。据悉,王小丽与杨大勇相识于大学时期,两人相恋多年...
我家办白事,朋友给我发红包,我... 我家办白事,朋友给我发红包,我该怎么说感谢话我家办白事,朋友给我发红包,我该怎么说感谢话当家里有事情...
顶级绝伦推理片100部介绍 顶级绝伦推理片100部介绍 《白夜追凶》;可以说是刑侦国剧天花板了,逻辑,叙事方式,主演演技,这些几...
小猿众包骗局 小猿众包骗局小猿众包是小猿旗下的可以在家做题赚钱的兼职,平时如果时间充裕的情况下可以做做小猿众包挣个...
女生说男生丑萌什么意思? 女生说男生丑萌什么意思?丑萌就是又丑又萌,意思是男生在她的审美里不是好看的那种,但是又很戳她萌点就是...
自从和女朋友确定关系后,女朋友... 自从和女朋友确定关系后,女朋友为什么每天晚上发视频要我给她讲故事哄她睡觉?每次给她讲一个小时她都不睡...
外婆发外孙朋友圈说说有哪些? 外婆发外孙朋友圈说说有哪些? 1、天伦之乐,幸福便是如此简单。2、难得好时光,携孙儿共享天伦之乐,哪...
一个女生。让我去找她玩。是什么... 一个女生。让我去找她玩。是什么意思呢?求解。她是对你有好感的,不然也不会邀请你去找她玩。有可能把你当...