mlsanityData to 7 Oct 2026

The Two-Stage Decision-Sampling Hypothesis: Understanding the Emergence of Self-Reflection in RL-Trained LLMs

arXiv 2601.01580 ↗First posted 4 Jan 2026cs.LG cs.AIcited by 0

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

  1. 2022Training language models to follow instructions with human feedbackcited by 15,712
  2. 2017Proximal Policy Optimization Algorithmscited by 14,441
  3. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  4. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  6. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  7. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  8. 2024Qwen2.5 Technical Reportcited by 4,713
  9. 2020Learning to summarize from human feedbackcited by 2,562
  10. 2015Trust Region Policy Optimizationcited by 2,503
  11. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  12. 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
17 more
  1. 2023Large Language Models Cannot Self-Correct Reasoning Yetcited by 965
  2. 2023MetaMath: Bootstrap Your Own Mathematical Questions for Large Language Modelscited by 724
  3. 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 655
  4. 2024Training Language Models to Self-Correct via Reinforcement Learningcited by 405
  5. 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 293
  6. 2022Exploring Length Generalization in Large Language Modelscited by 204
  7. 2025The Surprising Effectiveness of Negative Reinforcement in LLM Reasoningcited by 158
  8. 2024MathScale: Scaling Instruction Tuning for Mathematical Reasoningcited by 144
  9. 2024Marco-o1: Towards Open Reasoning Models for Open-Ended Solutionscited by 142
  10. 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 136
  11. 2023Teaching Arithmetic to Small Transformerscited by 121
  12. 2025On the Generalization of SFT: A Reinforcement Learning Perspective with Reward Rectificationcited by 121
  13. 2025Self-rewarding correction for mathematical reasoningcited by 67
  14. 2025PAG: Multi-Turn Reinforced LLM Self-Correction with Policy as Generative Verifiercited by 28
  15. 2025Boosting LLM Reasoning via Spontaneous Self-Correctioncited by 18
  16. 2025First Try Matters: Revisiting the Role of Reflection in Reasoning Modelscited by 12
  17. 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

  1. 2026Power Distribution Bridges Sampling, Self-Reward RL, and Self-Distillation
  2. 2026How RL Reshapes LLM Reasoning: Transferability, Coverage, and Scaling Laws
  3. 2026Rethinking RL for LLM Reasoning: It's Sparse Policy Selection, Not Capability Learning
  4. 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
  5. 2025Reflect, Retry, Reward: Self-Improving LLMs via Reinforcement Learning
  6. 2025No Free Lunch: Rethinking Internal Feedback for LLM Reasoning
  7. 2025From Emergence to Control: Probing and Modulating Self-Reflection in Language Models
  8. 2026The Cancellation Hypothesis in Critic-Free RL: From Outcome Rewards to Token Credits