arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.09668cs.LGcs.AIcs.CL
收藏

如何训练数据高效的LLMs

How to Train Data-Efficient LLMs

University of California, San Diego(加利福尼亚大学圣迭戈分校);Google DeepMind(谷歌DeepMind);Texas A&M University(德克萨斯农工大学)

Noveen Sachdeva, Benjamin Coleman, Wang-Cheng Kang, Jianmo Ni, Lichan Hong, Ed H. Chi, James Caverlee, Julian McAuley, Derek Zhiyuan Cheng

AI总结:

本文研究预训练LLMs的数据高效方法,提出Ask-LLM利用零样本推理评估数据质量,Density sampling优化覆盖率,实验表明两者在各自类别中最佳,Ask-LLM在拒绝90%数据时仍优于全量训练且收敛快70%。

摘要:

大型语言模型(LLMs)的训练成本高昂。在本文中,我们研究了预训练LLMs的数据高效方法,即旨在优化模型质量与训练资源/数据消耗之间帕累托前沿的技术。我们试图理解基于以下两种方法的数据选择例程的权衡:(i) 计算成本高昂的数据质量估计,以及 (ii) 在特征空间中最大化覆盖率和基于多样性的度量。我们的第一种技术Ask-LLM利用指令微调LLMs的零样本推理能力来直接评估训练样本的质量。为了实现覆盖率,我们提出了Density sampling,该方法对数据分布进行建模以选择多样化的样本。在我们对19个采样器的比较中,涉及数百个评估任务和预训练运行,我们发现Ask-LLM和Density是各自类别中的最佳方法。覆盖率采样能够恢复完整数据的性能,而在Ask-LLM数据上训练的模型始终优于完整数据训练——即使我们拒绝了原始数据集的90%,同时收敛速度提高了70%。

英文摘要:

The training of large language models (LLMs) is expensive. In this paper, we study data-efficient approaches for pre-training LLMs, i.e., techniques that aim to optimize the Pareto frontier of model quality and training resource/data consumption. We seek to understand the tradeoffs associated with data selection routines based on (i) expensive-to-compute data-quality estimates, and (ii) maximization of coverage and diversity-based measures in the feature space. Our first technique, Ask-LLM, leverages the zero-shot reasoning capabilities of instruction-tuned LLMs to directly assess the quality of a training example. To target coverage, we propose Density sampling, which models the data distribution to select a diverse sample. In our comparison of 19 samplers, involving hundreds of evaluation tasks and pre-training runs, we find that Ask-LLM and Density are the best methods in their respective categories. Coverage sampling can recover the performance of the full data, while models trained on Ask-LLM data consistently outperform full-data training -- even when we reject 90% of the original dataset, while converging up to 70% faster.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑