【深度学习】（9）--调整学习率

文章目录

调整学习率
- 一、学习率的定义
- 二、学习率的作用
- 三、实现调整学习率
- - 1. 使用库函数进行调整
  - 2. 手动调整学习率
总结

调整学习率

调整学习率的目的是：通过调整学习率，优化训练速度、提高训练稳定性、适应不同的训练阶段以及改善模型性能。那么，学习率是如何影响模型的呢？我们又该如何调整学习率呢？

一、学习率的定义

学习率（Learning Rate），也称为步长，是深度学习中控制参数更新步长的一个超参数。在每一次迭代中，学习率决定了基于损失函数梯度的参数更新幅度。它决定了模型在优化过程中向损失函数最优解移动的步长大小。

二、学习率的作用

控制参数更新的步长：学习率直接决定了在每次迭代中，参数根据梯度调整的幅度。过大的学习率可能导致参数更新幅度过大，从而在最优解附近震荡甚至发散；而过小的学习率则可能导致收敛速度过慢。
平衡速度与稳定性：合理的学习率需要在保证模型稳定收敛的同时，尽可能提高训练速度。过高的学习率可能导致模型训练不稳定，而过低的学习率则可能使训练过程过于缓慢。

由上可知，学习率是深度学习中的一个关键参数，它直接影响到模型的训练效果和收敛速度。选择合适的学习率对于训练一个高性能的深度学习模型至关重要。过高的学习率可能导致模型发散或震荡，而过低的学习率则可能使训练过程过于缓慢甚至无法收敛到最优解。

所以，我们要对学习率进行调整。

三、实现调整学习率

调整学习率有两种方法：1.使用库函数进行调整；2.手动调整学习率

1. 使用库函数进行调整

Pytorch学习率调整策略通过 torch.optim.lr_sheduler 接口实现。并提供3种调整方法：有序调整、自适应调整以及自定义调整。

（1）有序调整：等间隔调整(Step)，多间隔调整(MultiStep)，指数衰减(Exponential)，余弦退火(CosineAnnealing);

如下图所示：x轴为迭代次数，y轴表示学习率

在这里插入图片描述

等间隔调整(Step)：

等间隔调整学习率是一种简单直接的方法，它按照预定的间隔（通常是epoch数）来调整学习率。每次调整时，学习率会乘以一个预设的衰减因子（gamma）。这种方法适合于训练过程较为稳定，且需要逐步减小学习率以避免过拟合的场景。

torch.optim.lr_scheduler.StepLR(optimizer, step_size, gamma=0.1)

参数：
optimizer: 神经网络训练中使用的优化器，如optimizer=torch.optim.Adam(…)
step_size(int): 学习率下降间隔数，单位是epoch，而不是iteration.
gamma(float):学习率调整倍数，默认为0.1
每训练step_size个epoch，学习率调整为lr=lr*gamma.

代码体现：

注意：在每个epochs迭代训练时，使用**scheduler.step()**语句进行学习率更新。

optimizer = torch.optim.Adam(model.parameters(),lr=0.001,weight_decay=0.0001)
scheduler = torch.optim.lr_scheduler.StepLR(optimizer,step_size=5,gamma=0.5)

epochs = 80
train_dataloader = DataLoader(training_data,batch_size=64,shuffle=True)
test_dataloader = DataLoader(test_data,batch_size=64,shuffle=True)
for t in range(epochs):
    print(f"Epoch {t+1} \n-------------------------")
    train(train_dataloader,model,loss_fn,optimizer)
    test(test_dataloader,model,loss_fn)
    scheduler.step()

多间隔调整(MultiStep):

与Step方法类似，但可以在多个预设的epoch点进行学习率衰减，每个点可以有不同的衰减比例。这种方法提供了更多的灵活性来适应复杂的训练过程。

torch.optim.lr_shceduler.MultiStepLR(optimizer, milestones, gamma=0.1)

参数：
milestone(list): 一个列表参数，表示多个学习率需要调整的epoch值，如milestones=[10, 30, 80].

指数衰减(Exponential):

学习率按照指数形式进行衰减，衰减速度由衰减率（gamma）控制。这种方法适用于模型训练初期需要较大学习率以快速收敛，后期需要较小学习率以微调模型参数的情况。

torch.optim.lr_scheduler.ExponentialLR(optimizer, gamma)

参数：
gamma(float)：学习率调整倍数的底数，指数为epoch，初始值我lr, 倍数为γ的epoch次方

余弦退火(CosineAnnealing):

学习率按照余弦函数进行周期性调整，先下降后上升，形成多个学习率的波峰和波谷。这种方法有助于模型跳出局部最优解，探索更广阔的参数空间。

torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, T_max, eta_min=0)

参数：
Tmax(int):学习率下降到最小值时的epoch数，即当epoch=T_max时，学习率下降到余弦函数最小值，当epoch>T_max时，学习率将增大；
etamin: 学习率调整的最小值，即epoch=Tmax时，lrmin=etamin, 默认为0.

（2）自适应调整：依训练状况伺机而变，通过监测某个指标的变化情况(loss、accuracy)，当该指标不怎么变化时，就是调整学习率的时机(ReduceLROnPlateau);

torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, mode='min', factor=0.1,
patience=10,verbose=False, threshold=0.0001, threshold_mode='rel', cooldown=0, min_lr=0, eps=1e-08)

-- optimizer (Optimizer): 要调整学习率的优化器（如 SGD, Adam 等）。
-- mode (str, 可选): 指定触发学习率减少的条件。'min' 模式意味着当指标停止下降时减少学习率（即，当指标达到局部最小值				时）。'max' 模式则意味着当指标停止上升时减少学习率。默认值为 'min'。
-- factor (float, 可选): 学习率减少的因子。新的学习率将是旧学习率乘以这个因子。默认值为 0.1。
-- patience (int, 可选): 容忍多少个 epoch 指标没有改善后减少学习率。例如，如果 patience=10，则指标在 10 个 epoch 内没有		改善后，学习率将被减少。默认值为 10。
-- verbose (bool, 可选): 如果为 True，则在每次更新学习率时打印一条消息。默认值为 False。
-- threshold (float, 可选): 阈值，用于判断指标是否“真正”没有改善。只有当指标的变化小于这个阈值时，才认为指标没有改善。这有助			于避免由于指标的小幅波动而频繁调整学习率。默认值为 1e-4。
-- threshold_mode (str, 可选): 阈值的比较模式。'rel' 表示相对模式，即阈值是相对于当前最佳指标的百分比；'abs' 表示绝对模			式，即阈值是绝对数值。默认值为 'rel'。
-- cooldown (int, 可选): 在学习率被减少之后，等待多少个 epoch 再重新考虑是否减少学习率。这有助于在减少学习率后给模型一些时间		来适应新的学习率。默认值为 0，表示不等待。
-- min_lr (float 或 list, 可选): 学习率的最小值。如果为单个浮点数，则所有参数组的学习率都不会低于这个值。如果为列表，则每个参		数组的学习率都不会低于列表中对应位置的值。默认值为 0。
-- eps (float, 可选): 用于提高数值稳定性的小常数。当两个指标值非常接近时，它有助于避免除以零的错误。默认值为 1e-8。

（3）自定义调整：通过自定义关于epoch的lambda函数调整学习率(LambdaLR)。

torch.optim.lr_scheduler.LambdaLR(optimizer, lr_lambda)

参数：
lr_lambda(function or list): 自定义计算学习率调整倍数的函数，通常时epoch的函数，当有多个参数组时，设为list.

2. 手动调整学习率

手动调整学习率通常涉及在训练循环中定期检查模型的性能（如验证集上的损失或准确率），并根据这些性能指标来更新学习率。

比如：

# 训练循环  
for epoch in range(num_epochs):  
    # 训练模型（省略具体代码）  
    train(model, optimizer, train_loader)  
      
    # 评估模型在验证集上的性能（省略具体代码）  
    val_loss = validate(model, val_loader)  
      
    # 根据验证集损失手动调整学习率  
    # 这里只是一个简单的示例，实际中你可能需要更复杂的逻辑  
    if epoch % 10 == 0:  # 每10个epoch检查一次  
        if val_loss < best_val_loss:  # 假设best_val_loss在循环外初始化  
            best_val_loss = val_loss  
        else:  
            # 如果验证损失没有改善，则减少学习率  
            for param_group in optimizer.param_groups:  
                param_group['lr'] *= 0.1  # 将学习率乘以0.1

通过验证集损失，来调整学习率。

总结

本篇介绍了：

通过调整学习率能够优化训练速度、提高训练稳定性、适应不同的训练阶段以及改善模型性能。
调整学习率有两种方法：1.使用库函数进行调整；2.手动调整学习率
Pytorch学习率调整策略通过 torch.optim.lr_sheduler 接口实现。并提供3种库函数调整方法：有序调整、自适应调整以及自定义调整。
在库函数调整方法中，注意在每个epochs迭代训练时，使用**scheduler.step()**语句进行学习率更新。