arXiv:2509.19249cs.CLcs.AI2025-09ACL被引 12

用预训练数据做强化学习,让大模型自动提升推理能力。

Reinforcement Learning on Pre-Training Data

  • 用文本续写任务生成奖励信号,实现无需人工标注的强化学习。
  • 在多个基准上提升显著,数学推理最高增8.1分。
  • 适合想用算力放大模型推理能力的研究者与工程师。

当前大语言模型的算力增长远超高质量文本数据的增长,制约了传统扩展方法。为此,我们提出在预训练数据上进行强化学习(RLPT),一种新的训练扩展范式。与依赖监督学习的以往方法不同,RLPT使策略能自主探索有意义的文本轨迹,通过强化学习提升能力。不同于需人工标注奖励的RLHF和RLVR,RLPT直接从预训练数据中提取奖励信号,采用后续片段推理目标,奖励模型对上下文条件下后续文本的准确预测。该机制可在大规模预训练数据上扩展强化学习,促进更广泛上下文下的丰富轨迹探索,从而培养更强泛化推理能力。在多个模型和通用、数学推理基准上的实验验证了其有效性:以Qwen3-4B-Base为例,在MMLU、MMLU-Pro、GPQA-Diamond、KOR-Bench、AIME24、AIME25上分别取得3.0、5.1、8.1、6.0、6.6、5.3的绝对提升。结果还显示良好的扩展性,表明算力增加仍可带来持续收益。此外,RLPT为模型提供了坚实基础,拓展了推理边界,并增强了RLVR表现。

原文摘要 · Abstract (English)

The growing disparity between the exponential scaling of computational resources and the finite growth of high-quality text data now constrains conventional scaling approaches for large language models (LLMs). To address this challenge, we introduce Reinforcement Learning on Pre-Training data (RLPT), a new training-time scaling paradigm for optimizing LLMs. In contrast to prior approaches that scale training primarily through supervised learning, RLPT enables the policy to autonomously explore meaningful trajectories to learn from pre-training data and improve its capability through reinforcement learning (RL). While existing RL strategies such as reinforcement learning from human feedback (RLHF) and reinforcement learning with verifiable rewards (RLVR) rely on human annotation for reward construction, RLPT eliminates this dependency by deriving reward signals directly from pre-training data. Specifically, it adopts a next-segment reasoning objective, rewarding the policy for accurately predicting subsequent text segments conditioned on the preceding context. This formulation allows RL to be scaled on pre-training data, encouraging the exploration of richer trajectories across broader contexts and thereby fostering more generalizable reasoning skills. Extensive experiments on both general-domain and mathematical reasoning benchmarks across multiple models validate the effectiveness of RLPT. For example, when applied to Qwen3-4B-Base, RLPT yields absolute improvements of $3.0$, $5.1$, $8.1$, $6.0$, $6.6$, and $5.3$ on MMLU, MMLU-Pro, GPQA-Diamond, KOR-Bench, AIME24, and AIME25, respectively. The results further demonstrate favorable scaling behavior, suggesting strong potential for continued gains with more compute. In addition, RLPT provides a solid foundation, extending the reasoning boundaries of LLMs and enhancing RLVR performance.

强化学习大模型推理增强预训练

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