公众号:小红书技术(dots.llm)· 小红书技术REDtech·· 2026-07-20精选AI 评分76
小红书、北大开源 UltraEP:面向大规模 MoE 训推的「最优」负载均衡方案
AI 导读
小红书与北大开源 UltraEP,首次将基于精确路由信息的实时负载均衡引入生产系统,在每个 microbatch 和每一层动态复制热点专家,平均达到理想性能的 94.3%,相比业界 SOTA 训推框架提升 1.49 倍。该方案已在 288B 参数 MoE 模型的完整预训练中部署,保持不低于理想性能 92% 的水平。
推荐理由
动态复制热点专家替代了基于历史窗口的重排,将MoE训推吞吐从理想的一半拉至90%以上,对自研大规模模型的团队而言,是一套可落地的负载均衡方案。
来源:公众号:小红书技术(dots.llm) · mp.weixin.qq.com