首页AI 频道AI 术语词典梯度累积(Gradient Accumulation)
梯

梯度累积(Gradient Accumulation)

模型与训练 2 次浏览 更新于 2026-11-09 18:04
返回术语列表

一句话理解

显存不够时的“蚂蚁搬家”——小批次多算几步,把梯度攒够一份再统一更新,效果约等于大批次训练。

定义

梯度累积指连续前向反向计算多次、只做一次参数更新:每次算得的梯度先累加,攒够 N 份再一起更新。效果近似把批量扩大 N 倍,但显存占用始终只有一份小批次的水平。

你会在哪遇到它

训练脚本里的累积步数参数;小显卡微调大模型的教程;分布式训练配置里的惯用搭配。显存告急时的第一反应。

背后的原理

神经网络的梯度天然可加——多个小批次的梯度取平均,数学上等价于大批次的梯度。省的是中间激活值的显存,代价是训练时间变长。它与分布式训练(977)、梯度检查点一起,被戏称为“显存三板斧”,微调大模型的教程里几乎是必选项。

和相近概念的区别

1 与真大批次训练:数学上近似等价,但个别按批次统计的组件会有细微差异。 2 与模型并行:一个省数据维度的显存,一个拆模型本身,解决的问题不同。 3 与混合精度(本批):一个省激活显存,一个省权重与计算开销,常常叠满全用。

常见误区

1 等效完全无损——累积步数过大时优化轨迹与真大批次仍有偏差,学习率常需微调。 2 攒得越多越好——收敛变慢,时间成本线性上涨。 3 能替代好显卡——它省显存不省算力,该跑多久还是多久。 4 步数别乱定——累积步数翻倍等于批量翻倍,常见取值 4 到 32,配学习率一起调。

所属分类模型与训练
更新时间2026-11-09 18:04
浏览量2
点赞0
每个 IP 仅可点赞一次 · 点赞后不可取消

评论 (0)

发表评论

0/300

暂无评论,来发表第一条评论吧!

便签
已输入:0/500字

    扫码访问工具