TAG

#MoE

共 6 篇文章

MoE 专家容量因子与 Token 丢弃:容量溢出如何影响训练质量与推理一致性

训练稀疏 MoE 时,专家容量因子决定每个专家能接收的 token 上限,超出的 token 会被丢弃或经残差绕过。本文以多语言翻译训练为贯穿场景,解释容量因子、负载均衡损失与丢弃率的耦合关系,对比 GShard 与 Switch Transformer 的容量设计,并讨论丢弃对长尾 token 学习、推理期行为差异和可观测指标的影响。

MoE 路由的 z-loss 与 logit 钳制:路由器 logits 爆炸如何引发专家塌缩

稀疏 MoE 训练常在数千步后突然发散,根源往往不是任务本身,而是路由器 logits 数值膨胀导致 softmax 饱和、梯度消失与专家塌缩。文章以在线多语言训练服务为贯穿场景,解释 z-loss 如何惩罚 logits、logit 钳制与 float32 路由器精度各自修正了什么,并对比仅用负载均衡损失的方案。