从SGD到AdamW：深度学习优化器的进化逻辑与Transformer实战选择

在训练一个现代深度学习模型时，优化器的选择往往决定了模型能否收敛、收敛速度多快以及最终性能上限。2017年Transformer架构的横空出世，不仅改变了自然语言处理的游戏规则，也对优化算法提出了全新挑战。当你在PyTorch中简单写下 optimizer = AdamW(model.parameters(), lr=5e-5) 时，背后是优化器领域三十年的技术演进。本文将揭示为什么AdamW成为了训练BERT、GPT等Transformer架构的事实标准，而这条进化之路上的每个关键转折，都对应着深度学习模型训练中一个亟待解决的核心痛点。

1. 优化器演进的底层逻辑：解决深度学习训练的四大难题

深度神经网络的优化本质上是一个在高维非凸空间中的搜索问题。与传统的凸优化不同，深度学习面临的损失函数地形复杂——存在大量鞍点、局部极小值以及各向异性的峡谷地形。优化器的演进史，就是一部人类如何应对这些挑战的创新史。

1.1 基础SGD及其根本局限

随机梯度下降(SGD)作为最朴素的优化方法，其更新规则简单直接：

# 纯SGD的Python实现
def sgd(params, gradients, lr):
    for param, grad in zip(params, gradients):
        param -= lr * grad

这种朴素的实现方式在深度学习时代暴露出三个关键缺陷：

1. 学习率敏感 ：所有参数共享同一学习率，而不同层、不同参数的最佳学习率往往差异巨大。例如：

   • 底层特征提取器需要较小的学习率保持稳定
   • 高层语义组合器需要较大学习率快速适应

2. 梯度震荡 ：特别是在损失函数的峡谷地形中，SGD会沿着陡峭方向震荡前进。在Transformer的自注意力层中，这种震荡会被放大，因为注意力权重需要精确调整。

3. 缺乏动量记忆 ：每次更新仅依赖当前batch的梯度，在mini-batch方差较大时（尤其是NLP任务），优化轨迹会出现剧烈抖动。

下表对比了SGD在不同类型神经网络中的表现：

网络类型	收敛速度	最终性能	稳定性
CNN图像分类	慢	较高	中等
RNN语言模型	极慢	一般	差
Transformer	无法收敛	极差	极差

1.2 动量方法的突破与局限

为缓解SGD的震荡问题，动量方法引入了物理中的惯性概念：

# 带动量的SGD实现
def sgd_momentum(params, gradients, v, lr, gamma=0.9):
    for i, (param, grad) in enumerate(zip(params, gradients)):
        v[i] = gamma * v[i] + lr * grad
        param -= v[i]

动量方法在Transformer训练中展现出两个独特价值：

1. 在注意力权重更新时，能平滑不同head之间的梯度差异
2. 在FFN层的参数更新中，有助于穿越平坦的损失区域

但2013年ICLR论文《On the importance of initialization and momentum in deep learning》揭示：传统动量在自适应调整不同参数方向的学习率方面仍然无能为力。这正是RMSProp登上舞台的契机。

2. 自适应学习率革命：从RMSProp到Adam

2.1 RMSProp的分参数学习率适应

RMSProp的核心创新在于为每个参数维护一个梯度平方的指数移动平均：

def rmsprop(params, gradients, sq_grad, lr, beta=0.9, eps=1e-8):
    for i, (param, grad) in enumerate(zip(params, gradients)):
        sq_grad[i] = beta * sq_grad[i] + (1-beta)*grad**2
        param -= lr * grad / (np.sqrt(sq_grad[i]) + eps)

这种自适应机制在Transformer训练中表现出三个关键优势：

1. 为embedding层的大稀疏梯度提供稳定性
2. 自动平衡注意力层中query/key/value矩阵的更新幅度
3. 缓解层归一化参数与其它参数之间的学习率冲突

但RMSProp缺乏动量积累，在BERT的预训练中会出现前期收敛缓慢的问题。这直接催生了Adam的诞生。

2.2 Adam：自适应矩估计的集大成者

Adam将动量与自适应学习率完美结合，其核心实现如下：

def adam(params, gradients, m, v, t, lr, beta1=0.9, beta2=0.999, eps=1e-8):
    for i, (param, grad) in enumerate(zip(params, gradients)):
        m[i] = beta1 * m[i] + (1-beta1) * grad
        v[i] = beta2 * v[i] + (1-beta2) * grad**2
        m_hat = m[i] / (1 - beta1**t)
        v_hat = v[i] / (1 - beta2**t)
        param -= lr * m_hat / (np.sqrt(v_hat) + eps)

Adam在Transformer训练中的优势包括：

• 自注意力层的梯度方差大，Adam能自动调节更新步长
• 对预训练初期不稳定的梯度分布具有鲁棒性
• 超参数(β1,β2)的默认值在大多数情况下表现良好

但2017年ICLR论文《Decoupled Weight Decay Regularization》揭示：Adam与L2权重衰减的结合存在根本性缺陷。

3. AdamW：为Transformer量身定制的优化方案

3.1 权重衰减与自适应学习率的耦合问题

传统Adam+L2的实现方式：

# 传统Adam+L2的实现（有问题）
for param in model.parameters():
    param.grad += weight_decay * param  # L2正则直接加到梯度上
optimizer.step()

这种方式导致权重衰减项也被自适应学习率缩放，造成两个严重后果：

1. 实际权重衰减强度随训练过程变化
2. 不同参数接收到的正则化强度不一致

AdamW的解决方案是将权重衰减与梯度更新完全解耦：

# AdamW的正确实现
optimizer.step()
for param in model.parameters():
    param.data -= lr * weight_decay * param.data

3.2 AdamW在Transformer中的实证优势

在BERT预训练任务中，AdamW相比Adam展现出三大改进：

1. 更稳定的泛化性能 ：

   • 在GLUE基准上，AdamW平均提升1.2个点
   • 特别在RTE、MRPC等小数据集上优势明显

2. 训练曲线更平滑 ：

   优化器	训练loss波动幅度	验证集一致性
   Adam	±0.15	差
   AdamW	±0.08	优

3. 超参数鲁棒性 ：

   • 对学习率和权重衰减的选择不再敏感
   • 在1e-5到5e-5学习率范围内表现稳定

3.3 实际训练中的AdamW配置建议

对于典型的Transformer模型，推荐配置如下：

optimizer = AdamW(
    model.parameters(),
    lr=5e-5,         # 初始学习率
    betas=(0.9, 0.999),  # 动量参数
    eps=1e-8,        # 数值稳定性
    weight_decay=0.01  # 解耦权重衰减
)

# 学习率预热调度器
scheduler = get_linear_schedule_with_warmup(
    optimizer,
    num_warmup_steps=1000,  # 前1k步预热
    num_training_steps=100000
)

关键调整经验：

• 权重衰减在0.01到0.1之间调节
• 预热步数约为总步数的1-2%
• β2=0.98可能对某些生成任务更优

4. 超越AdamW：优化器的最新进展与选择策略

4.1 新兴优化器的性能对比

近年来出现的LAMB、NovoGrad等优化器在特定场景下表现优异：

优化器	内存占用	训练速度	适合场景
AdamW	1x	1x	大多数Transformer
LAMB	1.2x	1.1x	超大batch训练
NovoGrad	1.5x	0.9x	低精度训练

4.2 优化器选择的决策树

基于模型特点和硬件条件的选择指南：

是否需要训练超大模型(>10B参数)?
├─ 是 → 考虑LAMB或分布式AdamW
└─ 否 → 
    Batch Size是否大于4096?
    ├─ 是 → 使用LAMB + 学习率缩放
    └─ 否 → 
        是否使用混合精度?
        ├─ 是 → AdamW或NovoGrad
        └─ 否 → 标准AdamW

4.3 优化器调参的高级技巧

1. 分层学习率 ：

param_groups = [
    {'params': model.embeddings.parameters(), 'lr': 1e-5},
    {'params': model.encoder.parameters(), 'lr': 5e-5},
    {'params': model.head.parameters(), 'lr': 1e-4}
]
optimizer = AdamW(param_groups)

2. 梯度裁剪与AdamW的结合 ：

   • 全局范数裁剪阈值设为1.0
   • 在预训练初期尤为重要

3. 权重衰减调度 ：

   • 后期训练逐渐降低衰减强度
   • 类似学习率衰减的余弦调度

在HuggingFace Transformers库的实际实现中，AdamW的默认配置已经针对BERT类模型进行了充分优化。但理解这些参数背后的数学原理，才能在遇到训练问题时做出精准调整。