arXiv:2410.11840cs.LGcs.AI2024-10ICML被引 23

通过大规模实验优化了模型缩放定律的估计方法,提升预训练决策效率。

A Hitchhiker's Guide to Scaling Law Estimation

  • 利用中间检查点数据拟合缩放定律,显著提高预测精度
  • 相似规模模型间的缩放规律更可靠,小模型多试优于单一大模型
  • 同一架构模型可外推预测性能,适用于新模型快速评估

缩放定律通过从参数量较少或训练集较小的模型中外推,预测目标模型的损失,为研究者和实践者提供高效比较优化器、数据集和模型架构的工具。尽管广泛用于语言模型训练动态建模,但如何最佳估计与解释缩放定律仍缺乏系统研究。本文收集并发布包含485个已发表预训练模型损失及下游评估的大规模数据集,基于此估计超过1000条缩放定律,并总结出新模型族缩放定律估计的最佳实践。发现:使用训练过程中的中间检查点(而不仅是最终损失)可显著提升准确性;在其他条件相同时,基于同规模模型的估计最精确;由于模型种子间存在显著变异性,训练多个小模型有时比训练单个大模型更有用。此外,虽不同模型族缩放行为存在差异,但通常足够相似,使得仅需一个同架构模型及其来自其他模型族的缩放参数估计,即可有效预测目标模型表现。

原文摘要 · Abstract (English)

Scaling laws predict the loss of a target machine learning model by extrapolating from easier-to-train models with fewer parameters or smaller training sets. This provides an efficient way for practitioners and researchers alike to compare pretraining decisions involving optimizers, datasets, and model architectures. Despite the widespread use of scaling laws to model the dynamics of language model training, there has been little work on understanding how to best estimate and interpret them. We collect (and release) a large-scale dataset containing losses and downstream evaluations for 485 previously published pretrained models. We use these to estimate more than 1000 scaling laws, then derive a set of best practices for estimating scaling laws in new model families. We find that fitting scaling laws to intermediate checkpoints of training runs (and not just their final losses) substantially improves accuracy, and that -- all else equal -- estimates of performance are generally most accurate when derived from other models of similar sizes. However, because there is a significant degree of variability across model seeds, training multiple small models is sometimes more useful than training a single large one. Moreover, while different model families differ scaling behavior, they are often similar enough that a target model's behavior can be predicted from a single model with the same architecture, along with scaling parameter estimates derived from other model families.

缩放定律模型评估预训练

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