假如有一万张卡,RL训练该怎么扩大规模?十万张呢?

2026.09.22 · 观点 · 论文 · 论文预印本

腾讯混元团队提出RL训练中Batch Size的规模化准则,以最短时间达标。

腾讯混元团队研究LLM强化学习中的Batch Size规模化问题,提出通过调整学习率保持样本效率,并优化系统吞吐来缩短训练时间。实验显示,在GRPO中,Batch Size在16倍范围内近似不变,但过大Batch会因样本效率损失而减速。

  • Batch Size影响样本效率和系统效率
  • GRPO在16倍prompt batch范围内近似不变
  • PPO中actor和critic的尺度不同
  • 通过调整学习率可保持样本效率
  • 过大Batch会因样本效率损失而减速

为什么值得看

为大规模RL训练提供Batch Size调优的实用准则。

适合谁关注

大模型训练工程师、强化学习研究者

原始来源