The Two-Stage Decision-Sampling Hypothesis: Understanding the Emergence of Self-Reflection in RL-Trained LLMs
Not on the map. Why?
Abstract (arXiv)
Self-reflection capabilities emerge in Large Language Models after RL post-training, with multi-turn RL achieving substantial gains over SFT counterparts. Yet the mechanism of how a unified optimization objective gives rise to functionally distinct capabilities of generating solutions and evaluating when to revise them remains opaque. To address this question, we introduce the Gradient Attribution Property to characterize how reward gradients distribute across policy components, formalized through the Two-Stage Decision-Sampling (DS) Hypothesis, which decomposes the policy into sampling (πsample) for generation and decision (πd) for verification. We prove that surrogate rewards exhibit Balanced Gradient Attribution, while SFT and KL penalties exhibit Unbalanced Gradient Attribution, with length-weighting creating asymmetric regularization that constrains πsample while leaving πd under-optimized, providing an theoretical explanation of why RL succeeds where SFT fails. We also empirically validate our theoretical predictions on arithmetic reasoning demonstrates that RL's superior generalization stems primarily from improved decision-making (πd) rather than sampling capabilities, providing a first-principles mechanistic explanation for self-correction in thinking models.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 29 of its 36 references that are in the map. First posted on arXiv 4 Jan 2026.
Builds on
29 of 36 references.
References · 29
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2020Learning to summarize from human feedbackcited by 2,562
- 2015Trust Region Policy Optimizationcited by 2,503
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
17 more
- 2023Large Language Models Cannot Self-Correct Reasoning Yetcited by 965
- 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 724
- 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 655
- 2024Training Language Models to Self-Correct via Reinforcement Learningcited by 405
- 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 293
- 2022Exploring Length Generalization in Large Language Modelscited by 204
- 2025The Surprising Effectiveness of Negative Reinforcement in LLM Reasoningcited by 158
- 2024MathScale: Scaling Instruction Tuning for Mathematical Reasoningcited by 144
- 2024Marco-o1: Towards Open Reasoning Models for Open-Ended Solutionscited by 142
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 136
- 2023Teaching Arithmetic to Small Transformerscited by 121
- 2025On the Generalization of SFT: A Reinforcement Learning Perspective with Reward Rectificationcited by 121
- 2025Self-rewarding correction for mathematical reasoningcited by 67
- 2025PAG: Multi-Turn Reinforced LLM Self-Correction with Policy as Generative Verifiercited by 28
- 2025Boosting LLM Reasoning via Spontaneous Self-Correctioncited by 18
- 2025First Try Matters: Revisiting the Role of Reflection in Reasoning Modelscited by 12
- 2024Principled Understanding of Generalization for Generative Transformer Models in Arithmetic Reasoning Taskscited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation
- 2026How RL Reshapes LLM Reasoning: Transferability, Coverage, and Scaling Laws
- 2026Rethinking RL for LLM Reasoning: It's Sparse Policy Selection, Not Capability Learning
- 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
- 2025Reflect, Retry, Reward: Self-Improving LLMs via Reinforcement Learning
- 2025No Free Lunch: Rethinking Internal Feedback for LLM Reasoning
- 2025From Emergence to Control: Probing and Modulating Self-Reflection in Language Models
- 2026The Cancellation Hypothesis in Critic-Free RL: From Outcome Rewards to Token Credits