跳到正文
arXiv · cs.AI· Sifan Yang·· 13 天前AI 评分32

基于残差校正的内存高效子空间优化:从 Momentum SGD 到 Muon

Memory-Efficient Subspace Optimization with Residual Correction: From Momentum SGD to Muon

来源与原文

arXiv · cs.AI

作者:Sifan Yang

发布时间:(北京时间)

阅读原始来源 · arxiv.org ↗
完整链接与内容说明

https://arxiv.org/abs/2610.19844v1

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

一篇 arXiv 论文提出残差校正框架,在子空间中维护优化器状态并将正交残差梯度加入参数更新,以降低 LLM 训练的内存成本。该框架将 momentum SGD 扩展为 SMRC、将 Muon 扩展为 PMRC,二者均达到 O(T^{-1/4}) 收敛率,与全空间版本匹配;多阶段变体通过周期性重算基并引入边界校正保持同阶收敛率。实验验证了算法的有效性与效率。

来源:arXiv · cs.AI · arxiv.org