用国际象棋测试模型从预训练到强化学习的推理进化过程。
Understanding Reasoning from Pretraining to Post-Training

- 以国际象棋为可控实验场,研究预训练与强化学习的协同关系。
- 预训练损失越低、预训练数据越多,强化学习效果越好且提升更快。
- 强化学习在难题中发现模型原本忽略的正确走法,不只强化已有能力。
强化学习(RL)已成为提升大语言模型(LLM)复杂推理能力的核心方法,但其与前置预训练的关系长期被孤立研究。本文通过国际象棋这一可控环境,系统考察了从预训练到后训练的完整流程。我们训练500万至10亿参数的模型,在人类对局数据上进行预训练,再用合成推理轨迹进行监督微调(SFT),最后在可验证奖励的棋题上进行强化学习。结果表明:给定强化学习计算量下,模型后置性能可由预训练损失准确预测;强化学习收益随预训练数据量近似线性增长。进一步分析发现,强化学习并非简单放大监督微调策略:在简单题目中放大原有偏好动作,在困难题目中激活原几乎不存在的正确走法。该模式在数学领域10亿参数模型上同样成立,长预训练模型具有更高后置强化学习性能并更快收敛。本研究提供了一个定量理解预训练-强化学习接口的框架,也为推理研究提供了可复现的控制实验平台。
原文摘要 · Abstract (English)
Reinforcement learning (RL) has become central to improving large language models (LLMs) on complex reasoning tasks, yet RL post-training is largely studied in isolation from the pretraining that precedes it. As a result, two basic questions remain open: (1) how do pretraining choices (model size, data) shape the returns to RL compute, and (2) what does RL actually do to the model? These questions are difficult to study in the standard LLM setting: pretraining corpora are vast and uncontrolled, making it hard to attribute behaviors to pretraining versus RL, and systematic compute sweeps across both stages are prohibitively expensive. To address these challenges, we use chess as a controlled testbed for studying reasoning across the full pretraining-to-post-training pipeline. We follow the standard LLM training pipeline by pretraining language models from 5M to 1B parameters on human chess games, supervised fine-tuning on synthetic reasoning traces, and running RL on chess puzzles with verifiable rewards. Using this framework, we find that the post-RL performance at given RL compute level is well-predicted from the pretraining loss, and slope of the RL reward curves improves approximately linearly with the pretraining tokens. Beyond scaling, we find that RL does not simply sharpen the SFT policy: on easy puzzles it amplifies correct moves the SFT policy already preferred, while on hard puzzles it surfaces correct moves that were nearly absent under SFT. We further test whether our findings transfer beyond chess by training a 1B language model on math-domain text, where the same predictive pattern emerges: longer-pretrained checkpoints reach higher post-RL performance and improve faster under RL. In sum, we provide a quantitative account of the pretraining-to-RL interface and a controlled testbed for studying the science of reasoning across the full pretraining-to-post-training pipeline.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。