arXiv:2602.09006cs.LGcs.AI2026-02被引 5

提出新型矩阵优化框架ARO,通过自适应旋转梯度提升大模型训练效率。

ARO: A New Lens On Matrix Optimization For Large Models

  • 将梯度旋转作为核心设计,基于新范式进行归一化下降。
  • 在80亿参数模型上提速1.3至1.35倍,比AdamW和正交化方法更优。
  • 适合追求高效训练的大模型研究者,可拓展跨层耦合设计。

基于矩阵的优化器因能提升大语言模型(LLM)训练效率而受到越来越多关注,现有进展主要集中在正交化/白化方法。尽管性能显著提升,但一个根本问题浮现:能否超越正交化,开辟新范式以进一步推动效率边界?本文提出自适应旋转优化(Adaptively Rotated Optimization, ARO),将梯度旋转视为首要设计原则。ARO通过在由新型范数感知策略决定的旋转坐标系中执行归一化最速下降,实现更新规则,超越现有正交化与白化优化器,实际提升了样本效率。为确保比较可靠性,我们设计了严格受控的基准测试协议,减少混杂因素与偏差。在此协议下,ARO在高达80亿激活参数的LLM预训练中,持续优于AdamW(1.3~1.35倍)和正交化方法(1.1~1.15倍),并可在超8倍训练预算下保持优势,未见收益递减。最后,我们揭示ARO可被重构成基于残差流旋转对称性的对称感知优化器,为高效利用跨层/跨模块耦合提供计算友好的先进设计基础。

原文摘要 · Abstract (English)

Matrix-based optimizers have attracted growing interest for improving LLM training efficiency, with significant progress centered on orthogonalization/whitening based methods. While yielding substantial performance gains, a fundamental question arises: can we develop new paradigms beyond orthogonalization, pushing the efficiency frontier further? We present \textbf{Adaptively Rotated Optimization (ARO}, a new matrix optimization framework that treats gradient rotation as a first class design principle. ARO accelerates LLM training by performing normed steepest descent in a rotated coordinate system, where the rotation is determined by a novel norm-informed policy. This perspective yields update rules that go beyond existing orthogonalization and whitening optimizers, improving sample efficiency in practice. To make comparisons reliable, we propose a rigorously controlled benchmarking protocol that reduces confounding and bias. Under this protocol, ARO consistently outperforms AdamW (by 1.3 $\sim$1.35$\times$) and orthogonalization methods (by 1.1$\sim$1.15$\times$) in LLM pretraining at up to 8B activated parameters, and up to $8\times$ overtrain budget, without evidence of diminishing returns. Finally, we discuss how ARO can be reformulated as a symmetry-aware optimizer grounded in rotational symmetries of residual streams, motivating advanced designs that enable computationally efficient exploitation of cross-layer/cross module couplings.

矩阵优化大模型训练自适应旋转梯度下降

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。