arXiv:2510.01265cs.LGcs.AI2025-10被引 23

用强化学习提前训练模型自主思考,提升推理能力。

RLP: Reinforcement as a Pretraining Objective

  • 在预训练阶段引入强化学习,让模型先思考再预测
  • 数学与科学任务平均提升19%,大模型效果更显著
  • 无需验证器即可高效训练,适合各类模型规模

当前大型推理模型通常先通过海量数据的下一个词预测进行预训练,再在后期通过监督微调和强化学习进行后训练。本文提出一种信息驱动的强化预训练目标RLP,将强化学习的核心思想——探索——引入预训练末期。核心思路是将思维链视为探索性动作,奖励基于其对预测未来词带来的信息增益。具体而言,奖励衡量在结合上下文与采样推理链后,下一词对数似然相比仅用上下文时的提升。该方法生成无需验证器的密集奖励信号,支持在整个文档流中高效预训练。实验显示,在Qwen3-1.7B-Base上使用RLP预训练,八项数学与科学基准平均分提升19%;在相同后训练下,推理类任务如AIME25和MMLU-Pro表现更优。应用于Nemotron-Nano-12B-v2时,整体平均分从42.81%升至61.32%,科学推理平均提升23%,表明该方法在不同架构和规模下均具可扩展性。

原文摘要 · Abstract (English)

The dominant paradigm for training large reasoning models starts with pre-training using next-token prediction loss on vast amounts of data. Reinforcement learning, while powerful in scaling reasoning, is introduced only as the very last phase of post-training, preceded by supervised fine-tuning. While dominant, is this an optimal way of training? In this paper, we present RLP, an information-driven reinforcement pretraining objective, that brings the core spirit of reinforcement learning -- exploration -- to the last phase of pretraining. The key idea is to treat chain-of-thought as an exploratory action, with rewards computed based on the information gain it provides for predicting future tokens. This training objective essentially encourages the model to think for itself before predicting what comes next, thus teaching an independent thinking behavior earlier in the pretraining. More concretely, the reward signal measures the increase in log-likelihood of the next token when conditioning on both context and a sampled reasoning chain, compared to conditioning on context alone. This approach yields a verifier-free dense reward signal, allowing for efficient training for the full document stream during pretraining. Specifically, RLP reframes reinforcement learning for reasoning as a pretraining objective on ordinary text, bridging the gap between next-token prediction and the emergence of useful chain-of-thought reasoning. Pretraining with RLP on Qwen3-1.7B-Base lifts the overall average across an eight-benchmark math-and-science suite by 19%. With identical post-training, the gains compound, with the largest improvements on reasoning-heavy tasks such as AIME25 and MMLU-Pro. Applying RLP to the Nemotron-Nano-12B-v2 increases the overall average from 42.81% to 61.32% and raises the average on scientific reasoning by 23%, demonstrating scalability across architectures and model sizes.

强化学习预训练推理增强思维链

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