通过重写低质网页文本,提升大模型预训练数据质量与数量
Recycling the Web: A Method to Enhance Pre-training Data Quality and Quantity for Language Models
- 将过滤掉的低质网页文本经引导重写,转化为可用训练数据
- 在3个规模模型上提升1.0至2.5个百分点,超越两倍原始数据
- 适合追求低成本扩展高质量数据的研究者与工程师
Scaling laws 预测大语言模型性能随模型规模和数据量增长而提升。现实中,预训练依赖大规模网络爬取,但自然数据增长远不及算力发展。高质量文本更稀缺:现有过滤流程常丢弃99%原始数据以达到顶尖水平。为突破预训练的数据瓶颈,本文提出 REWIRE(REcycling the Web with guIded REwrite),通过引导重写将被丢弃的低质文档转化为有用训练内容,从而增加合成数据在最终训练集中的比例。在 DCLM 基准的 1B、3B、7B 模型规模上,混合高质量原始文本与重写文本,在22项多样化任务中分别带来1.0、1.3、2.5个百分点的提升,优于仅用过滤数据训练。使用原始-合成数据混合的训练效率,甚至超过拥有2倍原始网络数据的情况。分析显示,约82%的新增文本来自原本将被丢弃的低质文档。REWIRE 在生成合成数据方面也优于维基式改写、问答合成与知识抽取等方法。结果表明,回收网页文本是简单有效的预训练数据扩展路径。高质量合成数据已公开于 https://huggingface.co/datasets/facebook/recycling_the_web。
原文摘要 · Abstract (English)
Scaling laws predict that the performance of large language models improves with increasing model size and data size. In practice, pre-training has been relying on massive web crawls, using almost all data sources publicly available on the internet so far. However, this pool of natural data does not grow at the same rate as the compute supply. Furthermore, the availability of high-quality texts is even more limited: data filtering pipelines often remove up to 99% of the initial web scrapes to achieve state-of-the-art. To address the "data wall" of pre-training scaling, our work explores ways to transform and recycle data discarded in existing filtering processes. We propose REWIRE, REcycling the Web with guIded REwrite, a method to enrich low-quality documents so that they could become useful for training. This in turn allows us to increase the representation of synthetic data in the final pre-training set. Experiments at 1B, 3B and 7B scales of the DCLM benchmark show that mixing high-quality raw texts and our rewritten texts lead to 1.0, 1.3 and 2.5 percentage points improvement respectively across 22 diverse tasks, compared to training on only filtered web data. Training on the raw-synthetic data mix is also more effective than having access to 2x web data. Through further analysis, we demonstrate that about 82% of the mixed in texts come from transforming lower-quality documents that would otherwise be discarded. REWIRE also outperforms related approaches of generating synthetic data, including Wikipedia-style paraphrasing, question-answer synthesizing and knowledge extraction. These results suggest that recycling web texts holds the potential for being a simple and effective approach for scaling pre-training data. We make our high-quality synthetic data publicly available at https://huggingface.co/datasets/facebook/recycling_the_web.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。