清华大学与腾讯混元赢得MLSys2026 MoE推理挑战赛冠军,NPU推理速度提升4.1倍

2026年05月25日 10:34
本文共计661个字,预计阅读时长3分钟。
来源/aibase 责编/爱力方

清华大学存储实验室与腾讯混元AI Infra团队近日联合宣布,在国际机器学习系统顶级会议MLSys2026举办的MoE模型推理优化挑战赛中荣获全球冠军。

QQ20260525-090728.jpg

面对万亿参数时代混合专家(MoE)架构在异构芯片(NPU)上的推理性能瓶颈,联合团队针对官方指定模型及NPU硬件设计了全链路优化方案。通过引入按专家拆分双核任务的E-Shard策略、PSUM三维张量批量读出、打散输出至多Bank并发的GEMV路径,以及利用标量引擎降低初始搬运延迟等组合拳,团队成功攻克了数据搬运不充分、激活反复搬运等底层算子痛点。

同时,针对注意力模块,团队重构片上数据布局并融合Transformer关键算子,实现了比特级高精度对齐。

QQ20260525-090739.jpg

图3:MoE 优化结构示意图,包含E-Shard专家切分、连续 DMA、PSUM/GEMV并发、冷启动流水和预取控制。

在此次角逐中,团队还联合研发了基于Agent的推理算子优化器“Knight”,通过方案提出、代码落地与复盘迭代的自动化闭环,大幅拓展了优化搜索空间。最终,该方案将模型端到端推理时间由14.91s缩短至3.56s,性能加速达4.1倍;单步解码延迟从12.63ms降至5.45ms,权重加载期间的DMA引擎利用率攀升至约80%。

击败Stanford、MIT等国际顶尖高校夺冠,不仅彰显了我国团队在大模型底层系统适配与算子优化上的深厚积累,也为未来超节点算力平台部署万亿参数MoE模型提供了极具价值的工程范式。

来源:清华联合腾讯混元斩获MLSys2026MoE推理挑战赛冠军,NPU推理提速4.1倍 | AIbase

声明:本文来自aibase,版权归作者所有。文章内容仅代表作者独立观点,不代表A³·爱力方立场,转载目的在于传递更多信息。如有侵权,请联系 copyright#agent.ren。
0

热门资讯

推荐专栏

爱力方

爱力方

机器人前沿资讯及信息解读
机器人大讲堂

机器人大讲堂

中国顶尖的机器人专业媒体服务平台
关注爱力方,掌握前沿具身智能动态

© 2025 A³·爱力方

https://www.agentren.cn/