首次证明原始Adam在非光滑非凸优化中收敛,破解其实际有效性之谜。
Adam Converges in Nonsmooth Nonconvex Optimization
- 引入随机缩放学习率,首次完成原始Adam的有限时间分析
- 在非光滑非凸场景下达到1/T^(2/13)的收敛速率,适用于重尾噪声
- 理论支持β₁=β₂参数设置,与近期实证研究一致
Adam是当前最广泛使用且影响深远的现代优化器。它为何在各类实际问题中表现优异?这一问题过去十年一直是优化领域的核心挑战。然而,现有研究多集中于光滑非凸优化中的收敛速率,难以反映真实场景——因神经网络训练等问题普遍为非光滑。近期提出的Online-to-Nonconvex Conversion框架为分析非光滑非凸下的Adam提供了新思路。但此前工作存在两大缺陷:一、忽略了原版Adam中的偏差修正项;二、引入了原始算法未使用的额外操作(如裁剪)。因此,原始Adam的收敛性仍不明确。本文首次对经典形式的Adam(含偏差修正项,无额外修改)进行有限时间分析,证明随机缩放学习率下,其在非光滑非凸优化中可实现1/T^(2/13)的收敛速率。此外,结果适用于更贴近实践的重尾噪声情形。有趣的是,理论成立条件恰好对应近期实证研究推荐的β₁=β₂设置。
原文摘要 · Abstract (English)
Adam is one of the most widely implemented and influential modern optimizers. Why is it effective across different optimization problems in practice? This question arguably lies at the center of the optimization community over the last decade and has motivated a substantial body of work aimed at understanding its convergence behavior. However, existing studies have mainly focused on the convergence rate of Adam in smooth nonconvex optimization, which unfortunately does not adequately capture practical settings, since many real-world problems are nonsmooth, such as those arising in training neural networks. Thus, these studies cannot fully explain the popularity and empirical success of Adam. Recently, an insightful and powerful framework called Online-to-Nonconvex Conversion has opened a new way to analyze Adam for nonsmooth nonconvex optimization. Unfortunately, prior works along this line share two common limitations. First, all of them ignore the important bias-correction term in the original Adam algorithm. Second and more importantly, many of them require extra operations that are not used in Adam, such as a clipping step. Therefore, the convergence guarantee for the original Adam method still remains unclear. In this work, we present the first finite-time analysis for the classical form of Adam, i.e., with the bias-correction step and without further algorithmic modifications, and prove that a randomly scaled learning rate ensures a convergence rate of $1/T^{\frac{2}{13}}$ for nonsmooth nonconvex optimization. Moreover, our result provably applies to the modern heavy-tailed noise regime, which is closer to practice. Interestingly, our theory is established under the parameter choice $β_1=β_2$, aligning with the recent empirical studies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。