arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~

正确的二阶Muon:谱几何与曲率的原则性结合

Second-Order Muon Done Right: A Principled Marriage of Spectral Geometry and Curvature

Tong Che

arXiv 2608.09763首次发表:更新:

发表机构

NVIDIA Research(英伟达研究院)

机构由 AI 辅助整理,请以论文原文为准。

AI 中文总结

本文提出GO-MUON算法,将谱几何与曲率结合,通过复用匹配的依赖数据的几何实现精确更新,量化了softmax交叉熵中相关因子的接近情况,揭示了惰性几何是计算-统计的权衡。

AI 中文摘要

Muon的极性更新对于未加权谱几何是精确的。我们引入GO-MUON,其使用匹配的、依赖数据的几何并在多个优化步骤中复用该几何。在任意正定左右映射的条件下,其原始更新可精确求解对应的加权谱神谕;该结论与映射的估计方式或最近刷新的时间无关。对于softmax交叉熵,我们量化了观测标签反向因子接近模型Fisher和广义高斯-牛顿因子的情况。我们还表明,四步刷新几乎保留了缓慢变化几何的跟踪延迟,同时增加了平稳因子噪声,这使得惰性几何成为计算-统计的权衡,而非去噪机制。

英文摘要

Muon's polar update is exact for an unweighted spectral geometry. We introduce GO-MUON, which uses a matched data-dependent geometry and reuses it across several optimization steps. Conditioned on any positive-definite left and right maps, its raw update exactly solves the corresponding weighted spectral oracle; this statement is independent of how the maps are estimated or how recently they were refreshed. For softmax cross-entropy, we quantify when the observed-label backward factor approaches the model Fisher and generalized Gauss--Newton factor. We also show that four-step refresh nearly preserves the tracking delay of slowly changing geometry while increasing stationary factor noise, making lazy geometry a compute--statistics tradeoff rather than a denoising mechanism.

论文原文

arXiv 摘要页 · PDF 原文 · HTML 原文

↑