欢迎光临
我们一直在努力

MoE 负载均衡深度解析:专家累死、摸鱼与负载崩塌

简介

MoE(混合专家模型)最致命的原生缺陷:路由器马太正反馈引发负载崩塌。训练过程中路由器会持续把 Token 路由到少数表现更好的专家,其余专家长期接收样本不足陷入休眠;显存、参数资源大量浪费,MoE 稀疏加速增益直接消失,性能甚至劣于同规模稠密大模型。 行业主流解法是引入负载均衡辅助损失,在主任务损失之外新增惩罚项,约束路由器不能过度偏向部分专家。 本文拆解负载崩塌发生机理、危害、各类均衡损失公式、工程实现伪代码,最后分析 Kimi‑MoE 落地现状,同时明确整套方案的适用边界。

比喻

把 MoE 想象成一家拥有 8 个技术小组(专家)的外包公司,调度员就是路由器。 调度员考核原本只看交付质量,哪个小组活干得好,就源源不断把新项目派过去。 小组 A 拿到越多项目,实战历练越多,技术越强;小组 B 长期没活,手生能力持续退化。调度员进一步把活全部塞给 A,其余小组摸鱼躺平。

这就是负载崩塌:你花钱养了 8 组人,实际只有 1 组在干活。

公司新增 KPI 约束:调度员考核 = 交付质量 + 人力负载均衡惩罚。哪怕小组 A 能力最强,也必须分配一部分任务给其余小组;如果任务分配差距过大,调度员直接扣分。 对应技术逻辑:总损失 = 主任务损失 + λ× 负载均衡损失,强制路由器兼顾效果与负载分配,打破马太正反馈循环。

公式

路由器路由概率:

(p_i = \\text{softmax}(W_g \\cdot x)_i) x:token 特征向量;(W_g):路由器可学习参数;(p_i):token 分配给第i号专家的概率。

总损失函数(核心)

(\\mathcal{L}{total} = \\mathcal{L}{task} + \\lambda \\cdot \\mathcal{L}{balance}) (\\mathcal{L}{task}):主任务交叉熵损失;(\\lambda):均衡损失权重超参;(\\mathcal{L}_{balance}):负载均衡辅助损失。

  • 重要性损失(Importance Loss)
  • (\\mathcal{L}{importance} = N \\cdot \\sum{i=1}^N \\bar p_i^2) N专家总数,(\\bar p_i)专家 i 平均路由概率;分布越不均匀损失数值越大;完全均匀时损失等于 1。

  • 负载损失(Load Loss)
  • (\\mathcal{L}{load} = N \\cdot \\sum{i=1}^N l_i^2) (l_i):专家 i 实际接收 token 占比,统计真实调度负载,而非概率期望。

  • 熵正则
  • (\\mathcal{L}{entropy} = -\\sum{i=

    赞(0)
    未经允许不得转载:171主机测评 » MoE 负载均衡深度解析:专家累死、摸鱼与负载崩塌
    分享到: 更多 (0)

    评论 抢沙发

    • 昵称 (必填)
    • 邮箱 (必填)
    • 网址