Nora优化器通过正交对齐实现高效稳定的大模型训练。
Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizer
- 行级动量投影到权重正交补空间,稳定参数幅值与方向。
- 近似结构化预条件,计算复杂度仅O(mn),媲美高效方法。
- 代码仅需两行,适合大规模语言模型训练场景。
基于矩阵的优化器在大语言模型训练中展现出巨大潜力,但设计理想优化器仍具挑战。优秀优化器需满足效率、稳定性与速度三要素:实现类Muon的预条件加速优化,严格保持神经网络固有的尺度不变性,同时最小化计算开销。现有方法难以兼顾三者,或如Muon计算成本过高,或如RMNP存在径向抖动影响稳定性。为此,我们提出Nora,一种严格满足三项要求的优化器。Nora通过将行级动量投影至权重正交补空间,显式稳定权重范数与角速度,保障训练稳定性;利用Transformer海森矩阵的块对角主导特性,有效近似结构化预条件,维持最优计算复杂度O(mn)。此外,我们证明Nora具备可扩展性,并建立相应缩放定理。简化实现仅需两行代码,初步实验验证其在大规模训练中的高效性与前景。
原文摘要 · Abstract (English)
Matrix-based optimizers have demonstrated immense potential in training Large Language Models (LLMs), however, designing an ideal optimizer remains a formidable challenge. A superior optimizer must satisfy three core desiderata: efficiency, achieving Muon-like preconditioning to accelerate optimization; stability, strictly adhering to the scale-invariance inherent in neural networks; and speed, minimizing computational overhead. While existing methods address these aspects to varying degrees, they often fail to unify them, either incurring prohibitive computational costs like Muon, or allowing radial jitters that compromise stability like RMNP. To bridge this gap, we propose Nora, an optimizer that rigorously satisfies all three requirements. Nora achieves training stability by explicitly stabilizing weight norms and angular velocities through row-wise momentum projection onto the orthogonal complement of the weights. Simultaneously, by leveraging the block-diagonal dominance of the Transformer Hessian, Nora effectively approximates structured preconditioning while maintaining an optimal computational complexity of $\mathcal{O}(mn)$. Furthermore, we prove that Nora is a scalable optimizer and establish its corresponding scaling theorems. With a streamlined implementation requiring only two lines of code, our preliminary experiments validate Nora as an efficient and highly promising optimizer for large-scale training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。