精读笔记

Problem Setting

论文标题：OptMuon: Closed-Loop Orthogonalized Momentum Methods for Stochastic Optimization with Zero-Noise Optimality（arXiv preprint / 2026）。

这篇论文关注的是矩阵参数随机非凸优化中的 Muon-style update：更新方向来自 momentum matrix 的 polar factor，而不是原始梯度或普通 momentum。问题表面上是证明一个新的 optimizer 收敛；实际核心是解决 orthogonalized momentum 的幅度校准问题。

Muon 类方法已经把“方向”从 raw gradient 的尺度中剥离出来：polar factor 抹掉奇异值，只保留某种矩阵方向几何。这带来稳定性，但也制造了新的矛盾：一旦方向被归一化，更新幅度就不能再自然由梯度范数决定，必须由外部 learning rate 或某种 scaling rule 接管。已有 Muon 理论通常使用固定或 open-loop magnitude rule，因此要么需要问题常数/训练 horizon 的隐式调参，要么无法表达低噪声时自动加速、无噪声时恢复 deterministic rate。

这篇论文真正处理的困难点是：如何在 polar-factor 这种强非线性、非坐标式方向变换之后，仍然设计一个闭环 scalar schedule，使其能同时控制 stochastic noise、tracking error 和 descent magnitude。关键矛盾是：Muon 要方向归一化以获得几何稳定性，noise-adaptive 方法要幅度随观测轨迹收缩；二者如果耦合不好，方向稳定性会被噪声污染，或者自适应幅度会被 orthogonalization 后的非线性放大。

Motivation

作者的动机不是再发明一个 Muon variant，而是补上 Muon 理论中一个相当具体的缺口：orthogonalized direction 已经有若干 convergence analysis，但 magnitude control 仍不够闭环。

已有路线的不足可以概括为两类。第一类是 Muon / normalized / orthogonalized update：它们改变几何，但幅度往往是固定 LR、horizon-dependent rule 或工程 schedule，本质上仍是 open-loop。这样很难证明同一套参数在 σ>0 和 σ=0 下同时自适应。第二类是 AdaGrad-Norm、STORM+、META-STORM 等 Lipschitz-free/noise-adaptive 方法：它们擅长从轨迹中估计合适尺度，但通常直接作用在 gradient estimator 上，没有处理 polar-factor direction 这种矩阵几何归一化。

作者的核心观察是：Muon 的稳定性主要来自方向几何，而 noise adaptivity 主要来自 scalar normalization；这两个机制理论上可以拆开。缺的是一种 scalar rule，既能和 Muon 的 polar factor 对齐，又能在 proof 中给出 self-normalized 控制，不依赖 L、σ、bounded-gradient constant 作为输入。

Core Idea

核心思想是把优化更新重新建模为“方向几何”和“幅度统计”的解耦系统。方向由 Orth(M_t) 给出，其中 M_t 是 momentum 或 STORM-type estimator；幅度则由一个轨迹依赖的 scalar coefficient 决定。换句话说，OptMuon 不让 raw gradient 同时决定方向、大小和噪声尺度，而是让 polar factor 负责几何，让 AdaGrad-Norm-like scalar 负责噪声/尺度适配。

这个建模变化的本质在于：Muon 的 orthogonalization 不再只是 optimizer 的一个工程 trick，而被放进一个 closed-loop adaptive framework 中。更新形式大致是 X_{t+1}=X_t-θ γ_t ||M_t||_F Orth(M_t)。其中 Orth(M_t) 给出归一化方向，γ_t ||M_t||_F 给出可控幅度。理论 proof 利用 polar factor 的 alignment property：<Orth(M),M>=||M||_* ≥ ||M||_F，从而得到 descent 主项；同时用 adaptive γ_t 把累积能量项压到 logarithmic 量级。

与 prior 的本质区别不是“用了 momentum”或“用了 AdaGrad”，而是 scalar adaptation 的闭环性被明确设计成可以和 orthogonalized direction 共存，并且目标是得到显式 zero-noise optimality split：deterministic term 与 stochastic term 分开，σ=0 时不需要 retuning。

Method

方法层面真正必要的机制只有几个。

第一，polar-factor momentum direction。它解决的是 Muon 中的几何归一化问题：不沿 raw momentum 的奇异值尺度走，而沿其 polar factor 走。必要性在于 descent proof 需要一个既有 bounded Frobenius norm 又与 M_t 对齐的方向。核心变化是把更新方向的尺度信息剥离出来，为 scalar schedule 留出控制空间。

第二，lagged adaptive coefficient α_t=ρ_{t-1}。它解决的是 adaptive coefficient 与当前 stochastic noise 的可测性问题。若 coefficient 直接依赖当前梯度，很多 martingale / conditional expectation 控制会变复杂甚至失效。滞后一拍是理论上干净的设计，不一定是实践中最优，但对 proof 很关键。

第三，running-maximum corrected AdaGrad-Norm coefficient。普通 AdaGrad-Norm denominator 会因累积梯度能量增长而收缩步长，但在出现孤立 spike 后可能过度塌缩。这里 numerator 放入历史最大梯度平方，并对 ρ_t 取 running minimum，目的是保留单调性，同时避免 spike 造成永久性过小 coefficient。这个 correction 更像是为了让 self-normalization 在 bounded-gradient 条件下给出更好的 pathwise 控制。

第四，两个 estimator regime。OptMuon-A 使用 M_t=G_t+(1-α_t)M_{t-1}，对应 average smoothness，证明得到 σ^{1/2}T^{-1/4} stochastic term。OptMuon-I 使用 same-sample STORM correction，M_t=(1-α_t)(M_{t-1}-∇f(X_{t-1};ξ_t))+G_t，对应 individual smoothness，换取 σ^{1/3}T^{-1/3}。这里的改进主要来自 STORM-type variance reduction，而不是 orthogonalization 本身。

Key Insight / Why It Works

这篇论文最关键的 insight 是：orthogonalization 和 noise adaptation 不必争夺同一个梯度尺度。Muon-style polar factor 可以被视为一种 direction-only operator，而 AdaGrad-Norm/STORM-style schedule 是 magnitude-only operator；只要 proof 能控制 estimator tracking error，两者可以组合而不互相破坏。

为什么能成立？下降分析中，polar factor 给了一个非常强的结构性不等式：<Orth(M_t),M_t>=||M_t||_*≥||M_t||_F。于是即便 Orth(M_t) 不是 gradient direction，只要 M_t 跟 ∇f(X_t) 的误差 S_t 可控，就能得到形如 γ_t||M_t||_F^2 的下降主项，误差项则是 γ_t||S_t||^2 和 smoothness 二阶项 γ_t^2||M_t||^2。整个问题被转化成：如何让 ∑γ_t||S_t||^2 和 ∑γ_t^2||M_t||^2 只增长 logarithmically。

closed-loop scalar schedule 的作用正是在这里。它不是简单调大学习率或加一个 normalization，而是让能量项自归一化：当 stochastic gradient 能量累积快时，γ_t 收缩；当噪声小或为零时，累积项不会强迫算法降到 stochastic-scale，从而保留 T^{-1/2} deterministic stationarity rate。running maximum correction 的价值主要在 pathwise 稳定性和避免 coefficient collapse；它可能是理论上便利且实践上合理的补丁，但是否是核心贡献，文中没有实证支持。

我认为核心贡献是“zero-noise adaptive rate split for orthogonalized momentum”，而不是某个具体 coefficient 形式。OptMuon-I 的 stochastic improvement 很大程度来自 STORM-type recursive estimator，这是已有 variance reduction 思想迁移到 Muon geometry；OptMuon-A 的价值则在说明即使没有 individual smoothness / same-sample correction，也能得到低噪声自适应分解。

哪些部分可能只是辅助？γ_t 中的 min 结构、具体 q=1/2 或 2/3、running maximum numerator、lagged convention 很大程度服务于 proof。它们可能不是实践最优设计。尤其 q 的选择与最终 rate 的 exponent 对齐，理论味道很重。若放到真实 LLM training，增益来源可能被 global LR schedule、batch size scaling、normalization layer、weight decay 完全掩盖。

从归因看，这不是 data/retrieval/curriculum 类贡献，而是 optimizer scaling 与 inductive bias 的组合：polar factor 提供 matrix geometry inductive bias，closed-loop coefficient 提供 noise-aware scaling。真正值得迁移的是“direction operator 与 magnitude controller 解耦”的分析范式。

Relation To Prior Work

这篇属于三条谱系的交叉：Muon / orthogonalized momentum，AdaGrad-Norm / Lipschitz-free adaptive stepsize，STORM-type recursive variance reduction。

与 MuonNS、LiMuon、Muon-MVR 等相比，OptMuon 的新增信息不是 polar factor 本身，而是 magnitude rule 从 open-loop 变成 closed-loop，并且证明了 σ-dependent rate split。Muon-MVR 类工作已经说明 variance reduction 可以和 Muon 结合，但通常没有把 scalar coefficient 设计成 trajectory-dependent、Lipschitz-free、zero-noise adaptive 的形式。

与 AdaGO、AdaMuon、NAMO 等 adaptive Muon 变体相比，OptMuon 更理论化。AdaGO/AdaMuon/NAMO 也在做 scaling，但本文强调的是 lagged monotone self-normalized scalar schedule 和显式 CLZO recovery。看似新的地方其实大量借用了 AdaGrad-Norm 与 STORM 的 proof 技术；实质创新在于把这些 self-normalized 控制移植到 polar-factor momentum，并处理由 Orth(M_t) 引入的 alignment/error decomposition。

与 STORM+ / META-STORM 相比，OptMuon-I 的 variance-reduced estimator 并不新；不同点是 estimator 不直接作为 update vector，而先经过 Muon-style orthogonalization。换言之，本文不是提出新的 variance reduction，而是证明 variance reduction 的 tracking-error machinery 可以服务于 orthogonalized direction。

因此它更像是 optimization theory 中的组合式推进：把已有 closed-loop scalar adaptation 与 Muon geometry 进行理论缝合，并提炼出 zero-noise optimality 这个评价维度。

Dataset / Evaluation

本文没有真正的 dataset / empirical evaluation。它是 math.OC 取向的理论 preprint，claim 主要由收敛定理支撑，而不是由神经网络训练实验支撑。

从验证角度看，论文证明了在假设满足时的 expected stationarity bound，因此支持的是“理论上可以将 closed-loop scalar adaptation 与 exact polar-factor momentum 结合，并保留 noise adaptivity / zero-noise recovery”。它没有验证以下更实践的 claim：在 LLM 或大规模深度网络中是否比 Muon、AdamW、AdaMuon、NAMO 更稳定；running-maximum correction 是否真的避免 spike 后性能下降；Newton–Schulz 近似下 rate 是否保持；blockwise layer update 是否符合理论假设。

因此 evaluation 对核心理论 claim 是足够的，但对 optimizer paper 通常关心的实际训练收益完全不充分。若作者暗示大规模 deep learning 稳定性，这部分文中未充分说明。实际增益来源不清，可能主要来自 scaling rule 的保守性，而非 orthogonalization-adaptation 的协同。

Limitation

第一，bounded stochastic gradients 是很强的前提。论文也承认没有 bounded-gradient-free theory。真实深度学习中梯度范数、per-layer scale、mixed precision overflow、normalization dynamics 都可能破坏这个假设。该假设不只是技术细节，因为 running normalizer、tracking error 和 martingale control 都依赖 pathwise boundedness。

第二，exact polar-factor orthogonalization 与实践 Muon 的 Newton–Schulz approximation 有明显鸿沟。有限步 Newton–Schulz 会引入方向误差、spectral norm 偏差和 alignment 损失；这些误差是否只是 additive term，还是会被 adaptive scalar 放大，文中未充分说明。

第三，rate 中常数依赖 L、维度 n、θ、G_hat 和初始 gap。虽然算法参数不使用这些量，但理论上不是 scale-free。尤其 polar factor 的 Frobenius bound 带 n，矩阵宽度/层维度在大模型中可能让常数不可忽略。

第四，OptMuon-I 的更好 stochastic term 依赖 individual smoothness 和 same-sample 两点梯度，这在实际 stochastic data augmentation、dropout、large-scale data pipeline 中不一定自然成立。它的优势可能更多是 STORM 经典条件下的理论优势，而非 Muon-specific。

第五，论文没有实验证据，因此无法判断 running maximum correction 是关键 insight 还是 proof artifact。也无法判断与已有 adaptive Muon 方法相比是否有实际收益。增益归因不清：可能主要来自更强的 scalar damping，而不是更好的 geometry。

第六，算法把问题从“如何调学习率”转移成“如何设计并维护合适的 global scalar normalizer”。在多层、异质 block、不同 parameter groups 的真实训练中，一个 scalar 或 layerwise scalar 是否足够表达局部 curvature/noise 差异，文中未说明。

Takeaway

• 1. 最值得记住的是 direction-magnitude decoupling：Muon 的 polar factor 可以只作为方向 operator，幅度完全交给 closed-loop scalar controller。
• 这种拆分比“给 Muon 加自适应学习率”更干净。
• 2. Zero-noise recovery 是评价 adaptive stochastic optimizer 的一个重要维度。
• 很多方法只给 O(T^{-1/4}) 或 O(T^{-1/3}) stochastic rate，但在 σ=0 时没有自动恢复 deterministic T^{-1/2}；本文把这个标准带到 orthogonalized momentum。

一句话总结

OptMuon 是把 AdaGrad-Norm/STORM 式闭环幅度自适应嫁接到 Muon polar-factor momentum 上的理论框架，真正贡献是证明 orthogonalized updates 也能获得 noise-adaptive 与 zero-noise optimal 的 rate split。