arXiv:2603.18534cs.LG2026-03被引 6

用合成文档提升预训练效率,数据利用率最高提至1.8倍

Data-efficient pre-training by scaling synthetic megadocs

  • 将同一网页的合成重述拼成超长文档,替代多个短文档
  • 在32次生成下,数据效率达1.80倍,长上下文任务表现更优
  • 适合数据受限但算力充足的模型预训练场景

当预训练受限于数据而非算力时,合成数据增强成为有前景的解决方案。本文研究如何设计合成数据算法以实现更优的损失缩放:不仅在有限算力下降低损失,尤其在算力趋近无穷时表现更好。实验表明,将网页数据与合成重述混合预训练,可在不产生过拟合的情况下提升独立同分布验证损失和下游基准准确率。通过最优混合与轮次策略,32次生成/文档时数据效率达到1.48倍。进一步发现,从新视角出发:将同一文档的合成生成物拼接为单一超长文档(megadoc),可显著优于简单重述。两种构建方式——拼接相同文档的重述或插入推理内容拉伸文档——均能改善独立同分布损失、下游任务表现及长上下文损失。在32次生成时,数据效率提升至1.80倍,且随着合成数据增加,超长文档优势持续扩大。结果表明,在数据受限场景下,通过合理设计合成数据算法,可使模型从更多算力中获益。

原文摘要 · Abstract (English)

Synthetic data augmentation has emerged as a promising solution when pre-training is constrained by data rather than compute. We study how to design synthetic data algorithms that achieve better loss scaling: not only lowering loss at finite compute but especially as compute approaches infinity. We first show that pre-training on web data mixed with synthetically generated rephrases improves i.i.d. validation loss on the web data, despite the synthetic data coming from an entirely different distribution. With optimal mixing and epoching, loss and benchmark accuracy improve without overfitting as the number of synthetic generations grows, plateauing near $1.48\times$ data efficiency at 32 rephrases per document. We find even better loss scaling under a new perspective: synthetic generations from the same document can form a single substantially longer megadocument instead of many short documents. We show two ways to construct megadocs: stitching synthetic rephrases from the same web document or stretching a document by inserting rationales. Both methods improve i.i.d. loss, downstream benchmarks, and especially long-context loss relative to simple rephrasing, increasing data efficiency from $1.48\times$ to $1.80\times$ at $32$ generations per document. Importantly, the improvement of megadocs over simple rephrasing widens as more synthetic data is generated. Our results show how to design synthetic data algorithms that benefit more from increasing compute when data-constrained.

数据效率合成数据预训练长文本

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