跳到正文
arXiv cs.CL· Esther Xin·· 10 小时前AI 评分31

先调好基线:Muon 缩放权重衰减的大部分收益来自调参差距,但其小幅真实优势随模型规模增长

Tune the Baseline First: Most of Muon's Scaled Weight Decay Gain Is a Tuning Gap, but Its Small Real Advantage Grows With Model Size

AI 导读

研究在 MAI-Base-2 协议下重新对比 Muon-SW 与恒定解耦权重衰减,发现共享超参时 Muon-SW 领先 0.035 BPB(1.52x 算力节省),双方各自调参后差距仅 0.0026 BPB(1.03x),约 90% 优势消失。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:据报道,Muon 的缩放权重衰减(Muon-SW)在训练语言模型时比恒定解耦权重衰减更高效,但此类比较通常对两种方法复用同一套学习率(LR)和权重衰减(WD)设置。我们在 MAI-Base-2 协议下重新进行了比较:对每种方法,通过其乘积联合调优 LR 和 WD,在源缩放的专家混合(MoE)缩放阶梯(1.9M 至 15M 活跃参数)上进行,该阶梯在各规模间保持每个活跃参数的 token 数和数据重复次数固定。我们发现:(1)大部分增益来自调优差距。在共享设置下,Muon-SW 以每字节 0.035 比特(BPB)胜出,相当于 1.52 倍计算节省;在两种方法均调优后,差距为 0.0026 BPB(95% CI [0.0018, 0.0034]),相当于 1.03 倍节省,因此约 90% 的优势消失。(2)剩余的小优势是真实的:在最小规模下,四个配对种子在调优设置下均支持 Muon-SW。(3)该优势随模型规模增长:在 8 倍的活跃参数范围内,调优后的差距从 0.0036 升至 0.0051 BPB,而未调优的差距从 0.035 缩小至 0.024,因此存活份额从约 10% 升至 21%。(4)最佳权重衰减随规模变化:两种方法的最优 LRWD 乘积均增长,Muon-SW 增长更快(比率 1.40 至 1.91),而在两个最小规模上拟合的规律对第三个规模高估 1.3-1.5 倍,代价高达 0.002 BPB。整个研究在单块 NVIDIA T4 GPU 上运行;我们发布了代码和所有运行结果。
评论: 9 页,8 图,代码和数据:此 https URL
主题: 计算与语言(cs.CL);机器学习(cs.LG)
ACM 分类: I.2.6;I.2.7
引用为: arXiv:2611.02529 [cs.CL]
  (或 arXiv:2611.02529v1 [cs.CL] 用于此版本)
  https://doi.org/10.48550/arXiv.2611.02529

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Esther Xin [查看电子邮件]
[v1] 2026 年 11 月 2 日星期一 04:59:08 UTC(311 KB)

来源:arXiv cs.CL · arxiv.org