用大模型自评数据质量,提升预训练效果。
DataMan: Data Manager for Pre-training Large Language Models
- 让大模型自我判断数据质量与领域,自动标注4470亿词元数据。
- 用数据管理器筛选300亿词元训练小模型,性能超过基线且用更少数据。
- 发现质量评分与困惑度相关性低,适合需要高质量数据的场景。
大规模语言模型的性能随数据量增长而涌现,使得预训练数据选择愈发关键。现有方法依赖有限启发式和人工直觉,缺乏系统指导。受逆向思维启发,我们让大模型自我识别对性能有益的标准。基于困惑度(PPL)异常成因,提炼出14项质量标准,并引入15个常见应用领域以支持领域混合。本文训练了数据管理器(DataMan),通过逐项评分学习质量评级与领域识别,对4470亿词元的预训练语料库进行标注,涵盖14项质量评分与领域类型。实验验证:使用DataMan筛选300亿词元训练13亿参数模型,显著提升上下文学习(ICL)、困惑度与指令遵循能力,最优模型(综合得分5)优于使用50%更多数据均匀采样的基线模型。进一步用高分、领域特定数据继续预训练,增强领域内上下文学习能力,验证了其领域混合能力。研究强调质量排序的重要性、多标准互补性及其与困惑度的弱相关性,分析了困惑度与上下文学习性能间的不一致。还深入分析了预训练数据集组成、质量评分分布及原始文档来源。
原文摘要 · Abstract (English)
The performance emergence of large language models (LLMs) driven by data scaling laws makes the selection of pre-training data increasingly important. However, existing methods rely on limited heuristics and human intuition, lacking comprehensive and clear guidelines. To address this, we are inspired by ``reverse thinking'' -- prompting LLMs to self-identify which criteria benefit its performance. As its pre-training capabilities are related to perplexity (PPL), we derive 14 quality criteria from the causes of text perplexity anomalies and introduce 15 common application domains to support domain mixing. In this paper, we train a Data Manager (DataMan) to learn quality ratings and domain recognition from pointwise rating, and use it to annotate a 447B token pre-training corpus with 14 quality ratings and domain type. Our experiments validate our approach, using DataMan to select 30B tokens to train a 1.3B-parameter language model, demonstrating significant improvements in in-context learning (ICL), perplexity, and instruction-following ability over the state-of-the-art baseline. The best-performing model, based on the Overall Score l=5 surpasses a model trained with 50% more data using uniform sampling. We continue pre-training with high-rated, domain-specific data annotated by DataMan to enhance domain-specific ICL performance and thus verify DataMan's domain mixing ability. Our findings emphasize the importance of quality ranking, the complementary nature of quality criteria, and their low correlation with perplexity, analyzing misalignment between PPL and ICL performance. We also thoroughly analyzed our pre-training dataset, examining its composition, the distribution of quality ratings, and the original document sources.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。