Despite the fact that experimental neural scaling laws have substantially guided empirical progress in large-scale machine learning, no existing theory can quantitatively predict the exponents of these important laws for any modern LLM trained on any natural language dataset. We provide the first such theory in the case of data-limited scaling laws. We isolate two key statistical properties of language that alone can predict neural scaling exponents: (i) the decay of pairwise token correlations with time separation between token pairs, and (ii) the decay of the next-token conditional entropy with the length of the conditioning context. We further derive a simple formula in terms of these statistics that predicts data-limited neural scaling exponents from first principles without any free parameters or synthetic data models. Our theory exhibits a remarkable match with experimentally measured neural scaling laws obtained from training GPT-2 and LLaMA style models from scratch on two qualitatively different benchmarks, TinyStories and WikiText.
翻译:尽管实验神经缩放定律在很大程度上指导了大规模机器学习的实证进展,但现有理论尚无法针对任何现代大语言模型在任何自然语言数据集上的训练,定量预测这些重要定律的指数。我们在数据受限的缩放定律情形下提出了首个此类理论。我们分离出语言的两个关键统计特性,仅凭这些特性即可预测神经缩放指数:(i) 词元对相关性随时间间隔的衰减规律,以及 (ii) 下一词元条件熵随上下文长度增加的衰减规律。我们进一步推导出一个基于这些统计量的简洁公式,该公式无需任何自由参数或合成数据模型,即可从第一性原理预测数据受限的神经缩放指数。我们的理论在实验测量的神经缩放定律中表现出显著的一致性——这些实验数据通过从头训练GPT-2和LLaMA架构模型获得,并在两个性质不同的基准数据集(TinyStories和WikiText)上得到验证。