T 5: Twin-Critic Training for Token-Level Thoughts in Reinforcement Mid-Training
cs.AI
Submitted: 2026-09-26
Updated: 2026-10-02
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: Reinforcement mid-training lets language models learn internal thoughts from unlabeled text, but efficient token-level credit assignment remains challenging.
Terminology
Abstract
Reinforcement mid-training lets language models learn internal thoughts from unlabeled text, but efficient token-level credit assignment remains challenging. Existing group-relative methods require costly repeated generation. Learned critics offer single-rollout feedback, but accurate return prediction alone does not ensure reliable policy updates. Our analysis shows how training--inference mismatch and PPO clipping prevent a common offset in advantage estimates from cancelling out, introducing additional update drift. We propose, a twin-critic method that calibrates token-level advantages from a single generated trajectory. After warmup and held-out qualification, the critics provide two advantage estimates, combined using action-dependent weights learned through a conditional-moment saddle-point objective. This objective brings the average advantage at each prefix toward zero, while a signal-retention constraint prevents the correction from erasing the learning signal. Sharing information across text positions avoids repeated sampling of each prefix. Theoretically, we characterize optimal mixing under the signal-retention constraint and establish an upper bound on residual mean-induced drift. Experiments show that, compared with the state-of-the-art critic-free method, improves mean benchmark performance by 7.8% and reduces mean training-step time by up to 63.4%.
Sources
- PIQA: Reasoning about Physical Commonsense in Natural Language
- Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challenge
- Training Verifiers to Solve Math Word Problems
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Reinforcement Pre-Training
- RLP: Reinforcement as a Pretraining Objective
- Single-Rollout Asynchronous Optimization for Agentic Reinforcement Learning
- REINFORCE++: Stabilizing Critic-Free Policy Optimization with Global Advantage Normalization
- Fast Quiet-STaR: Thinking Without Thought Tokens
- Reinforcement Learning on Pre-Training Data
- Understanding R1-Zero-Like Training: A Critical Perspective
- Beyond Uniform Token-Level Trust Region in LLM Reinforcement Learning
- Score Centering Stabilizes Off-policy Reinforcement Learning
- OpenWebMath: An Open Dataset of High-Quality Mathematical Web Text
- The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scale
- Less is More: Clustered Cross-Covariance Control for Offline RL
- High-Dimensional Continuous Control Using Generalized Advantage Estimation
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Reinforcement Mid-Training
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