arXiv:2505.22509cs.LGmath.OC2025-05NeurIPS被引 2

让优化算法的停止时间可微,实现超快训练

Accelerating Optimization via Differentiable Stopping Time

  • 提出可微停止时间,通过微分方程理论保证其可导性
  • 在多种任务上比传统方法更快达到目标损失,提升训练效率
  • 适合需要动态调参或学习优化策略的场景

优化是现代机器学习的重要模块。现有方法多聚焦于固定时间内降低损失,但另一种更实际的目标——以最短时间达到目标损失——因停止时间不可导而难以优化。本文提出一种可微停止时间,并基于微分方程理论证明其可导性,设计高效反向传播算法。该方法实现了全新的可微优化加速框架,在在线超参数调优与学习优化等场景中表现优异,实验验证了其在多种任务上的有效性。

原文摘要 · Abstract (English)

Optimization is an important module of modern machine learning applications. Tremendous efforts have been made to accelerate optimization algorithms. A common formulation is achieving a lower loss at a given time. This enables a differentiable framework with respect to the algorithm hyperparameters. In contrast, its dual, minimizing the time to reach a target loss, is believed to be non-differentiable, as the time is not differentiable. As a result, it usually serves as a conceptual framework or is optimized using zeroth-order methods. To address this limitation, we propose a differentiable stopping time and theoretically justify it based on differential equations. An efficient algorithm is designed to backpropagate through it. As a result, the proposed differentiable stopping time enables a new differentiable formulation for accelerating algorithms. We further discuss its applications, such as online hyperparameter tuning and learning to optimize. Our proposed methods show superior performance in comprehensive experiments across various problems, which confirms their effectiveness.

优化算法可微停止超参数调优

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