MuonGS: Pretraining LLMs with Relaxed Weight-Norm Control
MuonGS:放宽权重范数控制的LLM预训练优化器
University of Minnesota
用Gram-Schmidt移除Muon更新径向分量,权重范数最多√t增长,长期困惑度更低并加速LLM预训练约15%
📌 工作参考价值(4 点)
- 训练电商/广告域的大模型(CTR、NLP、多模态等)时,可尝试用 MuonGS 替代 AdamW 或 MuonH,尤其在长训练 horizon 下:MuonGS 不固定权重范数,省去严格范数约束的超参调优,同时保持后期有效学习率,可能降低最终损失。
- 把累积有效学习率 C_T = Ση_t/||W_t||_F 作为监控指标:如果训练后期 C_T 增长明显放缓(对数增长),说明有效学习率相对权重范数过小,可考虑放松范数控制、调整学习率或改用 MuonGS 类方法。
- 对推荐模型中的矩阵参数,Muon 类优化器可直接作用于 2D 权重;但 embedding 表通常是高维稀疏结构,需先验证 Gram-Schmidt 投影在混合参数上的工程开销,建议小规模预训练对比后再推广。
- 结论上,权重范数不必严格固定,允许其以 √t 增长可维持长期有效学习率,这一原则可用于指导自定义优化器或学习率调度。