ROOT优化器提升大模型训练稳定性,解决梯度噪声与维度敏感问题。
ROOT: Robust Orthogonalized Optimizer for Neural Network Training
- 采用自适应牛顿迭代实现跨维度的稳定正交化
- 在噪声环境和非凸场景中收敛更快、性能更优
- 适合对训练稳定性要求高的大规模模型训练
大型语言模型(LLMs)的优化仍是关键挑战,模型规模扩大加剧了对算法精度和训练不稳定的敏感性。现有优化器通过动量正交化提升了收敛效率,但存在两个主要鲁棒性缺陷:正交化精度受维度影响显著,且易受异常值噪声干扰。为此,我们提出ROOT(Robust Orthogonalized Optimizer),通过双重鲁棒机制增强训练稳定性。首先,设计基于自适应牛顿迭代的维度鲁棒正交化方案,使用细粒度系数适配不同矩阵尺寸,确保各类架构下的精度一致性;其次,引入近端优化框架抑制异常值噪声,同时保留有效梯度方向。大量实验表明,相较于Muon和Adam基线,ROOT在噪声环境与非凸场景中均实现更快收敛和更优最终性能。本工作为构建可应对现代大规模模型训练复杂性的鲁棒高效优化器提供了新范式。代码将发布于 https://github.com/huawei-noah/noah-research/tree/master/ROOT。
原文摘要 · Abstract (English)
The optimization of large language models (LLMs) remains a critical challenge, particularly as model scaling exacerbates sensitivity to algorithmic imprecision and training instability. Recent advances in optimizers have improved convergence efficiency through momentum orthogonalization, but suffer from two key robustness limitations: dimensional fragility in orthogonalization precision and vulnerability to outlier-induced noise. To address these robustness challenges, we introduce ROOT, a Robust Orthogonalized Optimizer that enhances training stability through dual robustness mechanisms. First, we develop a dimension-robust orthogonalization scheme using adaptive Newton iterations with fine-grained coefficients tailored to specific matrix sizes, ensuring consistent precision across diverse architectural configurations. Second, we introduce an optimization-robust framework via proximal optimization that suppresses outlier noise while preserving meaningful gradient directions. Extensive experiments demonstrate that ROOT achieves significantly improved robustness, with faster convergence and superior final performance compared to both Muon and Adam-based optimizers, particularly in noisy and non-convex scenarios. Our work establishes a new paradigm for developing robust and precise optimizers capable of handling the complexities of modern large-scale model training. The code will be available at https://github.com/huawei-noah/noah-research/tree/master/ROOT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。