Why Muon Outperforms Adam: A Curvature Perspective
为什么 Muon 优于 Adam:曲率视角
🧭 优化器比较 · 曲率与 NDS
National University of Singapore · Yale University · University of Minnesota
Muon 因更低的标准化方向锐度 (NDS) 实现更小的二阶曲率惩罚,从而获得更大的一步损失下降。
📌 工作参考价值(4 点)
- 在大规模推荐或 Agent 模型训练中,可尝试用 Muon 替代 Adam,尤其在数据长尾分布下,Muon 可能带来更快的收敛和更好的泛化。
- 关注 NDS 作为优化器选择的度量:优先选用在训练后段 NDS 更低的优化器,这可能意味着更平滑的收敛和更小的曲率惩罚。
- Muon 的频谱归一化思想可迁移到其他自适应优化器中:通过显式约束更新矩阵的奇异值能量分布,避免梯度对齐高曲率方向导致的优化震荡。
- 数据不平衡(如电商用户行为幂率分布)会放大 Muon 的优势,可据此设计自动优化器选取策略:当数据偏度超过阈值时切换至 Muon 类优化器。