arXiv:2507.03253cs.CLcs.AI2025-07被引 19

用程序化编辑提升大规模预训练数据质量,精准高效。

RefineX: Learning to Refine Pre-training Data at Scale from Expert-Guided Programs

  • 通过专家指导的编辑程序,实现文档级细粒度数据优化。
  • 750M模型在轻量评估任务上平均提升2.6%-7.2%,用更少数据达同等效果。
  • 适合需要高质量预训练数据的LLM研发团队使用。

大型语言模型的基础能力深受其预训练语料质量影响。然而,在大规模下提升数据质量仍面临挑战,主要源于精炼效果与处理效率之间的权衡。尽管基于规则的过滤仍是主流,但通常仅在文档层面操作,缺乏对文档内特定内容的精细化处理能力。受ProX等工作的启发,我们提出RefineX,一种通过程序化编辑任务实现大规模、精准化预训练数据精炼的新框架。RefineX能够在保持原始文本多样性和自然性的前提下,高效实现细粒度数据优化。其核心在于将高质量、专家指导的端到端精炼结果提炼为最小化的删除型编辑程序,通过高精度蒸馏流程训练出一个高效可靠的精炼模型,可系统性地提升整个语料库的质量。我们在多个模型规模下进行从头预训练评估,结果表明,RefineX在多种下游任务中均优于基于原始数据、过滤后数据或其它精炼方式训练的模型。在750M模型上,RefineX在LightEval任务上实现2.6%-7.2%的平均提升,并显著减少所需训练词数。进一步分析显示,RefineX在效率与精度上均优于端到端生成和Prox-C等方法。该成果证明RefineX是现代大模型训练流水线中优化预训练数据的可扩展、有效且可靠方案。

原文摘要 · Abstract (English)

The foundational capabilities of large language models (LLMs) are deeply influenced by the quality of their pre-training corpora. However, enhancing data quality at scale remains a significant challenge, primarily due to the trade-off between refinement effectiveness and processing efficiency. While rule-based filtering remains the dominant paradigm, it typically operates at the document level and lacks the granularity needed to refine specific content within documents. Inspired by emerging work such as ProX, we propose $\textbf{RefineX}$, a novel framework for large-scale, surgical refinement of pre-training data through programmatic editing tasks. RefineX enables efficient and fine-grained data refinement while reliably preserving the diversity and naturalness of raw text. The core strength of RefineX lies in distilling high-quality, expert-guided end-to-end refinement results into minimal edit-based deletion programs. This high-precision distillation pipeline is used to train an efficient and reliable refine model that can systematically improve every instance in the corpus at scale. We evaluate RefineX across from-scratch pre-training at multiple model scales and find that it consistently outperforms models trained on raw, filtered, or alternatively refined data across diverse downstream tasks. On the 750M model, RefineX yields 2.6%-7.2% average gains on lighteval tasks, and achieves comparable performance using significantly fewer training tokens. Further analysis shows that RefineX reliably enhances text quality with both high efficiency and precision, outperforming prior approaches such as end-to-end generation and Prox-C. These results position RefineX as a scalable, effective, and reliable solution for optimizing pre-training data in modern LLM pipelines.

数据精炼大模型训练程序化编辑

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