arXiv:2410.05639cs.CL2024-10EMNLP被引 7

用大模型自动评分、打标、优化训练数据,提升大模型性能

DecorateLM: Data Engineering through Corpus Rating, Tagging, and Editing with Language Models

  • 用小模型自动评估文本质量并打上分层标签
  • 筛选出450亿高质量、多样化文本用于训练新模型
  • 适合想提升预训练数据质量的研究者和工程团队

大语言模型的性能高度依赖于预训练语料库,但其海量无标注数据的质量难以保证。本文提出DecorateLM,通过语言模型对语料进行评分、打标与格式化编辑,实现数据质量提升。为兼顾效率,我们用大模型构建精标注训练集,并将数据工程知识蒸馏成一个12亿参数的小模型。该方法应用于1000亿token的语料,最终筛选出450亿高质量、高多样性样本,用于训练另一个12亿参数的LMM。实验表明,使用此类高质量数据可显著提升模型性能,验证了该数据工程方法的有效性。

原文摘要 · Abstract (English)

The performance of Large Language Models (LLMs) is substantially influenced by the pretraining corpus, which consists of vast quantities of unsupervised data processed by the models. Despite its critical role in model performance, ensuring the quality of this data is challenging due to its sheer volume and the absence of sample-level quality annotations and enhancements. In this paper, we introduce DecorateLM, a data engineering method designed to refine the pretraining corpus through data rating, tagging and editing. Specifically, DecorateLM rates texts against quality criteria, tags texts with hierarchical labels, and edits texts into a more formalized format. Due to the massive size of the pretraining corpus, adopting an LLM for decorating the entire corpus is less efficient. Therefore, to balance performance with efficiency, we curate a meticulously annotated training corpus for DecorateLM using a large language model and distill data engineering expertise into a compact 1.2 billion parameter small language model (SLM). We then apply DecorateLM to enhance 100 billion tokens of the training corpus, selecting 45 billion tokens that exemplify high quality and diversity for the further training of another 1.2 billion parameter LLM. Our results demonstrate that employing such high-quality data can significantly boost model performance, showcasing a powerful approach to enhance the quality of the pretraining corpus.

数据工程大模型预训练

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