arXiv:2507.12466cs.CLcs.LG2025-07被引 17

用任务数据匹配预训练数据,让大模型性能提升2.1倍

Language Models Improve When Pretraining Data Matches Target Tasks

  • 基于任务样例相似性筛选预训练文本,构建可扩展的数据选择方法
  • 在10^19到10^22 FLOPs范围内,9/10任务性能提升,计算效率提高2.1倍
  • 适用于不同规模模型,尤其大模型需更宽松的过滤策略

每种数据筛选方法本质上都有其目标。实践中,这些目标通常通过基准测试驱动的迭代隐式形成:研究者开发筛选策略,训练模型,测量基准表现,再进行优化。这引出一个自然问题:若将这一优化过程显式化会如何?为此,我们提出基准目标排序(BETR),一种基于与基准训练样例相似性选择预训练文档的方法。BETR将基准样例和预训练文档样本嵌入共享空间,根据相似性对样本打分,并训练轻量级分类器预测全量语料的得分。我们通过训练超过500个模型(涵盖10^19至10^22 FLOPs)并拟合缩放定律进行比较。结果表明,仅通过BETR将预训练数据与评估基准对齐,即可实现相对于DCLM-Baseline 2.1倍的计算乘数(相比未过滤数据提升4.7倍),并在所有规模下9/10任务中表现更优。BETR还具有良好的泛化能力:当针对与评估集不重叠的多样化基准时,仍能匹配或超越基线。缩放分析进一步揭示:模型越大,所需过滤越弱。总体而言,我们的发现表明,直接将预训练数据与目标任务对齐可精准塑造模型能力,强调最优筛选策略必须随模型规模自适应调整。

原文摘要 · Abstract (English)

Every data selection method inherently has a target. In practice, these targets often emerge implicitly through benchmark-driven iteration: researchers develop selection strategies, train models, measure benchmark performance, then refine accordingly. This raises a natural question: what happens when we make this optimization explicit? To explore this, we propose benchmark-targeted ranking (BETR), a simple method that selects pretraining documents based on similarity to benchmark training examples. BETR embeds benchmark examples and a sample of pretraining documents in a shared space, scores this sample by similarity to benchmarks, then trains a lightweight classifier to predict these scores for the full corpus. We compare data selection methods by training over 500 models spanning $10^{19}$ to $10^{22}$ FLOPs and fitting scaling laws to them. From this, we find that simply aligning pretraining data to evaluation benchmarks using BETR achieves a 2.1x compute multiplier over DCLM-Baseline (4.7x over unfiltered data) and improves performance on 9 out of 10 tasks across all scales. BETR also generalizes well: when targeting a diverse set of benchmarks disjoint from our evaluation suite, it still matches or outperforms baselines. Our scaling analysis further reveals a clear trend: larger models require less aggressive filtering. Overall, our findings show that directly matching pretraining data to target tasks precisely shapes model capabilities and highlight that optimal selection strategies must adapt to model scale.

数据筛选预训练模型规模性能提升

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