Papers
arxiv:2606.29526

The Mirage of Optimizing Training Policies: Monotonic Inference Policies as the Real Objective for LLM Reinforcement Learning

Published on Jun 28
· Submitted by
Jing Liang
on Jul 6
#1 Paper of the day
Authors:
,
,
,
,
,
,
,
,
,
,
,

Abstract

Training-inference mismatch in reinforcement learning for large language models leads to instability, which is addressed through a new policy optimization objective and framework that ensures consistent policy improvements between training and inference phases.

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.

Community

Paper submitter

We introduce MIPI (Monotonic Inference Policy Improvement) and its instantiation MIPU, a two-step RL framework for large language models under training–inference mismatch.

fig1

As illustrated in the figure, standard LLM RL methods optimize training-side surrogates, which do not necessarily translate into improvements of the deployed inference policy due to persistent engine-level mismatch (e.g., FP8 quantization or backend inconsistency).

MIPU addresses this by explicitly decomposing inference-policy improvement into (i) training-side update and (ii) post-synchronization inference gap. Step 1 constructs sampler-referenced policy updates to improve candidate quality, while Step 2 performs inference-gap-aware acceptance to filter updates that do not yield consistent improvements after synchronization.

Under FP8-quantized rollout, MIPU achieves both improved reasoning performance and significantly enhanced training stability across Qwen3-1.7B and Qwen3-4B.

The figure summarizes the core idea: instead of optimizing only the training policy, MIPU directly aligns optimization with inference-policy improvement under real deployment mismatch.

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2606.29526
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2606.29526 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2606.29526 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2606.29526 in a Space README.md to link it from this page.

Collections including this paper 1