Reformulate LLM Reinforcement Learning for Efficient Training under Black-box Discrepancy

28d ago · Global · primary source: export.arxiv.org

A team of researchers has reformulated reinforcement learning for large language models to address a persistent train-inference mismatch that can cause training collapses and sub-optimal performance, according to a paper submitted to arXiv in June 2026 [1]. The work targets a hidden discrepancy between the engines and architectures used during training and those used during inference, which recent findings have linked to unpredictable failures in reinforcement learning (RL) post-training pipelines [1][2]. The authors report that a training policy can self-correct this mismatch when given an appropriate learning signal [2]. They further identify a discrepancy tolerance region: aggressively narrowing the discrepancy inside this zone can suppress policy exploration and reduce learning efficiency, while reducing excessive discrepancy outside it improves optimization consistency and raises the achievable local performance ceiling [2]. To operationalize these insights, the researchers formulate the problem as a Discrepancy-Constrained Markov Decision Process, or DCMDP, which couples reward maximization with a constraint that aligns training and inference behavior [1][2]. A Lagrangian relaxation mechanism dynamically adjusts the relative weight of performance improvement and discrepancy control based on the current degree of violation, allowing the policy to explore freely within the tolerance region while guiding it back when the discrepancy exceeds a safe boundary [2]. The approach was tested on an 8-billion-parameter dense model, Qwen-3-8b, and a 30-billion-parameter Mixture-of-Expert model, Qwen-3-30bA3b, with the authors reporting significant performance improvements for both [1][2]. The framework also enables a heterogeneous training paradigm in which large language models can be optimized in high-fidelity training setups while being explicitly aligned for low-cost, resource-constrained inference deployment [2]. Large language models are neural networks trained on vast text corpora for tasks such as generation, summarization, and translation, and they underpin modern chatbots [8]. The paper was posted on arXiv, an open-access repository of electronic preprints that, as of November 2024, receives about 24,000 submissions per month and hosts more than two million articles [6]. Submissions to arXiv are moderated but not peer-reviewed [6].

regulationinfrastructureresearch-paper

Background sources we checked (7)
  • arxiv.org ↗ 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 f…
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
  • info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository [...] # arXivLabs: Showcase [...] arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. [...] While the arXiv team is focused on our core miss…
  • blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
  • en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
  • en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
  • en.wikipedia.org ↗ A large language model (LLM) is a neural network trained on a vast amount of text for natural language processing tasks, especially language generation. LLMs can typically generate, summarize, translate, and analyze text in many contexts, and are a foundational technology behind …

Sources

Spot something wrong? Report an issue