让优化器在低谷中继续探索,找更优的模型参数
Exploring Landscapes for Better Minima along Valleys
- 在损失值相近的山谷区域持续搜索,不轻易停在局部最优
- 在大批次训练中使现有最优算法准确率提升2.5%
- 适用于追求更好泛化能力的深度学习训练场景
寻找更低且泛化能力更强的极小值是深度学习的关键。然而,现有优化器一旦到达局部极小值便停止搜索。由于损失曲面具有复杂的几何特性,难以保证该点为全局最低或泛化最佳。为此,我们提出一种梯度优化器的适配器「E」,使优化器在抵达局部极小值后仍能沿损失值相近的山谷区域继续探索,以寻找潜在的更优局部极小值。该方法提高了找到更低、更平坦极小值的概率,而平坦极小值通常与更好的泛化性能相关。我们还从理论上证明了该适配器在凸与非凸情形下的收敛性。实验表明,在大批次训练这一极具挑战性的场景中,基于Lamb的改进版本ALTO,在多种任务上平均提升了2.5%的测试准确率,超越当前最优基准。本工作或开启优化算法设计的新方向。
原文摘要 · Abstract (English)
Finding lower and better-generalizing minima is crucial for deep learning. However, most existing optimizers stop searching the parameter space once they reach a local minimum. Given the complex geometric properties of the loss landscape, it is difficult to guarantee that such a point is the lowest or provides the best generalization. To address this, we propose an adaptor "E" for gradient-based optimizers. The adapted optimizer tends to continue exploring along landscape valleys (areas with low and nearly identical losses) in order to search for potentially better local minima even after reaching a local minimum. This approach increases the likelihood of finding a lower and flatter local minimum, which is often associated with better generalization. We also provide a proof of convergence for the adapted optimizers in both convex and non-convex scenarios for completeness. Finally, we demonstrate their effectiveness in an important but notoriously difficult training scenario, large-batch training, where Lamb is the benchmark optimizer. Our testing results show that the adapted Lamb, ALTO, increases the test accuracy (generalization) of the current state-of-the-art optimizer by an average of 2.5% across a variety of large-batch training tasks. This work potentially opens a new research direction in the design of optimization algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。