JA EN
LearnLarge Language Models
·★ MEMBER·PAPER·9 min read

Scaling Laws from Scratch — Why Making Models Bigger Makes Them Smarter (and When It Doesn't)

Scaling laws turn 'bigger is smarter' into an equation. This article walks from the Kaplan-era recipe to the Chinchilla paper (Hoffmann et al., 2022) — 400+ training runs, three independent estimation methods, and the 70B-vs-280B showdown — using only the paper itself as evidence.

ModalitytextTaskbasics

Training Compute-Optimal Large Language Models

Primary source — what this article is built on

undefined2022-03-29undefined2026-08-134y 5mo later

Training Compute-Optimal Large Language ModelsJordan Hoffmann, Sebastian Borgeaud, Arthur Mensch et al. · 2022-03-29 · v1arXiv:2203.15556Paper page·PDF
undefined

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.


The budget is fixed. How do you spend it?

Suppose you have exactly 1,000 hours to prepare for an exam. You could master one massive textbook, or work through a thinner one plus a mountain of practice problems. Same total hours, very different outcomes depending on the split.

Training a large language model (LLM) has the same structure. Once you know how many GPUs you have and for how long, your total compute budget — measured in FLOPs, floating-point operations — is essentially fixed in advance. You can spend it on a bigger model or on more training data, but not both: a larger model burns more compute per token, so the same budget buys fewer tokens. And since frontier-scale training runs are effectively one-shot, the split has to be decided before you press go (§1).

The paper that tackled this allocation problem head-on is DeepMind's "Training Compute-Optimal Large Language Models" (Hoffmann et al., 2022), universally known as the Chinchilla paper. This article explains scaling laws from zero, using only that paper as its source.

The Kaplan era: "just make it bigger"

The story starts two years earlier. Kaplan et al. (2020) showed that a power law links a language model's parameter count to its loss: scale the model by orders of magnitude and the loss falls at a predictable rate. After that discovery, the field raced to build ever-larger models (§1, §2).

The Kaplan analysis went further and prescribed an allocation: given a 10× larger compute budget, grow the model 5.5× but the training tokens only 1.8× (§1). In other words, pour almost all of the extra budget into model size. And that is what happened: as the paper's Table 1 shows, GPT-3 (175B), Jurassic-1 (178B), Gopher (280B), and MT-NLG (530B) kept getting bigger while nearly all of them were trained on roughly the same ~300 billion tokens.

The Chinchilla paper's claim, in one sentence: that allocation was wrong, and today's large models are significantly undertrained (Abstract).

Turning the allocation problem into one equation

The paper first states the question as math. Treat the final pre-training loss LL as a function L(N,D)L(N, D) of the parameter count NN and the number of training tokens DD, then minimize it under a compute constraint CC (§1).

Nopt(C), Dopt(C)=argminN,D s.t. FLOPs(N,D)=CL(N,D)N_{opt}(C),\ D_{opt}(C) = \operatorname*{argmin}_{N,\,D\ \text{s.t.}\ \mathrm{FLOPs}(N,D)=C} L(N,D)
(1)

Underneath the notation this is a menu problem, which says: of every pairing of brain size and reading list you can afford, take the one that ends up predicting text best.

Reading the symbols one at a time: NN is the model's parameter count (the size of the brain), DD is the number of tokens it reads during training, and CC is the total compute available. The whole expression just says: among all (N,D)(N, D) pairs that cost exactly CC FLOPs, find the pair with the lowest loss. The functions NoptN_{opt} and DoptD_{opt} are the optimal model size and data amount for that budget.

A handy approximation makes the constraint concrete (§3.3):

C6NDC \approx 6ND

Total training compute is roughly "parameters × tokens × 6" (the 6 comes from 2 FLOPs per parameter for the forward pass plus twice that for the backward pass; Appendix F checks this carefully and finds the approximation accurate to within a few percent). Fix the budget CC and NN and DD become inversely proportional — which is exactly what makes this an allocation problem. Said in words: on a fixed budget, every parameter you add is paid for with tokens the model will never get to read.

FIG 1A loss over two variables forms a terrain you can draw with contour lines. What the Chinchilla paper did was measure this terrain with model size N and data D as the axes, then search along a fixed-compute line for the lowest point. Use this figure to build intuition for "loss as a landscape over two knobs."

The catch is that nobody knows the shape of . So the paper trained over 400 models, from 70 million to over 16 billion parameters, on 5 billion to hundreds of billions of tokens, and estimated the optimal allocation three independent ways (Abstract, §3).

What's behind this

§

Members-only from here

371 walkthroughs, 26 textbook chapters, 48 student units and 6 close readings — all included for $4.99/mo, with three new explainers every day. Cancel any time; access runs to the end of the period.

Already a member? Sign in to keep reading

References

  1. Jordan Hoffmann, Sebastian Borgeaud, Arthur Mensch, Elena Buchatskaya et al.. (2022-03-29) Training Compute-Optimal Large Language Models. arXiv:2203.15556Paper page·PDF

This article is written from the source paper above. Where they differ, the original is authoritative.

Comments

Sign in to comment