用物理启发的梯度方法,让神经网络更高效避开训练中的平坦区和鞍点。
Dimer-Enhanced Optimization: A First-Order Approach to Escaping Saddle Points in Neural Network Training
- 通过双点探测局部曲率,仅用梯度信息近似海森矩阵最小特征向量。
- 在变压器小模型上表现优于传统一阶优化器,提升复杂损失面导航能力。
- 适合追求训练稳定性与收敛速度的深度学习研究者使用。
一阶优化方法如SGD和Adam因计算效率高而广泛用于大规模深度神经网络训练。然而,仅依赖梯度信息的方法在面对平坦区域、平台区和鞍点时表现不佳。二阶方法虽可利用海森矩阵的曲率信息解决此问题,但对大模型计算成本过高。本文受分子动力学中‘二聚体’(Dimer)方法启发,提出一种名为Dimer-Enhanced Optimization(DEO)的新框架,该方法通过构造两个邻近点来探测损失函数的局部几何结构,仅用梯度信息即可高效估计曲率。DEO将二聚体法扩展至更广的损失空间,无需计算完整海森矩阵即可逼近其最小特征向量。通过周期性地将梯度投影到最小曲率方向的正交子空间,引导优化器脱离鞍点和平坦区域,实现非步进式更新,显著提升训练效率。初步实验在变压器小模型上验证了其性能,结果表明其在复杂损失景观中的导航能力优于标准一阶方法。本工作将物理启发的一阶曲率估计方法成功应用于高维神经网络训练。
原文摘要 · Abstract (English)
First-order optimization methods, such as SGD and Adam, are widely used for training large-scale deep neural networks due to their computational efficiency and robust performance. However, relying solely on gradient information, these methods often struggle to navigate complex loss landscapes with flat regions, plateaus, and saddle points. Second-order methods, which use curvature information from the Hessian matrix, can address these challenges but are computationally infeasible for large models. The Dimer method, a first-order technique that constructs two closely spaced points to probe the local geometry of a potential energy surface, efficiently estimates curvature using only gradient information. Inspired by its use in molecular dynamics simulations for locating saddle points, we propose Dimer-Enhanced Optimization (DEO), a novel framework to escape saddle points in neural network training. DEO adapts the Dimer method to explore a broader region of the loss landscape, approximating the Hessian's smallest eigenvector without computing the full matrix. By periodically projecting the gradient onto the subspace orthogonal to the minimum curvature direction, DEO guides the optimizer away from saddle points and flat regions, enhancing training efficiency with non-stepwise updates. Preliminary experiments on a Transformer toy model show DEO achieves competitive performance compared to standard first-order methods, improving navigation of complex loss landscapes. Our work repurposes physics-inspired, first-order curvature estimation to enhance neural network training in high-dimensional spaces.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。