Adam比SGD起步更稳,因其为每个参数自适应缩放步长:小梯度参数获“放大镜”、大梯度参数受“刹车”,同时偏差校正使初期更新更准;AdamW通过解耦权重衰减进一步提升稳定性和泛化性。

Adam 能加快大多数模型收敛,不是因为它“天生更快”,而是它在默认配置下更鲁棒地应对了真实训练中常见的三类问题:参数梯度量级不一致、初始阶段方向不稳定、小批量噪声干扰大。换言之,它把很多需要手动调的“适配工作”自动化了。
为什么 Adam 比 SGD 在多数任务上起步更稳?
SGD 所有参数共用一个 lr,而实际网络里:Embedding 层梯度稀疏且小,Linear 层最后几层梯度剧烈波动,BatchNorm 的 weight 和 bias 更新尺度差 2–3 个数量级。固定步长必然顾此失彼。
Adam 为每个参数单独维护 v_t(二阶矩估计),自动缩放步长为 lr / sqrt(v_t + eps):
- 小梯度参数 →
v_t小 → 步长被放大(相当于“放大镜”) - 大梯度参数 →
v_t大 → 步长被压缩(相当于“刹车”) - 这种按参数维度的自适应,让前 10–20 步更新路径平滑不少
注意:bias_correction=True(PyTorch 默认开启)会让初期更新偏大,前 5 步的 m_hat 和 v_hat 偏差可达 30%,所以别一上来就设 lr=1e-2——lr=1e-3 是安全起点,小数据或浅层模型甚至建议 lr=5e-4。
AdamW 为什么比 Adam 更适合现代模型?
关键不在“W”,而在权重衰减(weight_decay)是否解耦:
-
Adam把weight_decay加进梯度里:grad = grad + weight_decay * param
→ 自适应学习率会同时缩放这个衰减项,导致大梯度参数衰减被削弱,小梯度参数衰减被放大 -
AdamW是独立操作:param = param <em> (1 - lr </em> weight_decay)
→ 衰减量只和当前参数值、lr、weight_decay有关,不受梯度大小影响
实测中,Transformer 类模型用 Adam 容易早衰(Attention 权重过快收缩),而 AdamW(lr=3e-4, weight_decay=0.01) 在 BERT 微调中 F1 高出 0.4,训练稳定性提升 30% 以上。
立即学习“Python免费学习笔记(深入)”;
哪些场景下 Adam 反而拖慢收敛?
不是所有任务都受益于自适应机制:
- 小数据集(<1k 样本)或极浅模型(如 2 层 MLP):
Adam的状态变量(m,v)引入额外噪声,SGD+momentum=0.9更干净 - 混合精度训练(
torch.float16):eps=1e-8太小,易触发数值溢出;必须调大到eps=1e-5 - 需要后期泛化提升的任务(如 ImageNet 微调):
Adam易收敛到尖锐极小值(sharp minimum),而SGD的震荡反而帮它跳进平坦区域(flat minimum),最终准确率可能反超 0.8 个百分点
最常被忽略的一点:换优化器前,先检查是否开了 torch.nn.utils.clip_grad<em>norm</em> 和学习率 warmup。没开梯度裁剪的 RNN 训练几步就 nan,没 warmup 的 Adam 前 100 步震荡剧烈——这时换什么优化器都没用。
betas 的默认值((0.9, 0.999))在 PyTorch 和 TensorFlow 里一致,但旧版 Keras 是 (0.9, 0.9999),迁移代码时要注意。


















