When Do Larger Batches Help Scale LLM Reinforcement Learning?
Not on the map. Why?
Abstract (arXiv)
Larger batches reduce the variance of stochastic gradients per update and are therefore often expected to accelerate training. Yet whether this statistical benefit translates into lower wall-clock time-to-target remains unclear, because each update consumes more samples and may take longer to execute. We study this tradeoff in reinforcement learning for large language models. We separate its algorithmic and systems effects by comparing learning and execution along their natural axes. At the algorithmic level, we compare configurations at equal cumulative sample counts while retuning batch-dependent hyperparameters. Over a bounded range of batch sizes, this procedure yields an approximately batch-size-invariant family whose members follow similar sample-indexed learning trajectories. At the systems level, we exploit the computational asymmetry between rollout generation and training: autoregressive generation is often memory-bandwidth-bound at low concurrency, whereas training work scales approximately with the number of processed tokens. Combining these two views yields a direct decision rule: a larger-batch configuration reduces time-to-target only when its throughput gain exceeds its samples-to-target penalty. Experiments with GRPO and PPO support both sides of this decomposition. At the algorithmic level, square-root learning-rate scaling with Adam produces approximately batch-size-invariant learning curves over a bounded range of batch sizes. At the systems level, larger batches improve generation throughput by up to 2.29x on fixed hardware. In GRPO, combining higher throughput with learning-rate retuning reduces time-to-target by up to 29%, whereas increasing the batch without retuning is slower despite its higher throughput.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 24 of its 29 references that are in the map. First posted on arXiv 29 Aug 2026.
Builds on
24 of 29 references.
References · 24
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2024DistServe: Disaggregating Prefill and Decoding for Goodput-optimized Large Language Model Servingcited by 595
- 2022Efficiently Scaling Transformer Inferencecited by 529
- 2017Don't Decay the Learning Rate, Increase the Batch Sizecited by 313
- 2018An Empirical Model of Large-Batch Trainingcited by 268
12 more
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 234
- 2018Measuring the Effects of Data Parallelism on Neural Network Trainingcited by 190
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 184
- 2022On the SDEs and Scaling Rules for Adaptive Gradient Algorithmscited by 100
- 2021Batch size-invariance for policy optimizationcited by 33
- 2025PipelineRL: Faster On-policy Reinforcement Learning for Long Sequence Generationcited by 29
- 2025Improving Sampling Efficiency in RLVR through Adaptive Rollout and Response Reusecited by 28
- 2025BroRL: Scaling Reinforcement Learning via Broadened Explorationcited by 23
- 2025Critical Batch Size Revisited: A Simple Empirical Approach to Large-Batch Language Model Trainingcited by 20
- 2026Adaptive Rollout Allocation for Online Reinforcement Learning with Verifiable Rewardscited by 15
- 2023Beyond Implicit Bias: The Insignificance of SGD Noise in Online Learningcited by 9
- 2026IsoCompute Playbook: Optimally Scaling Sampling Compute for LLM RLcited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026EasyPPO: Stabilizing the Critic Is Keycited by 0
- 2026EasyPPO: Stabilizing the Critic Is Key29 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2025Scaling Behaviors of LLM Reinforcement Learning Post-Training: An Empirical Study in Mathematical Reasoning
- 2026BASIS: Batchwise Advantage Estimation from Single-Rollout Information Sharing for LLM Reasoning
- 2025On Predictability of Reinforcement Learning Dynamics for Large Language Models
- 2026On Training in Imagination
- 2026Spend Your Rollouts Where It Counts: Rollout Allocation for Group-Based RL Post-Training
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning
- 2026A Survey of Reinforcement Learning for Large Language Models under Data Scarcity: Challenges and Solutions
- 2020Provably Good Batch Reinforcement Learning Without Great Exploration