梯度优化把损失面的局部信息转化为参数更新
说明梯度下降、动量、自适应步长和学习率调度怎样影响训练,并分析曲率、噪声与非凸目标带来的限制。
梯度优化把损失面的局部信息转化为参数更新
训练模型通常被写成最小化经验损失:参数决定预测,预测与目标之间的差异形成标量目标。梯度给出目标对各参数的一阶局部变化率,优化器沿着降低目标的方向更新参数。这个过程只利用当前位置附近的信息,并不保证找到全局最优。
从批量梯度到随机更新
完整批量梯度使用全部训练样本,方向稳定但每一步成本高。随机梯度一次使用单个样本,更新噪声大却频繁;小批量训练在硬件并行、估计方差和内存之间折中,也是深度学习最常见的形式。
批量大小会改变噪声和有效步长。增大批量不只是加速计算,还可能需要重新调整学习率和正则化。不同批次的数据应来自明确的采样过程;类别、时间或地点高度相关时,批次梯度可能系统偏向某些区域。
一次更新常写成参数减去学习率乘以梯度。学习率过小,训练缓慢并可能停在平坦区域;过大则越过下降区域、振荡甚至发散。损失下降曲线、梯度范数和验证指标要一起观察。
动量积累连续方向
普通随机梯度会在狭长谷底两侧来回摆动。动量维护历史更新的指数平均,使持续方向得到累积、相反方向部分抵消。它能提高穿越平坦区域的速度,却引入额外状态和超参数。
Nesterov 类方法在前瞻位置估计梯度,试图更早修正过冲。不同框架对动量公式和权重衰减的实现可能不同,复现实验时要记录优化器版本、参数和更新顺序,不能只写“使用 SGD”。
动量不会自动解决坏数据、错误目标或梯度爆炸。序列模型中常用梯度裁剪限制一次更新的最大范数,但裁剪阈值也会改变学习动力学,应记录触发频率并检查是否长期处于裁剪状态。
自适应方法调整参数尺度
AdaGrad 根据历史平方梯度缩小频繁更新参数的步长,适合稀疏特征,但累计量只增不减,后期可能过早停止。RMSProp 使用指数平均减弱久远历史影响。Adam 同时维护一阶与二阶矩估计,并对初期偏差作修正。
自适应步长能减少手工处理参数尺度,却不表示无需调学习率。它们可能更快降低训练损失,但在某些任务上泛化不一定优于带动量的 SGD。权重衰减与 L2 正则在自适应优化器中也不总等价,需按具体算法解释。
数值稳定常依赖 epsilon、混合精度缩放和累加精度。半精度训练若出现下溢,梯度可能变成零;损失缩放能缓解问题,但溢出跳步和缩放策略要进入运行记录。
学习率调度控制训练阶段
固定学习率简单,但后期可能在较优区域持续震荡。阶梯衰减、余弦衰减和指数衰减逐渐缩小步长;预热则在训练开始时从小值升高,避免大批量或深层模型初期不稳定。
按训练步还是按 epoch 调度,在数据量和批量改变后含义不同。恢复检查点时若只恢复模型权重而不恢复优化器和调度器状态,后续轨迹不会延续原训练。分布式训练还要记录全局批量和梯度累积次数。
以验证指标触发降率或早停,会把验证集用于模型选择。反复观察同一验证集并改方案,仍可能过拟合;最终效果需要独立测试集或外部验证。
非凸目标中的局部困难
神经网络损失面包含平坦方向、鞍点和许多等价参数表示。梯度接近零可能表示局部极小、鞍点或饱和平坦区,仅看梯度大小无法区分。初始化、归一化、残差结构和数据尺度都会改变优化几何。
二阶方法利用曲率信息选择方向,但完整 Hessian 对大模型成本过高,工程中多用近似、预条件或一阶自适应方法。曲率很大时,统一步长容易在敏感方向过冲;参数重标度可改善条件数。
训练损失下降也可能来自记忆噪声标签。优化目标必须与业务目标和评价一致,交叉熵更低不一定让校准、公平性或鲁棒性更好。优化器只执行给定目标,不会修正目标定义。
诊断和停止需要多种证据
应同时监测训练与验证损失、任务指标、梯度与参数范数、学习率、吞吐和数值异常。训练损失不降可能来自数据管线、冻结参数或学习率错误,而不是模型容量不足。
比较优化器要控制初始化、数据顺序、训练预算和调参范围,并运行多个随机种子。只比较各自默认配置或最佳一次运行,会把调参投入与随机波动混入算法结论。
停止条件可以是预算耗尽、验证指标长期无改善、目标达到或风险门禁失败。最佳检查点选择规则要预先定义,并保存优化器状态和数据版本。梯度优化提供的是有效搜索机制,可靠模型仍取决于数据、目标、评价和部署条件。
训练记录与复现实务
优化器比较必须固定数据划分、初始化策略、批大小和评价预算,并同时记录按步数与按样本量计算的训练进度。只比较最终一次损失容易遗漏收敛速度、波动和计算成本差异。训练日志应包含学习率、梯度范数、参数范数、验证指标和停止原因;发生数值异常时,还应保留首次异常之前的检查点。这样才能区分算法本身、超参数选择与实现错误带来的影响。
资料依据
- Ian Goodfellow、Yoshua Bengio、Aaron Courville:Deep Learning,https://www.deeplearningbook.org/
- Stuart Russell、Peter Norvig:Artificial Intelligence: A Modern Approach,https://aima.cs.berkeley.edu/