Note
Scaling Laws 揭示了模型、数据与计算量的什么关系?
解释尺度定律及其对参数规模、数据规模和训练预算分配的指导意义。
什么是Scaling Laws?它揭示了模型性能、计算量和数据量之间的什么关系?这对LLM的研发有什么指导意义?
什么是Scaling Laws?
Scaling Laws(尺度定律)是由OpenAI、DeepMind等机构通过大量实验发现的一系列经验性规律。它揭示了大型语言模型的性能(通常以交叉熵损失函数Loss来衡量)与三个关键资源要素——模型参数规模(N)、训练数据集大小(D)和训练所用的计算量(C)——之间存在着可预测的幂律关系(Power-Law Relationship)。
揭示了什么关系?
- 性能的可预测性: Scaling Laws表明,模型的性能损失会随着N、D、C的增加而平滑地、可预测地下降。这种关系可以用一个幂律公式来描述,例如,当数据和计算量足够时,模型损失 L 与模型参数量 N 的关系大致为: ,其中 是一个小的正指数。这意味着我们可以通过在小规模模型上的实验结果,来外推(predict)更大规模模型可能达到的性能。
- 瓶颈效应: 模型的最终性能会被N、D、C中最受限的那个因素所制约。如果仅仅增加模型大小而不增加数据量,性能提升会很快达到瓶颈;反之亦然。为了有效提升模型性能,必须协同扩展这三个要素。
- 资源的最优分配: 对于一个给定的计算预算(FLOPs),存在一个最优的模型大小(N)和数据量(D)的组合。DeepMind的Chinchilla论文是一个里程碑式的发现,它修正了早期认为应该优先扩大模型规模的观点,指出为了达到计算最优,模型参数量和训练数据量应该近似1:20的比例进行扩展。例如,训练一个70B参数的模型,大约需要1.4万亿个token的数据。
对LLM研发的指导意义:
- 科学指导项目规划: 在投入数百万甚至数千万美元进行一次大规模训练之前,研究机构可以先通过小规模实验拟合出自己数据集和模型架构下的Scaling Law。这使得他们能够科学地预测最终模型的性能,评估项目的投资回报率,并合理申请计算资源。
- 优化资源配置,避免浪费: Scaling Laws,特别是Chinchilla定律,为如何高效使用计算预算提供了明确的指导。它告诉我们,与其训练一个参数巨大但数据不足的模型(over-trained),不如用同样的算力去训练一个参数稍小但数据更充分的模型(under-trained),后者效果可能更好。这促使业界从单纯追求“大参数”转向“大参数与大数据的平衡”。
- 强调数据的重要性: Scaling Laws的发现,让学术界和工业界都更加深刻地认识到,高质量、大规模的训练数据和模型参数规模同等重要,甚至在某些阶段更为关键。这推动了数据工程、数据清洗和高质量合成数据生成等领域的发展。