通过约束训练与推理差异,提升大模型强化学习的稳定性与效率。
Reformulate LLM Reinforcement Learning for Efficient Training under Black-box Discrepancy

- 将差异控制纳入马尔可夫决策过程,实现双目标协同优化。
- 在8B和30B模型上显著提升性能,支持高低配置环境混合训练。
- 动态调节权重机制,自动平衡探索与偏差控制,避免训练崩溃。
强化学习作为大模型后训练的关键范式,常因训练-推理不一致导致性能下降甚至训练崩溃。我们发现,当提供合适的学习信号时,训练策略可主动校正此类差异。实证揭示存在一个差异容忍区间:在此区间内过度缩小差异会抑制探索、降低效率;而超出该区间时,减小过大的差异能增强优化一致性并提升局部性能上限。基于此,我们提出受限于差异的马尔可夫决策过程(DCMDP),将奖励最大化与训练-推理行为对齐约束结合,实现稳定双目标优化。引入拉格朗日松弛机制,根据当前差异违反程度动态调整两项目标权重,使策略在容忍区内自由探索,在超出安全边界时被引导回归。实验表明,该方法显著提升了8B稠密模型(Qwen-3-8b)和30B混合专家模型(Qwen-3-30bA3b)的性能,并支持异构训练范式——大模型可在高保真训练环境中优化,同时显式适配低成本、资源受限的推理部署。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。