arXiv:2411.05746cs.LGcs.AI2024-11被引 4

用连续时间模型解析Adam优化器,揭示超参与归一化的作用机制。

Continuous-Time Analysis of Adaptive Optimization and Normalization

  • 构建Adam和AdamW的连续时间模型,分析训练动态稳定性。
  • 发现超参稳定区域,超出则参数呈指数爆炸增长,实证验证成立。
  • 提出2-Adam、k-Adam等新优化器,解释归一化层的隐式自适应效果。

自适应优化算法(如Adam及其变体AdamW)是现代深度学习的核心组件,但其训练动态缺乏系统的理论理解,尤其对常见实践(如超参数选择、归一化层)为何促进泛化尚不清晰。本文提出Adam和AdamW的连续时间形式,实现可解析的训练动态分析。理论上推导出保证更新有界的超参稳定区域(β, γ),并通过实验验证:该区域外出现不稳定指数级参数增长。进一步,理论解释了归一化层的成功,源于尺度不变结构的隐式元自适应效应。基于此,提出显式优化器2-Adam,并推广为k-Adam——在k次自适应归一化下运行,其中k=1对应Adam,k=2对应带归一化层的Adam。整体上,该连续时间框架提供了对现代深度学习中最优超参与架构选择的原理性理解。

原文摘要 · Abstract (English)

Adaptive optimization algorithms, particularly Adam and its variant AdamW, are fundamental components of modern deep learning. However, their training dynamics lack comprehensive theoretical understanding, with limited insight into why common practices -- such as specific hyperparameter choices and normalization layers -- contribute to successful generalization. This work presents a continuous-time formulation of Adam and AdamW, facilitating a tractable analysis of training dynamics that can shed light on such practical questions. We theoretically derive a stable region for Adam's hyperparameters $(β, γ)$ that ensures bounded updates, empirically verifying these predictions by observing unstable exponential parameter growth outside of this stable region. Furthermore, we theoretically justify the success of normalization layers by uncovering an implicit meta-adaptive effect of scale-invariant architectural components. This insight leads to an explicit optimizer, $2$-Adam, which we generalize to $k$-Adam -- an optimizer that applies an adaptive normalization procedure $k$ times, encompassing Adam (corresponding to $k=1$) and Adam with a normalization layer (corresponding to $k=2$). Overall, our continuous-time formulation of Adam facilitates a principled analysis, offering deeper understanding of optimal hyperparameter choices and architectural decisions in modern deep learning.

优化器自适应连续时间归一化

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