Rationale-Guided Policy Optimization: Learning to Reason with Adaptive Rationale Scaffolding
cs.AI
Submitted: 2026-10-05
Updated: 2026-10-05
Comments: NeurIPS 2026
Code: https://github.com/VietHoang1512/rgpo
License: http://creativecommons.org/publicdomain/zero/1.0/
The gist: On-policy reinforcement learning has become a central paradigm for improving the reasoning abilities of large language models.
Terminology
Abstract
On-policy reinforcement learning has become a central paradigm for improving the reasoning abilities of large language models. However, its effectiveness is often limited by reward sparsity: when a model fails to discover correct trajectories for difficult problems, the optimization process receives little useful signal and may stagnate. Existing approaches mitigate this issue by incorporating off-policy demonstrations, expert traces, or model-generated solutions, but they typically require the auxiliary data to match the format of the reinforcement-learning task, often relying on rejection sampling from stronger models to obtain suitable training trajectories. We introduce Rationale-Guided Policy Optimization (RGPO), a framework that adaptively leverages ground-truth rationale information according to the model's current capability while preserving its freedom to explore. Rather than treating reference solutions as fixed imitation targets, RGPO uses them as temporary scaffolds: rationales help the model generate improved responses, after which only higher-reward, model-generated solutions are transferred back to the original unguided setting. This design allows training to exploit available ground-truth information without requiring off-policy data to follow the same format as the RL task. Across both language-only and vision-language reasoning settings, RGPO consistently improves performance over RLVR baselines, and ablation studies show that adaptive rationale guidance is a key contributor to these gains. These results suggest that RGPO offers a practical and general approach for reducing reward sparsity, stabilizing reinforcement learning, and improving reasoning performance in both text-only and multimodal models.
Sources
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Learning to Reason under Off-Policy Guidance
- GHPO: Adaptive Guidance for Stable and Efficient LLM Reinforcement Learning
- Don't Tell the Answer, Truly Guide the Reasoning During RL Rollouts
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weighting
- A Minimalist Approach to LLM Reasoning: from Rejection Sampling to Reinforce
- QuestA: Expanding Reasoning Capacity in LLMs via Question Augmentation
- Training language models to follow instructions with human feedback
- Mulberry: Empowering MLLM with o1-like Reasoning and Reflection via Collective Monte Carlo Tree Search
- REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization
- Open-Reasoner-Zero: An Open Source Approach to Scaling Up Reinforcement Learning on the Base Model
- Rethinking Entropy Interventions in RLVR: An Entropy Change Perspective
- Qwen2.5 Technical Report
- DAPO: An Open-Source LLM Reinforcement Learning System at Scale
- A Study of BFLOAT16 for Deep Learning Training
- Training Deep Nets with Sublinear Memory Cost
- Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challenge
- GPQA: A Graduate-Level Google-Proof Q&A Benchmark
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems