arXiv:2510.25366cs.LGcs.AI2025-10

根据损失函数凸性变化,分阶段切换优化算法提升训练效率

A Convexity-dependent Two-Phase Training Algorithm for Deep Neural Networks

  • 先用Adam处理非凸区域,再切换至CG求解凸区域
  • 实验验证该策略显著加速收敛并提高模型精度
  • 适合追求高效训练的深度学习研究者与工程师

机器学习的核心任务是最小化衡量模型拟合程度的损失函数。高效的数值优化方法依赖于损失函数的性质,其中最关键的是其凸性或非凸性。尽管损失函数常具有非凸区域,导致普遍采用Adam等非凸优化方法,但局部最优解附近必然存在函数呈现凸性的区域。在此区域内,共轭梯度(CG)等二阶优化方法可保证超线性收敛。本文提出一种新框架,假设真实任务中损失函数从初始非凸逐渐过渡到接近最优时的凸性。基于此,设计了一种两阶段优化算法,通过观察梯度范数随损失的变化来检测凸性转变点:非凸阶段使用Adam,凸阶段启用CG。计算实验验证了该简单凸性结构在实际中足够普遍,足以通过此策略显著提升收敛速度与模型准确率。

原文摘要 · Abstract (English)

The key task of machine learning is to minimize the loss function that measures the model fit to the training data. The numerical methods to do this efficiently depend on the properties of the loss function. The most decisive among these properties is the convexity or non-convexity of the loss function. The fact that the loss function can have, and frequently has, non-convex regions has led to a widespread commitment to non-convex methods such as Adam. However, a local minimum implies that, in some environment around it, the function is convex. In this environment, second-order minimizing methods such as the Conjugate Gradient (CG) give a guaranteed superlinear convergence. We propose a novel framework grounded in the hypothesis that loss functions in real-world tasks swap from initial non-convexity to convexity towards the optimum. This is a property we leverage to design an innovative two-phase optimization algorithm. The presented algorithm detects the swap point by observing the gradient norm dependence on the loss. In these regions, non-convex (Adam) and convex (CG) algorithms are used, respectively. Computing experiments confirm the hypothesis that this simple convexity structure is frequent enough to be practically exploited to substantially improve convergence and accuracy.

优化算法深度学习凸性分析

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