arXiv:2504.20096cs.LGmath.OC2025-04被引 3

提出AdaFisher优化器,用近似费舍尔信息矩阵提升训练稳定性与速度。

Towards Practical Second-Order Optimizers in Deep Learning: Insights from Fisher Information Analysis

  • 用对角块克罗内克分解逼近费舍尔信息矩阵,自适应调整梯度。
  • 在图像分类和语言建模上比SOTA优化器更快收敛且更稳定。
  • 适合追求高鲁棒性与快速收敛的深度学习任务开发者使用。

一阶优化方法仍是训练深度神经网络的标准。如Adam等优化器通过使用对角矩阵预处理随机梯度,引入有限的曲率信息。尽管一阶方法广泛采用,二阶优化算法通常表现出优于Adam和SGD的收敛性。然而,其在训练DNN时的实际应用受限于相比一阶方法显著更高的每轮计算成本。本文提出AdaFisher,一种新型自适应二阶优化器,利用费舍尔信息矩阵的对角块-克罗内克近似来自适应地预处理梯度。该方法旨在弥合二阶方法在收敛性和泛化性能上的优势与训练DNN所需的计算效率之间的差距。尽管传统上二阶优化器速度较慢,AdaFisher在图像分类和语言建模任务中仍表现出色,具有显著的稳定性与鲁棒性。实验表明,其在准确率和收敛速度上均超越现有最优优化器。代码已公开于 https://github.com/AtlasAnalyticsLab/AdaFisher。

原文摘要 · Abstract (English)

First-order optimization methods remain the standard for training deep neural networks (DNNs). Optimizers like Adam incorporate limited curvature information by preconditioning the stochastic gradient with a diagonal matrix. Despite the widespread adoption of first-order methods, second-order optimization algorithms often exhibit superior convergence compared to methods like Adam and SGD. However, their practicality in training DNNs is still limited by a significantly higher per-iteration computational cost compared to first-order methods. In this thesis, we present AdaFisher, a novel adaptive second-order optimizer that leverages a diagonal block-Kronecker approximation of the Fisher information matrix to adaptively precondition gradients. AdaFisher aims to bridge the gap between the improved convergence and generalization of second-order methods and the computational efficiency needed for training DNNs. Despite the traditionally slower speed of second-order optimizers, AdaFisher is effective for tasks such as image classification and language modeling, exhibiting remarkable stability and robustness during hyperparameter tuning. We demonstrate that AdaFisher outperforms state-of-the-art optimizers in both accuracy and convergence speed. The code is available from https://github.com/AtlasAnalyticsLab/AdaFisher.

二阶优化自适应优化深度学习费舍尔信息

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