arXiv:2601.18420cs.LGcs.AI2026-01被引 1

让二阶优化更稳定高效,提升模型泛化能力

Gradient Regularized Natural Gradients

  • 将梯度正则化融入自然梯度,改进优化路径
  • 在视觉与语言任务上比SGD、AdamW等更快收敛且更优
  • 支持显式/隐式正则化,还可用贝叶斯方法免去逆矩阵计算

梯度正则化(GR)已被证明可提升模型泛化能力。尽管自然梯度下降在训练初期能加速优化,但对二阶优化器如何受益于GR的研究仍不足。本文提出梯度正则化自然梯度(GRNG),一类可扩展的二阶优化器,将显式梯度正则化与自然梯度更新结合。框架包含两种频数算法:显式自然梯度正则化(RENG),通过双反向传播显式最小化梯度范数;隐式自然梯度正则化(RING),将正则化隐式嵌入更新方向。还提出基于正则化-卡尔曼公式的贝叶斯变体,完全避免了FIM求逆。我们建立了GRNG的收敛性保证,表明梯度正则化提升了稳定性并实现全局极小值收敛。实验显示,相比一阶方法(SGD、AdamW)和二阶基线(K-FAC、Sophia),GRNG在视觉与语言基准上始终提升优化速度与泛化性能。

原文摘要 · Abstract (English)

Gradient regularization (GR) has been shown to improve the generalizability of trained models. While Natural Gradient Descent has been shown to accelerate optimization in the initial phase of training, little attention has been paid to how the training dynamics of second-order optimizers can benefit from GR. In this work, we propose Gradient-Regularized Natural Gradients (GRNG), a family of scalable second-order optimizers that integrate explicit gradient regularization with natural gradient updates. Our framework introduces two frequentist algorithms: Regularized Explicit Natural Gradient (RENG), which utilizes double backpropagation to explicitly minimize the gradient norm, and Regularized Implicit Natural Gradient (RING), which incorporates regularization implicitly into the update direction. We also propose a Bayesian variant based on a Regularized-Kalman formulation that eliminates the need for FIM inversion entirely. We establish convergence guarantees for GRNG, showing that gradient regularization improves stability and enables convergence to global minima. Empirically, we demonstrate that GRNG consistently enhances both optimization speed and generalization compared to first-order methods (SGD, AdamW) and second-order baselines (K-FAC, Sophia), with strong results on vision and language benchmarks.

优化器自然梯度正则化二阶方法

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