English

Reinforcement Learning on Pre-Training Data

Computation and Language 2025-09-26 v2 Artificial Intelligence Machine Learning

Abstract

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.03.0, 5.15.1, 8.18.1, 6.06.0, 6.66.6, and 5.35.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.

Keywords

Cite

@article{arxiv.2509.19249,
  title  = {Reinforcement Learning on Pre-Training Data},
  author = {Siheng Li and Kejiao Li and Zenan Xu and Guanhua Huang and Evander Yang and Kun Li and Haoyuan Wu and Jiajia Wu and Zihao Zheng and Chenchen Zhang and Kun Shi and Kyrierl Deng and Qi Yi and Ruibin Xiong and Tingqiang Xu and Yuhao Jiang and Jianfeng Yan and Yuyuan Zeng and Guanghui Xu and Jinbao Xue and Zhijiang Xu and Zheng Fang and Shuai Li and Qibin Liu and Xiaoxue Li and Zhuoyu Li and Yangyu Tao and Fei Gao and Cheng Jiang and Bo Chao Wang and Kai Liu and Jianchen Zhu and Wai Lam and Wayyt Wang and Bo Zhou and Di Wang},
  journal= {arXiv preprint arXiv:2509.19249},
  year   = {2025}
}

Comments

Work in progress

R2 v1 2026-07-01T05:52:32.368Z