神经语言模型的缩放定律

Scaling Laws for Neural Language Models

贾里德·卡普兰 Jared Kaplan · OpenAI · 2020-01-23 · arXiv:2001.08361 ↗ · 被引 8416

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们研究了语言模型在交叉熵损失上的经验缩放定律。损失随模型大小、数据集大小以及用于训练的计算量呈幂律缩放,某些趋势跨越了七个数量级以上。其他架构细节,如网络宽度或深度,在广泛范围内影响甚微。简单的方程控制了过拟合对模型/数据集大小的依赖性以及训练速度对模型大小的依赖性。这些关系使我们能够确定固定计算预算的最优分配。更大的模型在样本效率上显著更高,因此计算效率最优的训练涉及在相对适中的数据量上训练非常大的模型,并在显著收敛之前停止。

We study empirical scaling laws for language model performance on the cross-entropy loss. The loss scales as a power-law with model size, dataset size, and the amount of compute used for training, with some trends spanning more than seven orders of magnitude. Other architectural details such as network width or depth have minimal effects within a wide range. Simple equations govern the dependence of overfitting on model/dataset size and the dependence of training speed on model size. These relationships allow us to determine the optimal allocation of a fixed compute budget. Larger models are significantly more sample-efficient, such that optimally compute-efficient training involves training very large models on a relatively modest amount of data and stopping significantly before convergence.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 43)

阅读逐段中英对照全文 →