跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

Muon 何时以及如何发挥预条件作用?结构化梯度噪声的角色

文章摘要

该论文从理论与实验两方面研究梯度噪声几何结构如何决定 Muon 优化器的预条件收益,指出在传统各向同性噪声假设下 Muon 只是等效于学习率缩放,而任务频率不均导致的结构化噪声才是其真正优势所在。

来源arXiv Machine Learning作者: Xiaoran Chen, Binghui Li, Liwei Wang, Pinyan Lu
Muon 何时以及如何发挥预条件作用?结构化梯度噪声的角色
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

Muon 近年来已成为大语言模型训练中颇受欢迎的优化器,但其理论基础仍缺乏充分解释。由 Xiaoran Chen 等四位作者完成的 arXiv 论文《When and How Does Muon Precondition? The Role of Structured Gradient Noise》(arXiv:2610.31728,机器学习 cs.LG 分类,2026 年 10 月 22 日提交)正是围绕这一空白展开,核心问题是:梯度噪声的几何结构如何决定 Muon 预条件带来的收益。

论文首先给出一个带有否定意味的结论。在经典各向同性噪声假设下,作者从理论和实验两方面表明,Muon 除了相当于对学习率做一次重新缩放外,并不提供额外的加速效果。这一结果说明,以往在均匀噪声框架下讨论 Muon 的优越性并不充分,需要寻找更贴近真实训练过程的噪声模型。

为此,作者转向结构化噪声:在任务频率分布不均的联想记忆任务中,小批量采样会带来具有特定结构的梯度噪声。针对两个任务的情形,论文给出了精确的平衡阈值,并将分析推广到多任务场景,揭示任务覆盖度与梯度可靠性如何共同塑造学习过程。其中一项关键发现是,Muon 能把对任务频率的依赖从线性削弱为平方根级别,从而帮助频繁任务与稀有任务得到更均衡的学习。

论文还识别出 SignSGD 中由旋转引起的负迁移现象,而 Muon 能够避免这一问题;同时解释了动量如何一方面降低采样噪声,另一方面跟踪不断变化的梯度。作者通过受控数值实验验证了上述机制,并用一项初步的大语言模型预训练研究展示了 fast-slow momentum 的实际潜力。该预印本共 62 页、11 幅图,并提供音频摘要等附加功能。

总体而言,这项工作并未笼统宣称 Muon 必然优于其他优化器,而是界定了其预条件收益成立的条件:收益依赖于结构化、非各向同性的梯度噪声,以及任务频率异质等具体设定。这一结论为理解 Muon 的理论基础提供了更清晰的边界,也为后续优化器设计和大模型训练实践提供了可检验的方向。

展开要点与分析

文章情报

工程师进阶

要点

  • 在各向同性噪声假设下 Muon 除学习率缩放外没有加速效果
  • 结构化噪声下 Muon 可将任务频率依赖从线性降为平方根
  • Muon 避免了 SignSGD 中旋转引起的负迁移
  • 动量在降低采样噪声的同时跟踪变化的梯度,快慢动量在初步 LLM 预训练中展现潜力

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。