Jing Liang, Hongyao Tang, Yi Ma, Yancheng He, Weixun Wang, Xiaoyang Li, Ju Huang, Wenbo Su, Jinyi Liu, Yan Zheng, Jianye Hao, Bo Zheng
5 min
Modern LLM reinforcement learning (RL) pipelines often use separate engines for training (high precision) and inference (high efficiency). This architectural split leads to a training-inference mismatch, where the training policy and the inference policy assign different probabilities to the same trajectories, even when model parameters are synchronized. The authors investigate whether optimizing the training policy—the standard approach—is actually the correct objective when the inference policy is the one used for deployment.
The authors propose a new principle called Monotonic Inference Policy Improvement (MIPI). Instead of focusing on the training-side surrogate, MIPI targets the monotonic improvement of the inference policy directly. To implement this, they introduce the Monotonic Inference Policy Update (MIPU) framework, which operates in two steps:
Experiments conducted under high-mismatch conditions (using FP8-quantized rollouts) demonstrate that MIPU significantly improves both reasoning performance and training stability compared to standard RL baselines. The authors show that the two-step process effectively decouples the construction of candidate updates from the validation of their deployment-side performance, preventing the model from accepting updates that appear beneficial during training but degrade performance during inference.
Reinforcement learning (RL) has gained growing attention in large language model (LLM) post-training, yet RL training remains fragile and can suffer from instability or collapse. One vital cause is training-inference mismatch: LLM adopts separate inference and training engines for generation efficiency and training precision, which in practice exhibits inconsistent probabilities for the same trajectories on training and inference sides, even with synchronized model parameters. This naturally induces a special type of off-policyness ever existing and poisoning the training. Prior works have made various efforts in addressing the off-policyness to stabilize the training policies under the mismatch. In this paper, we point out the objective misalignment neglected by existing works that an effective update to the policy in the training engine not necessarily ensures the improvement of the inference policy, i.e., the one used in deployment. To this end, we propose a new policy optimization objective for LLM RL, named Monotonic Inference Policy Improvement (MIPI). Following this principle, we introduce Monotonic Inference Policy Update (MIPU), a two-step LLM RL framework that constructs sampler-referenced candidate updates and selectively accepts synchronized candidates using an inference-side gap proxy. Experiments conducted on two model scales under high mismatch show that MIPU improves average reasoning performance and training stability.
Sam: Standard methods like GRPO compare groups of responses to identify the best one. That's useful, but those methods have no awareness of the gap between the training and deployment environments. MIPU takes those same updates and adds the two-step filter on top. The result, according to the researchers, is more stable training and stronger reasoning performance — particularly in settings where the inference engine uses lower-precision arithmetic, which tends to make the mismatch significantly worse.
Alex: If the filter is that selective, doesn't it risk the model becoming too cautious? Only learning from a very narrow slice of updates?
Sam: That's a fair concern. The two steps actually divide the work to address it. Step one improves the direction of each update — making sure the model is learning from the right signal. Step two controls acceptance — making sure only verified improvements get through. Together, they create a controlled trajectory rather than a bottleneck.
Alex: So it's not just being picky. It's being picky for the right reasons.
Sam: Exactly. The researchers actually tested what would happen if you just randomly rejected updates at the same rate MIPU does — which turns out to be roughly seventy percent of the time. Even though that random filter was more conservative overall, the model still eventually collapsed. It couldn't tell the difference between a harmful update and a helpful one. The stability MIPU provides doesn't come from slowing things down. It comes from filtering out the specific risks caused by the mismatch.
Alex: That's a meaningful distinction. It's like a teacher who doesn't just fail most of the class — they identify which students actually understood the material versus which ones got lucky on a particular question.
Sam: That's a fair analogy. The system learns to ignore updates that only appear to work because of a temporary quirk in the training environment. It only moves the model forward when the improvement is real and verifiable in the deployment context.
Alex: Are there downsides? This sounds like it requires considerably more computation.
Sam: That's the main limitation the paper acknowledges. The extra checks require more processing power, and the current research is limited to moderate-scale models. We don't yet know how well this approach scales to the very largest systems. The "post-update gap" measurement is also currently an estimate rather than an exact calculation, so future work will likely focus on making that measurement faster and more precise — or even building it directly into the training process from the start.
Alex: So this isn't just a patch for a specific bug. It's a different way of thinking about what training is actually for.
Sam: That's the core argument. The authors suggest that training-inference mismatch represents a fundamental problem with how we currently define success in policy training. We tend to treat deployment as an afterthought — something to worry about after the model is trained. This paper argues that the deployment environment should be the primary target from the very beginning.
Alex: It's a meaningful shift in perspective. Thanks for walking through the logic, Sam.
Sam: It was a good one to dig into. Thanks for listening to ResearchPod.