深度学习基础——神经网络优化算法

1.1梯度下降法

梯度下降法:函数沿梯度方向有最大的变化率,优化目标损失函数时,根据负梯度方向进行。
θ t + 1 = θ t − η ⋅ ∇ θ L ( θ t ) \theta_{t+1} = \theta_t - \eta \cdot \nabla_\theta L(\theta_t) θt+1=θt−η⋅∇θL(θt)

其中:
η \eta η为学习率,即每次更新的步长。
∇ θ \nabla_\theta ∇θ为梯度。

1.2改进的梯度下降法

以下是神经网络梯度下降法中几种经典改进方法的讲解,包括核心思想、公式及适用场景:


1. Momentum(动量法)

核心思想 :模拟物理中的动量,在参数更新时引入历史梯度方向的加权平均,加速收敛并减少震荡。
公式
v t = β v t − 1 + ( 1 − β ) ∇ θ L ( θ t ) θ t + 1 = θ t − η ⋅ v t v_{t} = \beta v_{t-1} + (1-\beta) \nabla_\theta L(\theta_t) \\ \theta_{t+1} = \theta_t - \eta \cdot v_t vt=βvt−1+(1−β)∇θL(θt)θt+1=θt−η⋅vt

  • 特点
    • β \beta β 为动量系数(通常取0.9),控制历史梯度的权重。
    • 在梯度方向一致时加速更新,梯度方向变化时减少震荡。
  • 适用场景:损失函数存在局部最小值或鞍点时效果显著。

2. AdaGrad(自适应梯度)

核心思想 :为每个参数自适应调整学习率,历史梯度平方的累积值越大,学习率越小。
公式
G t = G t − 1 + ( ∇ θ L ( θ t ) ) 2 θ t + 1 = θ t − η G t + ϵ ⋅ ∇ θ L ( θ t ) G_t = G_{t-1} + (\nabla_\theta L(\theta_t))^2 \\ \theta_{t+1} = \theta_t - \frac{\eta}{\sqrt{G_t + \epsilon}} \cdot \nabla_\theta L(\theta_t) Gt=Gt−1+(∇θL(θt))2θt+1=θt−Gt+ϵ η⋅∇θL(θt)

  • 特点
    • 学习率随训练逐步衰减,适合稀疏数据(如自然语言处理)。
    • 缺点:累积梯度平方可能导致学习率过早趋近于零。
  • 适用场景:特征稀疏或需要自适应调整学习率的任务。

3. RMSProp(均方根传播)

核心思想 :改进AdaGrad的累积方式,引入指数衰减平均,避免学习率过早下降。
公式
E t = β E t − 1 + ( 1 − β ) ( ∇ θ L ( θ t ) ) 2 θ t + 1 = θ t − η E t + ϵ ⋅ ∇ θ L ( θ t ) E_t = \beta E_{t-1} + (1-\beta)(\nabla_\theta L(\theta_t))^2 \\ \theta_{t+1} = \theta_t - \frac{\eta}{\sqrt{E_t + \epsilon}} \cdot \nabla_\theta L(\theta_t) Et=βEt−1+(1−β)(∇θL(θt))2θt+1=θt−Et+ϵ η⋅∇θL(θt)

  • 特点
    • 通过衰减系数 β \beta β(通常取0.9)控制历史梯度的影响。
    • 解决了AdaGrad学习率单调下降的问题。
  • 适用场景:非平稳目标函数或需要动态调整学习率的场景。

4. Adam(自适应矩估计)

核心思想 :结合Momentum和RMSProp,利用梯度的一阶矩(均值)和二阶矩(方差)进行自适应调整。
公式
m t = β 1 m t − 1 + ( 1 − β 1 ) ∇ θ L ( θ t ) ( 一阶矩 ) v t = β 2 v t − 1 + ( 1 − β 2 ) ( ∇ θ L ( θ t ) ) 2 ( 二阶矩 ) m ^ t = m t 1 − β 1 t , v ^ t = v t 1 − β 2 t ( 偏差修正 ) θ t + 1 = θ t − η v ^ t + ϵ ⋅ m ^ t m_t = \beta_1 m_{t-1} + (1-\beta_1)\nabla_\theta L(\theta_t) \quad (\text{一阶矩}) \\ v_t = \beta_2 v_{t-1} + (1-\beta_2)(\nabla_\theta L(\theta_t))^2 \quad (\text{二阶矩}) \\ \hat{m}_t = \frac{m_t}{1-\beta_1^t}, \quad \hat{v}t = \frac{v_t}{1-\beta_2^t} \quad (\text{偏差修正}) \\ \theta{t+1} = \theta_t - \frac{\eta}{\sqrt{\hat{v}_t} + \epsilon} \cdot \hat{m}_t mt=β1mt−1+(1−β1)∇θL(θt)(一阶矩)vt=β2vt−1+(1−β2)(∇θL(θt))2(二阶矩)m^t=1−β1tmt,v^t=1−β2tvt(偏差修正)θt+1=θt−v^t +ϵη⋅m^t

  • 特点
    • 超参数 β 1 \beta_1 β1(通常0.9)和 β 2 \beta_2 β2(通常0.999)分别控制一阶和二阶矩的衰减。
    • ϵ \epsilon ϵ为一个很小的常数。
    • 偏差修正(Bias Correction)避免初始阶段估计偏差。
    • 综合了动量加速和自适应学习率的优点。
  • 适用场景:通用性强,尤其适合大数据集和复杂模型(如深度学习)。

总结与对比
方法 核心改进点 优点 缺点
Momentum 引入动量加速收敛 减少震荡,加速平坦区域收敛 对噪声敏感,需调参(\beta)
AdaGrad 自适应学习率(历史梯度平方) 适合稀疏数据 学习率过早下降,需手动设置(\epsilon)
RMSProp 指数衰减的梯度平方累积 解决AdaGrad学习率下降过快问题 超参数(\beta)需调节
Adam 一阶矩+二阶矩自适应 收敛快,适应性强,通用性好 内存占用略高,超参数较多

实际应用建议
  • 默认选择:优先尝试Adam,因其在大多数任务中表现稳定。

  • 特殊场景:稀疏数据可尝试AdaGrad或RMSProp;简单模型可用Momentum。

  • 框架实现

    python 复制代码
    # PyTorch示例
    optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
    optimizer = torch.optim.SGD(model.parameters(), lr=0.01, momentum=0.9)
相关推荐
星光银河1 分钟前
链表操作场景与循环条件技巧总结表(含指针初始化)
数据结构·算法·链表
2301_7944615714 分钟前
力扣-两数之和
数据结构·算法·leetcode
技术帮扶户15 分钟前
Leetcode-3 判断根结点是否等于子结点之和
算法·leetcode
强盛小灵通专卖员32 分钟前
DL00987-基于深度学习YOLOv11的红外鸟类目标检测含完整数据集
人工智能·深度学习·算法·目标检测·计算机视觉
小喵喵生气气43 分钟前
Python60日基础学习打卡D32
python·学习·算法
老唐7771 小时前
前沿技术:如何从多模态文件中自动提取有效信息?
人工智能·深度学习·机器学习·ai·自然语言处理·大模型·多模态
自由的风.1 小时前
回溯法求解N皇后问题
算法·c#·剪枝·迭代加深
何双新1 小时前
第9.1讲、Tiny Encoder Transformer:极简文本分类与注意力可视化实战
深度学习·分类·transformer
yuanManGan1 小时前
领略算法真谛:单源最短路问题
算法
学算法的程霖1 小时前
目标检测135个前沿算法模型汇总(附源码)!
图像处理·人工智能·pytorch·深度学习·目标检测·计算机视觉·transformer