From P(y|x) to P(y): Investigating Reinforcement Learning in Pre-train Space
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
While reinforcement learning with verifiable rewards (RLVR) significantly enhances LLM reasoning by optimizing the conditional distribution P(y|x), its potential is fundamentally bounded by the base model's existing output distribution. Optimizing the marginal distribution P(y) in the Pre-train Space addresses this bottleneck by encoding reasoning ability and preserving broad exploration capacity. Yet, conventional pre-training relies on static corpora for passive learning, leading to a distribution shift that hinders targeted reasoning enhancement. In this paper, we introduce PreRL (Pre-train Space RL), which applies reward-driven online updates directly to P(y). We theoretically and empirically validate the strong gradient alignment between log P(y) and log P(y|x), establishing PreRL as a viable surrogate for standard RL. Furthermore, we uncover a critical mechanism: Negative Sample Reinforcement (NSR) within PreRL serves as an exceptionally effective driver for reasoning. NSR-PreRL rapidly prunes incorrect reasoning spaces while stimulating endogenous reflective behaviors, increasing transition and reflection thoughts by 14.89x and 6.54x, respectively. Leveraging these insights, we propose Dual Space RL (DSRL), a Policy Reincarnation strategy that initializes models with NSR-PreRL to expand the reasoning horizon before transitioning to standard RL for fine-grained optimization. Extensive experiments demonstrate that DSRL consistently outperforms strong baselines, proving that pre-train space pruning effectively steers the policy toward a refined correct reasoning subspace.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 67 of its 71 references that are in the map. First posted on arXiv 15 Apr 2026.
Builds on
67 of 71 references.
Closest ancestors · 18
Same direction, newest first
- 2026How to Allocate, How to Learn? Dynamic Rollout Allocation and Advantage Modulation for Policy Optimization22 Feb
- 2026VI-CuRL: Stabilizing Verifier-Independent RL Reasoning via Confidence-Guided Variance Reduction13 Feb
- 2026Unveiling Implicit Advantage Symmetry: Why GRPO Struggles with Exploration and Difficulty Adaptation5 Feb
- 2025On the Interplay of Pre-Training, Mid-Training, and RL on Reasoning Language Models8 Dec
- 2025RLP: Reinforcement as a Pretraining Objective26 Sep
- 2025Pass@k Training for Adaptively Balancing Exploration and Exploitation of Large Reasoning Models14 Aug
- 2025Squeeze the Soaked Sponge: Efficient Off-policy Reinforcement Finetuning for Large Language Model9 Jul
- 2025OctoThinker: Mid-training Incentivizes Reinforcement Learning Scaling25 Jun
- 2025The Surprising Effectiveness of Negative Reinforcement in LLM Reasoning2 Jun
- 2025Learning to Reason under Off-Policy Guidance21 Apr
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
- 2025Open-Reasoner-Zero: An Open Source Approach to Scaling Up Reinforcement Learning on the Base Model31 Mar
6 more
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025SimpleRL-Zoo: Investigating and Taming Zero Reinforcement Learning for Open Base Models in the Wild24 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
- 2023Let's Verify Step by Step31 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 49
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2024GPT-4o System Cardcited by 4,467
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 1,916
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
37 more
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,758
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2024Do NOT Think That Much for 2+3=? On the Overthinking of o1-Like LLMscited by 564
- 2026Kimi K2.5: Visual Agentic Intelligencecited by 449
- 2019ERNIE 2.0: A Continual Pre-training Framework for Language Understandingcited by 230
- 2025Reasoning Models Can Be Effective Without Thinkingcited by 182
- 2020What Matters In On-Policy Reinforcement Learning? A Large-Scale Empirical Studycited by 157
- 2023OpenWebMath: An Open Dataset of High-Quality Mathematical Web Textcited by 151
- 2023Continual Pre-Training of Large Language Models: How to (re)warm your model?cited by 135
- 2022Continual Pre-Training Mitigates Forgetting in Language and Visioncited by 82
- 2023MathPile: A Billion-Token-Scale Pretraining Corpus for Mathcited by 67
- 2025SEAL: Steerable Reasoning Calibration of Large Language Models for Freecited by 65
- 2022Reincarnating Reinforcement Learning: Reusing Prior Computation to Accelerate Progresscited by 63
- 2025MegaMath: Pushing the Limits of Open Math Corporacited by 47
- 2025Test-Time Learning for Large Language Modelscited by 46
- 2025Reinforcement Pre-Trainingcited by 40
- 2025Scaling Agents via Continual Pre-trainingcited by 39
- 2025The Choice of Divergence: A Neglected Key to Mitigating Diversity Collapse in Reinforcement Learning with Verifiable Rewardcited by 36
- 2026How Far Can Unsupervised RLVR Scale LLM Training?cited by 26
- 2025How Do LLMs Acquire New Knowledge? A Knowledge Circuits Perspective on Continual Pre-Trainingcited by 22
- 2025A Comprehensive Survey on Trustworthiness in Reasoning with Large Language Modelscited by 22
- 2025Bottom-up Policy Optimization: Your Language Model Policy Secretly Contains Internal Policiescited by 22
- 2025Beyond the Sampled Token: Preserving Candidate Support in RLVRcited by 18
- 2026Proximity-Based Multi-Turn Optimization: Practical Credit Assignment for LLM Agent Trainingcited by 18
- 2025Adaptive Social Learning via Mode Policy Optimization for Language Agentscited by 15
- 2025QFFT, Question-Free Fine-Tuning for Adaptive Reasoningcited by 15
- 2026Automated Optimization Modeling via a Localizable Error-Driven Perspectivecited by 14
- 2025Reinforcement Learning on Pre-Training Datacited by 12
- 2025Tree-based Dialogue Reinforced Policy Optimization for Red-Teaming Attackscited by 8
- 2025PretrainZero: Reinforcement Active Pretrainingcited by 8
- 2026ResAdapt: Adaptive Resolution for Efficient Multimodal Reasoningcited by 8
- 2025RedOne: Revealing Domain-specific LLM Post-Training in Social Networking Servicescited by 6
- 2026Anchored Policy Optimization: Mitigating Exploration Collapse Via Support-Constrained Rectificationcited by 5
- 2025The Zero-Step Thinking: An Empirical Study of Mode Selection as Harder Early Exit in Reasoning Modelscited by 4
- 2026ReMiT: RL-Guided Mid-Training for Iterative LLM Evolutioncited by 4
- 2026Mitigating the Safety-utility Trade-off in LLM Alignment via Adaptive Safe Context Learningcited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Reinforcement Learning with Verifiable Rewards Implicitly Incentivizes Correct Reasoning in Base LLMs
- 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
- 2026Understanding Reasoning from Pretraining to Post-Training
- 2026Reward Modeling for Reinforcement Learning-Based LLM Reasoning: Design, Challenges, and Evaluation
- 2026Gradients Know What Outcomes Don't: Unlocking Reinforcement Learning for LLM Reasoning with Gradient-Aligned Rewards
- 2026Unlocking the Critic: Reward-Free Policy Optimization for LLM Post-Training
- 2026Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs
- 2025RLPR: Extrapolating RLVR to General Domains without Verifiers