Reformulate LLM Reinforcement Learning for Efficient Training under Black-box Discrepancy
Not on the map. Why?
Abstract (arXiv)
Reinforcement Learning (RL) has emerged as a pivotal post-training paradigm, yet it frequently suffers from unpredictable sub-optimum performance or even training collapses. Recent findings attribute these failures to a hidden train-inference discrepancy (or mismatch), stemming from the disparate underlying engines and architecture. We find that the training policy can actively self-correct such a discrepancy when provided with an appropriate learning signal. Then, we further empirically identify a discrepancy tolerance region: within this region, aggressively narrowing the discrepancy can suppress policy exploration and reduce learning efficiency, whereas outside this region, reducing excessive discrepancy improves optimization consistency and raises the achievable local performance ceiling. According to such findings, we formulate this problem as a Discrepancy-Constrained Markov Decision Process (DCMDP), where reward maximization is coupled with a constraint that aligns training-Inference behavior, achieving stable dual-objective optimization. To adaptively balance performance improvement and discrepancy control, we introduce a Lagrangian relaxation mechanism that dynamically adjusts the relative weight of the two objectives according to the current degree of discrepancy violation. This enables stable dual-objective optimization: the policy is allowed to explore freely within the tolerance region, while being guided back when the discrepancy exceeds the safe boundary. Empirically, DCMDP significantly improves the performance of 8B dense model (Qwen-3-8b) and 30B Mixture-of-Expert model (Qwen-3-30bA3b), and enables a heterogeneous training paradigm, where LLMs can be optimized in high-fidelity training setup while being explicitly aligned for low-cost, resource-constrained inference deployment.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 24 references that are in the map. First posted on arXiv 7 Jun 2026.
Builds on
20 of 24 references.
References · 20
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2023Let's Verify Step by Stepcited by 3,598
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
- 2024Qwen2.5-Coder Technical Reportcited by 1,151
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2017Constrained Policy Optimizationcited by 928
- 2025Open-Reasoner-Zero: An Open Source Approach to Scaling Up Reinforcement Learning on the Base Modelcited by 450
- 2023OmniSafe: An Infrastructure for Accelerating Safe Reinforcement Learning Researchcited by 76
8 more
- 2025Defeating the Training-Inference Mismatch via FP16cited by 62
- 2025Every Step Evolves: Scaling Reinforcement Learning for Trillion-Scale Thinking Modelcited by 50
- 2025Stabilizing Reinforcement Learning with LLMs: Formulation and Practicescited by 38
- 2025Trust Region Masking for Long-Horizon LLM Reinforcement Learningcited by 13
- 2025Asymmetric Proximal Policy Optimization: mini-critics boost LLM reasoningcited by 9
- 2026Beyond Precision: Training-Inference Mismatch is an Optimization Problem and Simple LR Scheduling Fixes Itcited by 8
- 2026QuRL: Efficient Reinforcement Learning with Quantized Rolloutcited by 6
- 2026Probing RLVR training instability through the lens of objective-level hackingcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Rethinking Training-Inference Mismatch in LLM Reinforcement Learning: Where It Arises and How to Correct It
- 2026A Step Back: Prefix Importance Ratio Stabilizes Policy Optimization
- 2026Where Should RL Post-Training Compute Go? Model Size, Search, Learning, and Feedback
- 2026EVPO: Explained Variance Policy Optimization for Adaptive Critic Utilization in LLM Post-Training
- 2026COPC: Coupled Off-Policy Correction for Asynchronous LLM Reinforcement Learning
- 2026Unlocking the Critic: Reward-Free Policy Optimization for LLM Post-Training
- 2026Rethinking the Trust Region in LLM Reinforcement Learning
- 2021Continuous Doubly Constrained Batch Reinforcement Learning