腾讯混元:LLM 强化学习批大小扩展研究
⚡️ As LLM reinforcement learning scales to larger GPU clusters and more training data, training efficiency becomes a first-order concern.
腾讯混元将经典临界批大小理论扩展到在线 LLM 强化学习,在 GRPO 和 PPO 上发现重新调整学习率可在有界批大小范围内保持每条回复的学习效果。在固定硬件上,扩大批大小使 PPO 生成阶段吞吐量最高提升 2.29×,最佳 GRPO 配置达到相同验证目标所需时间减少 29%。
正文
当前显示原文,中文全文译文尚未就绪。
⚡️ As LLM reinforcement learning scales to larger GPU clusters and more training data, training efficiency becomes a first-order concern.
Our new research revisits classical critical-batch-size theory and extends it to online LLM RL, where the model generates its own training data and rollout generation and training scale differently.
Across GRPO and PPO, we find that learning-rate retuning can preserve learning per response over a bounded range of batch sizes.
On fixed hardware, scaling up the batch size improves PPO generation-stage throughput by up to 2.29×, while our best measured GRPO configuration reaches the same validation target in 29% less time. 🚀