arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.00992math.OCcs.LGcs.NE

Power Law 谱条件下优化的紧收敛速率界

Tight Convergence Rate Bounds for Optimization Under Power Law Spectral Conditions

Technology Innovation Institute(技术创新研究所);Ecole Polytechnique(巴黎综合理工学院);Skolkovo Institute of Science and Technology(斯科尔科沃科技学院)

Maksim Velikanov, Dmitry Yarotsky

AI总结:

针对谱呈power law分布的二次优化问题,本文提出新的谱条件以获得更紧收敛上界,为多种梯度算法构建了完整的上下界图景并统一推导出最优加速策略,同时首次证明了最速下降法与共轭梯度法的紧下界。

摘要:

二次问题上的优化性能敏感地依赖于谱的低频部分。对于大型(实际上是无限维的)问题,这部分谱通常可以自然地由 power law 分布表示或近似,从而使得基于梯度的算法对这些问题的迭代求解产生 power law 收敛速率。在本文中,我们提出了一种新的谱条件,为具有 power law 优化轨迹的问题提供了更紧的上界。我们利用该条件为一系列广泛的优化算法——Gradient Descent、Steepest Descent、Heavy Ball 和 Conjugate Gradients——构建了上下界的完整图景,并重点关注学习率和动量的底层调度。特别地,我们展示了如何针对给定的谱形状,以统一的方式获得最优加速方法、其调度以及收敛上界。此外,我们首次为 Steepest Descent 和 Conjugate Gradients 在具有一般指数的 spectral power laws 下的收敛速率提供了紧下界证明。我们的实验表明,所获得的收敛界和加速策略不仅与精确的二次优化问题相关,而且在应用于神经网络训练时也相当准确。

英文摘要:

Performance of optimization on quadratic problems sensitively depends on the low-lying part of the spectrum. For large (effectively infinite-dimensional) problems, this part of the spectrum can often be naturally represented or approximated by power law distributions, resulting in power law convergence rates for iterative solutions of these problems by gradient-based algorithms. In this paper, we propose a new spectral condition providing tighter upper bounds for problems with power law optimization trajectories. We use this condition to build a complete picture of upper and lower bounds for a wide range of optimization algorithms -- Gradient Descent, Steepest Descent, Heavy Ball, and Conjugate Gradients -- with an emphasis on the underlying schedules of learning rate and momentum. In particular, we demonstrate how an optimally accelerated method, its schedule, and convergence upper bound can be obtained in a unified manner for a given shape of the spectrum. Also, we provide first proofs of tight lower bounds for convergence rates of Steepest Descent and Conjugate Gradients under spectral power laws with general exponents. Our experiments show that the obtained convergence bounds and acceleration strategies are not only relevant for exactly quadratic optimization problems, but also fairly accurate when applied to the training of neural networks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑