
Bayes-Adaptive MDPs for Rollout Allocation to Enhance GRPO Training
AdaRoll-GRPO is a method that allocates GRPO rollout budgets per question by difficulty instead of uniformly. It frames allocation as a sequential decision problem and learns an allocation policy with a Bayes-adaptive MDP, computed offline so training adds no online optimization cost. Experiments on several reasoning datasets and LLM architectures show gains over standard GRPO at the same total rollout budget.
training, researchΒ· Added 28 Oct 2026, published 27 Oct 2026





