把 MoE 负载均衡看成 PID 控制器:ID Balancing 到底统一了谁
九月底挂出、十月九日被技术社区重新翻出来讨论的 ID Balancing(arXiv 2609.39137),干的事听上去很跨界:它把两种主流的 MoE 负载均衡方法,统一成一个控制论里的 PID 框架。作者指出,DeepSeek 那套广为人知的「无辅助损失」负载均衡,本质上是一个固定步长的积分控制器;而 Kimi K3 用的分位数均衡,本质上是一个广义的比例控制器。在这个视角下,它们都是不完整的 PID,于是论文提出一个真正带积分和微分项的 ID 控制器,去稳定极稀疏 MoE 的训练。 MoE 训练为什么会「专家吃不饱、有的撑死」? MoE 靠路由器把每个 token 分给几个专家,理想情况下每个专家被用到的频率差不多。但现实里路由器会「偏心」:少数几个专家越用越顺手,大部分专家常年冷门。冷门专家等于白训,模型有效容量被浪费;更糟的是训练会因此震荡、发散。传统解法是加一个辅助损失,强行惩罚不均衡——但这个损失的权重是个难调的超参,调重了模型被迫为了均衡而牺牲专业分工,调轻了又压不住。后来 DeepSeek 干脆去掉辅助损失,改成在路由器上加一个动态偏置,按历史负载微调;Kimi K3 则用分位数统计来校准路由分数。 PID 这个老框架,怎么就套到路由器头上了? 控制论里,比例(P)控制器按当前误差大小回拉,积分(I)控制器对累积误差做修正,微分(D)控制器对误差变化率做预判。作者的洞察是:Kimi K3 的分位数均衡只看「当前负载离目标多远」并据此调整路由分数,正是比例项;DeepSeek 的无损失法按每步固定量累积修正,不随误差大小变化,正是一个没有比例、没有微分的积分项。两者都缺了一角。ID Balancing 于是保留积分项(随负载误差缩放),再补上一个微分项——只在负载误差的变化率足够大时才激活,用来在路由突然偏斜的瞬间提前刹车,而不是等偏完了再补。 这种「换个视角看老方法」,真能带来新收益吗? 统一视角的价值,不只是好看。一旦把两种方法摆进同一个控制论坐标系,设计者就能像调工业控制器那样,系统性地考察积分饱和、微分噪声、稳定性裕度这些老问题。论文报告说这种组合在极稀疏 MoE 上训练更稳、不需要艰难调超参。它提醒的是:大模型里很多「新算法」其实是控制论、运筹学里成熟思想的转世,换个坐标系,老工程经验就能直接搬过来。对做 MoE 训练的团队,这种把调度问题翻译成控制问题的思路,比单个超参更值得收藏。 去论坛讨论 前往 silicon-agi.com 论坛 参与讨论