Beyond Precision: Training-Inference Mismatch is an Optimization Problem and Simple LR Scheduling Fixes It
Not on the map. Why?
Abstract (arXiv)
Reinforcement Learning (RL) for training Large Language Models is notoriously unstable. While recent studies attribute this to "training inference mismatch stemming" from inconsistent hybrid engines, standard remedies, such as Importance Sampling, might fail during extended training runs. In this work, we analyze this instability through the lens of optimization, demonstrating that gradient noise and training-inference mismatch escalate in tandem as training progresses. Meanwhile, we find that the mismatch can be effectively suppressed by shrinking the update size. Taken together, we deduce that the mismatch is not merely a static numerical discrepancy, but a dynamic failure coupled with the model's optimization. Based on this insight, we propose a simple yet effective solution: a specialized Learning Rate (LR) scheduler. Instead of pre-defined decay schedule in traditional LR scheduler, our method dynamically triggers LR decay based on response length, which we identify as a reliable early-warning signal for impending instability. Empirical evidence suggests that by reducing the learning rate as gradient noise rises, we can consistently stabilize RL training and keep the training-inference mismatch at a safe level.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 24 of its 31 references that are in the map. First posted on arXiv 2 Feb 2026.
Builds on
24 of 31 references.
References · 24
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2015Trust Region Policy Optimizationcited by 2,503
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
- 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,107
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
12 more
- 2018IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architecturescited by 866
- 2025Group Sequence Policy Optimizationcited by 788
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 463
- 2023A Long Way to Go: Investigating Length Correlations in RLHFcited by 285
- 2025Logic-RL: Unleashing LLM Reasoning with Rule-Based Reinforcement Learningcited by 260
- 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 248
- 2025Defeating the Training-Inference Mismatch via FP16cited by 62
- 2025Stable Reinforcement Learning for Efficient Reasoningcited by 27
- 2025Shorten After You're Right: Lazy Length Penalties for Reasoning RLcited by 25
- 2024MA-RLHF: Reinforcement Learning from Human Feedback with Macro Actionscited by 15
- 2025Every Attention Matters: An Efficient Hybrid Architecture for Long-Context Reasoningcited by 9
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026RAGEN-2: Reasoning Collapse in Agentic RLcited by 16
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RLcited by 5
- 2026Joint Training of Multi-Token Prediction in Reinforcement Learning via Optimal Coefficient Calibrationcited by 2
- 2026The Mirage of Optimizing Training Policies: Monotonic Inference Policies as the Real Objective for LLM Reinforcement Learningcited by 1
- 2026Off-Policy Value-Based Reinforcement Learning for Large Language Modelscited by 0
- 2026Reformulate LLM Reinforcement Learning for Efficient Training under Black-box Discrepancycited by 0
- 2026MemoryWalker: Stop Training Agents on Contexts They Never Sawcited by 0
- 2026Rethinking Training-Inference Mismatch in LLM Reinforcement Learning: Where It Arises and How to Correct Itcited by 0
- 2026Rethinking Training-Inference Mismatch in LLM Reinforcement Learning: Where It Arises and How to Correct It26 Sep 2026
- 2026MemoryWalker: Stop Training Agents on Contexts They Never Saw1 Sep 2026
- 2026The Mirage of Optimizing Training Policies: Monotonic Inference Policies as the Real Objective for LLM Reinforcement Learning28 Jun 2026
- 2026Reformulate LLM Reinforcement Learning for Efficient Training under Black-box Discrepancy7 Jun 2026
- 2026Joint Training of Multi-Token Prediction in Reinforcement Learning via Optimal Coefficient Calibration27 May 2026
- 2026RAGEN-2: Reasoning Collapse in Agentic RL7 Apr 2026
- 2026Off-Policy Value-Based Reinforcement Learning for Large Language Models24 Mar 2026
- 2026The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL6 Feb 2026