arXiv:2509.24375cs.CL2025-09被引 3

提出强化中训练框架,显著提升大模型性能并减少推理步骤

Reinforcement Mid-Training

  • 引入动态令牌预算机制,避免模型过度思考
  • 通过课程式采样实现从易到难的渐进学习,性能提升64.91%
  • 融合强化学习与下一词预测,充分挖掘令牌信息,适合模型优化者

大型语言模型的发展通常分为预训练和后训练两个阶段。本文指出需要增加一个中间阶段——强化中训练,以实现显著性能提升。针对训练效率低、令牌熵分布不均、令牌信息利用不足三大挑战,提出RMT框架。该框架包含动态令牌预算机制,限制冗余推理步骤;课程式自适应采样方法,引导从简单到复杂的渐进学习;以及结合强化学习与下一词预测的双重训练策略,确保关键令牌的精准学习并充分利用所有信息。大量实验表明,RMT在语言建模任务中仅需21%的推理长度,性能最高提升64.91%。此外,强化中训练得到的检查点可显著增强后续后训练,在数学领域最多提升18.76%。

原文摘要 · Abstract (English)

The development of state-of-the-art large language models is commonly understood as a two-stage process involving pre-training and post-training. We point out the need for an additional intermediate stage called reinforcement mid-training with potential for strong performance gains. In this paper, we formally define the problem and identify three key challenges: (1) inefficient training due to excessive reasoning steps, (2) disregard of the imbalanced token entropy distribution, and (3) underutilization of token information. To address these challenges, we propose RMT, a framework for efficient, adaptive, and unified reinforcement mid-training with various innovative components. In particular, we first introduce a dynamic token budget mechanism that constrains unnecessary reasoning steps and mitigates model overthinking. Next, we design a curriculum-based adaptive sampling method that fosters a progressive learning trajectory from easy to hard tokens. Finally, we present a dual training strategy that combines reinforcement learning with next-token prediction, ensuring targeted learning on key tokens and full exploitation of all token information. Extensive experiments demonstrate the superiority of RMT over state-of-the-art methods, achieving up to +64.91% performance improvement with only 21% of the reasoning length in language modeling. We also show that checkpoints obtained after reinforcement mid-training can benefit the subsequent post-training, yielding up to +18.76% improvement in the mathematical domain.

大模型强化学习训练优化推理效率

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。