重新审视权重衰减与学习率的关系,提出更优的设置方式
Correction of Decoupled Weight Decay
- 基于更新独立于权重的假设,推导出权重衰减应正比于学习率平方
- 实验证明该设定能稳定权重和梯度范数,提升训练表现
- 适用于优化器调参、追求训练稳定的深度学习研究者
解耦权重衰减长期被视为AdamW优于Adam的关键因素,传统上设为与学习率γ成正比。近期有研究基于稳态正交性论证,主张应设为γ²。本文发现,消除更新中垂直于权重方向的贡献对训练动态影响甚微。相反,基于更新在稳态下独立于权重的简单假设,我们推导出解耦权重衰减应∝γ²,可实现稳定的权重范数。同时,基于相同假设,我们推导并实证表明:在Scion优化器下,小批量的总更新贡献(TUC)由依赖动量的有效学习率决定,其最优值可迁移。采用γ²设定能稳定权重与梯度范数,更好控制训练动态,提升模型性能。
原文摘要 · Abstract (English)
Decoupled weight decay, solely responsible for the performance advantage of AdamW over Adam, has long been set to proportional to learning rate $γ$ without questioning. Some researchers have recently challenged such assumption and argued that decoupled weight decay should be set $\propto γ^2$ instead based on orthogonality arguments at steady state. To the contrary, we find that eliminating the contribution of the perpendicular component of the update to the weight norm leads to little change to the training dynamics. Instead, we derive that decoupled weight decay $\propto γ^2$ results in stable weight norm based on the simple assumption that updates become independent of the weights at steady state, regardless of the nature of the optimizer. Based on the same assumption, we derive and empirically verify that the Total Update Contribution (TUC) of a minibatch under the Scion optimizer is better characterized by the momentum-dependent effective learning rate whose optimal value transfers and we show that decoupled weight decay $\propto γ^2$ leads to stable weight and gradient norms and allows us to better control the training dynamics and improve the model performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。