arXiv:2510.10681cs.CLcs.LG2025-10被引 3

用强化学习重制网络数据,让小模型高效生成高质量预训练文本。

RePro: Training Language Models to Faithfully Recycle the Web for Pretraining

  • 训练小型语言模型通过强化学习重写原始网页内容,保持语义不变。
  • 在22个下游任务中提升4.7%~14.0%准确率,数据效率提升2-3倍。
  • 开源代码与重写数据,适合关注数据增效的LLM研究者使用。

高质量预训练数据是大语言模型的‘化石燃料’,但前沿模型面临资源枯竭。本文提出RePro,一种新型网页数据重用方法:训练一个40亿参数的小型语言模型,利用强化学习生成有效且忠实的改写文本。设计1个质量奖励和3个忠实度奖励,使重写模型在保留原始语义与结构的前提下提升数据质量。实验中,该40亿参数重写器对720亿词元的DCLM-RefinedWeb样本进行重写。在4亿和14亿参数模型上预训练的结果显示,RePro在22项下游任务中相比纯原始数据基线提升4.7%至14.0%的相对准确率。其性能优于当前最优的基于提示的700亿参数重写方法(ReWire),也超越了使用四倍数据池的原始数据基线。不同数据量实验表明,RePro将原始数据效率提升2至3倍。个体与分布分析证实,相较于提示方法,RePro能更完整保留关键信息并忠实反映原始数据特征。整体结果表明,RePro为高效、可控地挖掘大模型预训练‘化石燃料’提供了新路径。代码、重写模型与重写数据已开源:https://github.com/cxcscmu/RePro。

原文摘要 · Abstract (English)

High-quality pretraining data is the fossil fuel of large language models (LLMs), yet its reserves are running low for frontier models. In this paper, we introduce RePro, a novel web recycling method that trains a relatively small LM with reinforcement learning to generate effective and faithful rephrasings of pretraining data. Specifically, we design one quality reward and three faithfulness rewards, optimizing the LM rephraser to convert organic data into high-quality rephrasings while maintaining its core semantics and structure. In our experiment, we train a 4B rephraser to recycle 72B tokens sampled from DCLM-RefinedWeb. Pretraining results on 400M and 1.4B models demonstrate that RePro delivers 4.7%-14.0% relative accuracy gains over organic-only baseline on 22 downstream tasks. RePro also outperforms ReWire, the state-of-the-art web recycling method that prompts a 70B rephraser, as well as the organic baseline with a 4x larger data pool. Experiments with different amounts of recycled data highlight that RePro improves organic data efficiency by 2-3x. Individual and distributional analyses validate that RePro preserves more critical information and faithfully reflects the characteristics of organic data compared to prompting-based methods. Together, these results show that RePro provides an efficient and controllable path to effectively harness the fossil fuel of LLM pretraining. We open-source our code, rephraser, and recycled data at https://github.com/cxcscmu/RePro.

数据重用强化学习预训练大模型

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