NVIDIA 推出 LoGRA:RL 训练显存占用大降 45.7%

NVIDIA 发布论文 LoGRA,提出一种内存高效的大模型强化学习后训练方法。它不同于冻结主模型只训 LoRA 的做法,而是在反向传播时通过 rank-256 随机投影的低秩梯度草图、每步重绘的方式直接累积低秩梯度,全程不构建完整梯度矩阵,从而将 RL 训练显存占用降低 45.7%,低于 LoRA,并可在 8 卡上跑通 27B 模型的训练。

2026-10-12 ~ 2026-10-13 · 2 条相关