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

洛拉风格矩阵分解中穆伦的谱增长与收敛

Uniform Spectral Growth and Convergence of Muon in LoRA-Style Matrix Factorization

KAIST(韩国科学技术院)

Changmin Kang, Jihun Yun, Baekrok Shin, Yeseul Cho, Chulhee Yun

AI总结:

研究揭示了LoRA微调LLM时穆伦的谱特性,证明了奇异值等速率增长并收敛至全局极小值。

摘要:

谱梯度下降(SpecGD)正交化矩阵参数更新,并启发了实际优化器如穆伦。它们在大型语言模型(LLM)训练中表现良好,但其动态仍缺乏深入理解。在低秩适应(LoRA)设置中,权重更新被参数化为两个低秩因子的乘积,在LoRA微调LLM时,我们发现穆伦下出现了一种独特的谱现象:LoRA乘积的奇异值在谱上表现出近似均匀增长,尽管分别对两个因子进行了正交化。受此观察启发,我们分析了谱梯度流(SpecGF)——谱GD的连续时间类比——在简化LoRA风格矩阵分解设置中的情况,并证明了“等速率”动态:所有奇异值在小偏差内以相同速率增长。因此,较小的奇异值比较大的奇异值更早达到目标值,这与标准梯度流中观察到的最大值优先的阶梯式学习形成鲜明对比。此外,我们证明在我们的设置中,SpecGF从几乎所有初始值收敛到全局极小值,只要因子范数保持有界;加入ℓ2正则化,我们获得全局收敛。最后,我们通过相同设置的实验验证了我们的理论。

英文摘要:

Spectral gradient descent (SpecGD) orthogonalizes the matrix parameter updates and has inspired practical optimizers such as Muon. They often perform well in large language model (LLM) training, but their dynamics remain poorly understood. In the low-rank adaptation (LoRA) setting, where weight updates are parameterized as a product of two low-rank factors, we find a distinctive spectral phenomenon under Muon in LoRA fine-tuning of LLMs: singular values of the LoRA product show near-uniform growth across the spectrum, despite orthogonalization being performed on the two factors separately. Motivated by this observation, we analyze spectral gradient flow (SpecGF)-a continuous-time analogue of SpecGD-in a simplified LoRA-style matrix factorization setting and prove "equal-rate" dynamics: all singular values grow at equal rates up to small deviations. Consequently, smaller singular values attain their target values earlier than larger ones, sharply contrasting with the largest-first stepwise learning observed in standard gradient flow. Moreover, we prove that SpecGF in our setting converges to global minima from almost all initializations, provided the factor norms remain bounded; with $\ell_2$ regularization, we obtain global convergence. Lastly, we corroborate our theory with experiments in the same setting.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑