跳到正文
热点事件持续更新

梯度裁剪提升不稳定RNN训练稳定性

1 篇报道1 个报道来源2 小时前更新

先了解这件事

AI 综述

作者用 PyTorch 对刻意设置成不稳定的 RNN 进行对照实测:不裁剪时,5 次运行均在 17 步内发散;将梯度范数裁剪至 1.0 后,5 次均完成 320 步训练,验证损失中位数为 0.043,而裁剪阈值升至 5.0 时效果明显较差。 测试在初始权重、数据和学习率相同的前提下,以 5 个随机种子重复。结果表明,裁剪阈值并非越高越好;此外,将学习率降至 0.005,即使不裁剪也能避免该测试中的发散。该结论仅对应这项刻意不稳定的 RNN 实验。

AI 根据报道生成 · 2 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月24日
  1. AGI Hunt
    PyTorch 实测梯度裁剪对 RNN 训练稳定性的影响

    作者用 PyTorch 训练刻意不稳定的 RNN,对比有无梯度裁剪的效果,5 次不裁剪运行均在 17 步内发散。将梯度范数裁剪到 1.0 后,5 次运行全部完成 320 步,验证损失中位数为 0.043,而裁剪到 5.0 的效果明显较差。文章还给出裁剪公式,指出 clip 1.0 时 94.7% 的步骤触发裁剪,并将学习率降至 0.005 也可避免不裁剪时发散。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。