Adam通过自适应更新降低一种独特尖锐度,带来比SGD更好的泛化性能。
Adam Reduces a Unique Form of Sharpness: Theoretical Insights Near the Minimizer Manifold
- 用随机微分方程建模Adam在最小值流形附近的自适应半梯度搜索
- 在带标签噪声的过参数模型中,Adam最小化Hessian平方根对角线迹,而非迹本身
- 适用于多种自适应优化器,为设计新优化器提供统一理论视角
尽管Adam在实践中广泛应用,但多数理论分析以随机梯度下降(SGD)为代理,对Adam所找到解的差异知之甚少。本文揭示,Adam通过其自适应更新隐式减小了一种独特的尖锐度度量,导致与SGD qualitatively different(定性不同)的解。具体而言,当训练损失较小时,Adam在最小值流形附近游走,并以自适应方式采用半梯度来最小化该尖锐度。我们通过连续时间近似(随机微分方程)严格刻画了这一行为。进一步地,在过参数模型加标签噪声的经典设定下,已知SGD最小化Hessian矩阵的迹$ r(oldsymbol{H})$,而我们证明Adam实际最小化$ r( ext{Diag}(oldsymbol{H})^{1/2})$。在对角线线性网络上的稀疏线性回归任务中,这一差异使Adam获得更优的稀疏性和泛化能力。最后,我们的分析框架可推广至包括RMSProp、Adam-mini、Adalayer和Shampoo在内的广泛自适应梯度方法,为这些优化器如何减小尖锐度提供了统一视角,有望启发未来优化器设计。
原文摘要 · Abstract (English)
Despite the popularity of the Adam optimizer in practice, most theoretical analyses study Stochastic Gradient Descent (SGD) as a proxy for Adam, and little is known about how the solutions found by Adam differ. In this paper, we show that Adam implicitly reduces a unique form of sharpness measure shaped by its adaptive updates, leading to qualitatively different solutions from SGD. More specifically, when the training loss is small, Adam wanders around the manifold of minimizers and takes semi-gradients to minimize this sharpness measure in an adaptive manner, a behavior we rigorously characterize through a continuous-time approximation using stochastic differential equations. We further demonstrate how this behavior differs from that of SGD in a well-studied setting: when training overparameterized models with label noise, SGD has been shown to minimize the trace of the Hessian matrix, $\tr(\mH)$, whereas we prove that Adam minimizes $\tr(\Diag(\mH)^{1/2})$ instead. In solving sparse linear regression with diagonal linear networks, this distinction enables Adam to achieve better sparsity and generalization than SGD. Finally, our analysis framework extends beyond Adam to a broad class of adaptive gradient methods, including RMSProp, Adam-mini, Adalayer and Shampoo, and provides a unified perspective on how these adaptive optimizers reduce sharpness, which we hope will offer insights for future optimizer design.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。