Your LR Schedule May Not Be Principled: Effective Learning Rate Scheduling in Weight-Normalized Training
你的学习率调度可能并不合理:权重归一化训练中的有效学习率调度
🧭 大模型预训练优化 · Effective LR
ByteDance Seed · Peking University
证明 WN 训练增益主要来自有效学习率轨迹,提出 AIR 调度,在 128B MoE 预训练上比 cosine 降损约 0.01
📌 工作参考价值(4 点)
- 若你在 LLM4Rec / 生成式推荐模型预训练中采用 weight normalization 或固定范数约束,应把 LR schedule 设计对象从 nominal LR 切换到 effective LR,否则会错失 WN 的降损收益。
- 工程上可直接用 AIR 替代 cosine:用 affine 修正 1/√t 衰减以匹配预设 peak/terminal LR,在 128B MoE 实验中比 cosine 的 terminal training loss 低约 0.01,且超参鲁棒性更好,适合线上预训练 pipeline 快速替换。
- 在优化器或训练约束切换(如 WN ↔ WD)时,保持相同 ELR 轨迹可复现训练动态,便于做 infra 对齐和消融实验,减少重新调参成本。
- 优先优化 late-stage LR schedule:早期 LR 选择对最终损失影响较弱,晚期 ELR 调度是主要收益来源;调参资源应集中在后期 schedule 上。