模型规模与训练数据应同比扩大,才能最优利用计算资源。
Training Compute-Optimal Large Language Models

- 模型参数与训练数据量按相同比例增长,实现计算最优。
- Chinchilla 模型在相同算力下性能超越多个更大模型。
- 适合关注高效训练和部署的AI研究者与工程师。
我们研究在给定计算预算下训练Transformer语言模型的最佳模型规模与训练词元数量。通过训练400多个参数从7000万到超过160亿不等、使用500亿至5000亿词元的数据集的语言模型,发现计算最优训练的关键是:模型规模与训练词元数应同比扩大——每将模型规模翻倍,训练数据也应翻倍。我们据此训练了预测的计算最优模型Chinchilla,其使用与Gopher相同的计算预算,但仅含700亿参数且训练数据为后者的4倍。Chinchilla在大量下游评估任务中均显著优于Gopher(2800亿)、GPT-3(1750亿)、Jurassic-1(1780亿)和Megatron-Turing NLG(5300亿)。此外,其更小的规模使微调和推理所需计算大幅减少,极大促进实际应用。值得一提的是,Chinchilla在MMLU基准上达到67.5%的平均准确率,较Gopher提升超7%。
原文摘要 · Abstract (English)
We investigate the optimal model size and number of tokens for training a transformer language model under a given compute budget. We find that current large language models are significantly undertrained, a consequence of the recent focus on scaling language models whilst keeping the amount of training data constant. By training over 400 language models ranging from 70 million to over 16 billion parameters on 5 to 500 billion tokens, we find that for compute-optimal training, the model size and the number of training tokens should be scaled equally: for every doubling of model size the number of training tokens should also be doubled. We test this hypothesis by training a predicted compute-optimal model, Chinchilla, that uses the same compute budget as Gopher but with 70B parameters and 4$\times$ more more data. Chinchilla uniformly and significantly outperforms Gopher (280B), GPT-3 (175B), Jurassic-1 (178B), and Megatron-Turing NLG (530B) on a large range of downstream evaluation tasks. This also means that Chinchilla uses substantially less compute for fine-tuning and inference, greatly facilitating downstream usage. As a highlight, Chinchilla reaches a state-of-the-art average accuracy of 67.5% on the MMLU benchmark, greater than a 7% improvement over Gopher.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。