Where Should RL Post-Training Compute Go? Model Size, Search, Learning, and Feedback
Not on the map. Why?
Abstract (arXiv)
Reinforcement Learning (RL) post-training is increasingly used to adapt foundation models for reasoning, planning, and feedback-driven robot-learning pipelines, but constrained post-training resources are often summarized by a single total FLOP budget. We study the fixed-budget decision problem behind this practice: under the same post-training budget, should one use a larger policy, train a smaller policy longer, generate more rollout search, or spend compute on stronger reward feedback? We introduce a FLOP-accounting framework for GRPO post-training that decomposes compute into rollout/search, policy-update/learning, and reward- or feedback-model evaluation. Across LoRA-adapted Qwen2.5 policies, we find conditional allocation frontiers: the best observed allocation changes with model size, compute budget, reward system, and evaluation target. Same-FLOP model-size comparisons show that model choice and training allocation are coupled because larger policies consume more per-token compute and therefore buy fewer updates or rollouts under the same budget. Reward systems also change the accounting: rule-based rewards spend nearly all non-update compute on policy rollouts, while PRM-style feedback allocates a visible part of the budget to reward-model inference. We present RACE as a diagnostic pilot-grid protocol, not a guarantee of held-out improvement, for identifying allocation regimes before expensive validation runs; our results suggest that RL post-training papers should report total FLOPs together with how compute is divided among model size, search, learning, and feedback.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 18 of its 18 references that are in the map. First posted on arXiv 15 Jul 2026.
Builds on
18 of 18 references.
References · 18
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2017Deep reinforcement learning from human preferencescited by 3,912
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 3,080
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2024OpenVLA: An Open-Source Vision-Language-Action Modelcited by 2,887
- 2022Do As I Can, Not As I Say: Grounding Language in Robotic Affordancescited by 2,348
- 2023PaLM-E: An Embodied Multimodal Language Modelcited by 2,071
6 more
- 2022RT-1: Robotics Transformer for Real-World Control at Scalecited by 2,061
- 2022Code as Policies: Language Model Programs for Embodied Controlcited by 1,210
- 2022Inner Monologue: Embodied Reasoning through Planning with Language Modelscited by 1,081
- 2022Language Models as Zero-Shot Planners: Extracting Actionable Knowledge for Embodied Agentscited by 1,035
- 2025The Art of Scaling Reinforcement Learning Compute for LLMscited by 101
- 2025Scaling Behaviors of LLM Reinforcement Learning Post-Training: An Empirical Study in Mathematical Reasoningcited by 29
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Spend Your Rollouts Where It Counts: Rollout Allocation for Group-Based RL Post-Training
- 2026Demystifying Reinforcement Learning Post-Training of Language Models
- 2026On the Role of Computation in Reinforcement Learning
- 2026RL Excursions during Pre-Training: Re-examining Policy Optimization for LLM training
- 2026Trust the Batch, On- or Off-Policy: Adaptive Policy Optimization for RL Post-Training
- 2026Unlocking the Critic: Reward-Free Policy Optimization for LLM Post-Training
- 2026Neglected Free Lunch from Post-training: Progress Advantage for LLM Agents
- 2024Reinforcement Learning for LLM Post-Training: A Survey