arXiv:2508.13898cs.LGcs.AI2025-08AAAI被引 5

提出FOP方法,让大批次训练时的二阶优化更有效。

Beyond the Mean: Fisher-Orthogonal Projection for Natural Gradient Descent in Large Batch Training

  • 用两个子批次梯度差构建正交方向,增强更新信息
  • 在超大批次下实现更快收敛与更好泛化性能
  • 适合需要高效大批次训练的深度学习场景

现代GPU具备大量高带宽内存,支持数十万样本的大批次训练。然而,现有优化器在大批次下表现不佳。随着批次增大,梯度噪声降低,一阶方法难以逃离尖锐或次优极小值,难以达到全局最优。而二阶方法如基于克罗内克积近似曲率(KFAC)的自然梯度,需施加过高阻尼以保持稳定,导致曲率信息被削弱,性能退化为普通梯度下降。本文提出费舍尔正交投影(Fisher-Orthogonal Projection, FOP),通过利用两个子批次的梯度,构造一个在费舍尔度量下与平均梯度正交的差异分量,形成具有方差感知的更新方向。该方法在超大批次下恢复了二阶方法的有效性,实现可扩展训练,提升泛化能力并加快收敛速度。

原文摘要 · Abstract (English)

Modern GPUs are equipped with large amounts of high-bandwidth memory, enabling them to support mini-batch sizes of up to tens of thousands of training samples. However, most existing optimizers struggle to perform effectively at such a large batch size. As batch size increases, gradient noise decreases due to averaging over many samples, limiting the ability of first-order methods to escape sharp or suboptimal minima and reach the global minimum. Meanwhile, second-order methods like the natural gradient with Kronecker-Factored Approximate Curvature (KFAC) often require excessively high damping to remain stable at large batch sizes. This high damping effectively washes out the curvature information that gives these methods their advantage, reducing their performance to that of simple gradient descent. In this paper, we introduce Fisher-Orthogonal Projection (FOP), a novel technique that restores the effectiveness of the second-order method at very large batch sizes, enabling scalable training with improved generalization and faster convergence. FOP constructs a variance-aware update direction by leveraging gradients from two sub-batches, enhancing the average gradient with a component of the gradient difference that is orthogonal to the average under the Fisher-metric.

优化算法大批次训练二阶优化

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