| over 400 models, 70M to over 16B parameters, 5 to 500 billion tokens | scale of the study | [2] Abstract
By training over 400 language models ranging from 70 million to over 16 billion parameters on 5 to 500 billion tokens |
| double the model, double the tokens | the headline result | [2] Abstract
for every doubling of model size the number of training tokens should also be doubled. |
| about 300 billion | tokens of the largest models before Chinchilla | [2] Table 1 caption
Other than LaMDA, most models are trained for approximately 300 billion tokens. |
| eight in C, six in N, two in D | orders of magnitude covered | [1] §1.2
These relations hold across eight orders of magnitude in C_min, six orders of magnitude in N, and over two orders of magnitude in D. |
| an aspect ratio varied by a factor of 40 changes the loss only slightly | depth against width | [1] Fig. 5 caption
Aspect ratio in particular can vary by a factor of 40 while only slightly impacting performance |
| N ≈ 12 n_layer d_model² | parameter count formula | [1] §2.1
= 12 n_layer d_model^2 with the standard d_attn = d_ff/4 = d_model |
| non-embedding parameters | definition of N | [1] §1.3
N -- the number of model parameters, excluding all vocabulary and positional embeddings |
| cleaner laws | why embeddings are excluded | [1] §2.1
we do not include these when discussing the model size N; we will see that this produces significantly cleaner scaling laws. |
| in parameters and FLOPs | embeddings counted | [2] App. F
We include all training FLOPs, including those contributed to by the embedding matrices, in our analysis. Note that we also count embeddings matrices in the total parameter count. |
| non-embedding versus total parameters, at small scale | Pearce and Song's claim | [4] Abstract
This paper finds that much of this discrepancy can be attributed to Kaplan counting non-embedding rather than total parameters, combined with their analysis being performed at small scale. |
| tied embeddings, perhaps | why the head was left out of the compute | [5] §3.2
Perhaps because their model head and embeddings had tied weights, this led them to also discount the contribution of the model head to the model's FLOPs per token |
| exponent falls by more than 0.1 | effect of counting the head | [5] §3.2
switching to our model size count also reduces the exponent a by more than 0.1 |
| 1,024 tokens | context length | [1] §2
We optimize the autoregressive log-likelihood (i.e. cross-entropy loss) averaged over a 1024-token context, which is also our principal performance metric. |
| improve in parallel with WebText2 | other text distributions | [1] §3.2.2
We see that the loss on these other data distributions improves smoothly with model size, in direct parallel with the improvement on WebText2. |
| 6N | compute per training token | [1] §2.1
we then define the estimated non-embedding compute as C ≈ 6 N floating point operators per training token. |
| 2 d_model n_vocab | de-embedding FLOPs | [1] Table 1
De-embed & --- & 2 d_model n_vocab |
| d_model > n_ctx/12 | when the context term is negligible | [1] §2.1
For contexts and models with d_model > n_ctx / 12, the context-dependent computational cost per token is a relatively small fraction of the total compute. |
| 10% at large models, 90% at small | the head's share of FLOPs | [5] §3.2
not accounting for the model head leads to under-approximation that grows smoothly as model size decreases, from roughly 10% at larger models to roughly 90% at smaller models. |
| 0.003239 − 0.0001395 log N | learning-rate rule | [1] App. D.6
LR(N) ≈ 0.003239 + -0.0001395 log(N) |
| Adam, 2.5×10⁵ steps, 512 sequences of 1,024 tokens | training recipe | [1] §2.2
Unless otherwise noted, we train models with the Adam optimizer for a fixed 2.5 × 10^5 steps with a batch size of 512 sequences of 1024 tokens. |
| 3,000-step warmup, cosine to zero | learning-rate schedule | [1] §2.2
all training runs included in our data used a learning rate schedule with a 3000 step linear warmup followed by a cosine decay to zero. |
| one schedule for all models | the first difference from Kaplan | [2] §2
First, the authors use a fixed number of training tokens and learning rate schedule for all models; this prevents them from modelling the impact of these hyperparameters on the loss. |
| 4 horizons per size, 16× range | Approach 1 design | [2] §3.1
For each parameter count N we train 4 different models, decaying the learning rate by a factor of 10× over a horizon (measured in number of training tokens) that ranges by a factor of 16×. |
| 9 FLOP counts from 6×10¹⁸ to 3×10²¹ | Approach 2 design | [2] §3.2
(ranging from 6 ×10^18 to 3 ×10^21 FLOPs) |
| less than one epoch | single epoch | [2] §6
Finally, the training runs for our analysis have all been trained on less than an epoch of data |
| 10% dropout and early stopping | regularisation for the data-limited runs | [1] §4.2
We regularize all our models with 10% dropout, and by tracking test loss and stopping once it is no longer decreasing. |
| poor for the smallest D | small-data fits | [1] App. C
We did not thoroughly investigate the small data regime, and our fits for L(N,D) were poor for the smallest values of D |
| all models, not truly optimal | fixed batch size | [1] §3.2
Note that in these results the batch size B remains fixed for all models, which means that these empirical results are not truly optimal. |
| noise scale rises as the loss falls | the critical batch size | [9] Abstract
We find that the noise scale increases as the loss decreases over a training run and depends on the model size primarily through improved model performance. |
| trained to convergence on sufficiently large datasets | L(N) is defined at convergence | [1] §1.2
For models with a limited number of parameters, trained to convergence on sufficiently large datasets: |
| 0.076 | exponent of L(N) | [1] App. A, key parameters
alpha_N=0.076 & N_c=8.8×10^13 params (non-embed) |
| power-law generalization error scaling | power laws before Kaplan et al. | [10] Abstract
Our empirical results show power-law generalization error scaling across a breadth of factors |
| non-zero entropy | the trends must level off | [1] §6.3
Our trends must eventually level off, though, since natural language has non-zero entropy. |
| C* ~ 10⁴ PF-days, N* ~ 10¹², D* ~ 10¹², L* ~ 1.7 nats | where the laws must break | [1] §6.3
C^* sim 10^4 PF-Days, N^* sim 10^12 parameters, D^* sim 10^12 tokens, L^* sim 1.7 nats/token |
| an order of magnitude either way | uncertainty of that point | [1] §6.3
though the numerical values are highly uncertain, varying by an order or magnitude in either direction depending on the precise values of the exponents from the power-law fits. |
| α_N 0.076, α_D 0.103, N_c 6.4×10^13, D_c 1.8×10^13 | fit of L(N, D) | [1] §4.2
Value & 0.076 & 0.103 & 6.4 × 10^13 & 1.8 × 10^13 |
| three principles | the L(N, D) ansatz | [1] §4.1
We conjecture that this functional form may also parameterize the trained log-likelihood for other generative modeling tasks. |
| 8× the model, about 5× the data | data needed per model growth | [1] §1.1
every time we increase the model size 8x, we only need to increase the data by roughly 5x to avoid a penalty. |
| 5×; data 2× | model growth per 10× compute | [1] Fig. 14 caption
Optimal model size grows very rapidly with C_min, increasing by 5x for each 10x increase in compute. The number of data examples processed makes up the remainder of the increase, growing relatively modestly by only 2x. |
| 0.73 | optimal model size exponent | [1] App. A, compute-efficient training
N_opt = N_e·C_min^p_N & p_N = 0.73 & N_e = 1.3·10^9 params |
| 0.27 | optimal data exponent | [1] App. A, compute-efficient training
D_opt = D_e·C_min^p_D (1 epoch) & p_D = 0.27 & D_e = 2·10^10 tokens |
| 0.03 | optimal steps exponent | [1] App. A, compute-efficient training
S_min = S_e·C_min^p_S (lower bound) & p_S = 0.03 & S_e = 5.4·10^3 steps |
| stop significantly short of convergence | convergence is inefficient | [1] §1.1
we attain optimal performance by training very large models and stopping significantly short of convergence |
| big models over big data | the paper's conclusion | [1] §8
Big models may be more important than big data. |
| 5.5× model, 1.8× tokens | what Kaplan implies for 10× compute | [2] §1
Specifically, given a 10× increase computational budget, they suggests that the size of the model should increase 5.5× while the number of training tokens should only increase 1.8×. |
| 0.050 | exponent of L(C_min) | [1] App. A, key parameters
alpha_C^min=0.050 & C_c^min=3.1×10^8 PF-days |
| 0.054 | exponent predicted from L(N, S_min) | [1] §6.2
α_C^min ≡ 1/(1/α_S + 1/α_B + 1/α_N) ≈ 0.054 |
| 0.052 | the same exponent in the appendix | [1] App. B.2
α_C = 1/(1/α_S + 1/α_B + 1/α_N) ≈ 0.052 |
| Figure X | unresolved figure reference | [1] App. B.4
The result is shown in Figure X. |
| 1,500 log-spaced FLOP values | Approach 1 evaluation points | [2] §3.1
At 1500 logarithmically spaced FLOP values, we find which model size achieves the lowest loss of all models along with the required number of training tokens. |
| within the last 15% of training | where the envelope points sit | [2] §3.1 footnote
Note that all selected points are within the last 15% of training. |
| a = 0.50, b = 0.50 | Approach 1 exponents | [2] Table 2
1. Minimum over training curves & 0.50 (0.488, 0.502) & 0.50 (0.501, 0.512) |
| a parabola per budget | IsoFLOP minima | [2] §3.2
We fit a parabola to each IsoFLOPs curve to directly estimate at what model size the minimum loss is achieved |
| a = 0.49, b = 0.51 | Approach 2 exponents | [2] Table 2
2. IsoFLOP profiles & 0.49 (0.462, 0.534) & 0.51 (0.483, 0.529) |
| bootstrap, 80% resampled 100 times | how the intervals were made | [2] Table 2 caption
The 10th and 90th percentiles are estimated via bootstrapping data (80% of the dataset is sampled 100 times) and are shown in parenthesis. |
| about 0.02 | seed-to-seed spread | [1] §4.2
We estimate that the variation in the loss with different random seeds is roughly 0.02 |
| E + A/N^α + B/D^β | parametric form | [2] §3.3
L̂(N,D) ≜ E + A/N^α + B/D^β |
| entropy of natural text | meaning of E | [2] §3.3
The first term captures the loss for an ideal generative process on the data distribution, and should correspond to the entropy of natural text. |
| a = β/(α+β), b = α/(α+β) | closed-form frontier | [2] §3.3
a = β/(α+β), and b = α/(α + β) |
| δ = 10⁻³ | Huber threshold | [2] §3.3, App. D.2
We use delta = 10^-3 for the Huber loss. |
| a = 0.46, b = 0.54 | Approach 3 exponents | [2] Table 2
3. Parametric modelling of the loss & 0.46 (0.454, 0.455) & 0.54 (0.542, 0.543) |
| 0.33917084 for the parameter slope | unrounded exponent, left in a TeX comment | [2] main.tex, after App. D.2
'param_slope': array(0.33917084, dtype=float32), |
| 0.2849083 for the token slope | unrounded token exponent, left in a TeX comment | [2] main.tex, after App. D.2
'token_slope': array(0.2849083, dtype=float32) |
| β = 0.2849, reported as 0.28 | rounding in the body of the paper | [13] §3
The true value of the data exponent beta should be around 0.2849 according to the comments in the TeX source code |
| averaged, not summed, Huber loss | why the original fit stopped early | [13] §3.1
they averaged the Huber loss values over different data points instead of summing them during loss minimization and the high loss scale of their optimizer caused their optimization to terminate early |
| E 1.8172, A 482.01, B 2085.43, α 0.3478, β 0.3658 | their refit | [13] §3
L(N, D) = 1.8172 + 482.01/N^0.3478 + 2085.43/D^0.3658 |
| comparable predictions | the three approaches agree | [2] §3.4
We find that the three approaches, despite using different fitting methodologies and different trained models, yield comparable predictions for the optimal scaling in parameters and tokens with FLOPs |
| treated as outliers by the Huber loss | low-compute points | [2] §3.4
The fitted model places increased weight on the points with more FLOPs---automatically considering the low-computational budget points as outliers due to the Huber loss. |
| concave at high compute | curvature of the frontier | [2] §6
we observe some concavity in log(N_opt) at high compute budgets |
| left as future work | curvature not taken into account | [2] App. E
In this work, we do not take this in to account and we leave this as interesting future work |
| 20.2 billion tokens | tokens per parameter at 1B | [2] Table 3
1 Billion & 1.21e+20 & 1/4,761 & 20.2 Billion |
| smaller and longer than Chinchilla-optimal | inference-aware sizing | [14] Abstract
LLM researchers expecting reasonably large inference demand (~1B requests) should train models smaller and longer than Chinchilla-optimal. |
| quality keeps improving up to 10,000 | extreme tokens per parameter | [14] Abstract
model quality continues to improve as we scale tokens per parameter to extreme ranges (up to 10,000) |
| 5.76×10²³ FLOPs | Gopher's compute | [2] Fig. 2 caption
based on the number of FLOPs used to train Gopher (5.76 × 10^23) |
| 4× smaller, 4× the tokens | Gopher against Chinchilla | [2] §1
we predict that for the compute budget used to train Gopher, an optimal model should be 4 times smaller, while being training on 4 times more tokens. |
| 70B parameters, 1.4T tokens | Chinchilla | [2] §4
We test this hypothesis by training a model on the larger end of this range---70B parameters---for 1.4T tokens |
| 80 layers, 64 heads, d_model 8,192 | Chinchilla architecture | [2] Table 4
Chinchilla 70B & 80 & 64 & 128 & 8,192 & 1 × 10^-4 & 1.5M → 3M |
| AdamW | optimiser change | [2] §4.1
We use AdamW for Chinchilla rather than Adam as this improves the language modelling loss and the downstream task performance after finetuning. |
| 67.6% | MMLU in the body | [2] §4.2.2
with an average accuracy of 67.6% (improving upon Gopher by 7.6%) |
| two comparable runs | validation at large scale | [2] §6
Due to the cost of training large models, we only have two comparable training runs at large scale (Chinchilla and Gopher), and we do not have additional tests at intermediate scales. |
| 4.68B against 2.86B recommended | head-to-head at 10²¹ FLOPs | [2] App. D.3
From Kaplan et al. (2020), we find that the optimal model size should be 4.68 billion parameters. From our approach 1, we estimate a 2.86 billion parameter model should be optimal. |
| 1.3×10⁹ (C / 8.64×10¹⁹)^0.73 | Kaplan's adjusted power law, as Porian et al. write it | [5] §3.5
= 1.3e9 (C / 8.64e19)^0.73 which Kaplan et al. (2020) obtain by estimating the compute required to reach the same results at a sufficiently low batch size |
| 8.0 billion tokens | Table 3, 400 million parameters | [2] Table 3
400 Million & 1.92e+19 & 1/29,968 & 8.0 Billion |
| 205.1 billion tokens | Table 3, 10 billion parameters | [2] Table 3
10 Billion & 1.23e+22 & 1/46 & 205.1 Billion |
| 21.2 trillion tokens | Table 3, 1 trillion parameters | [2] Table 3
1 Trillion & 1.27e+26 & 221.3 & 21.2 Trillion |
| D ≳ 5×10³ N^0.74 | data needed to avoid overfitting | [1] §4.2
D gtrsim (5 × 10^3) N^0.74 |
| up to 4 epochs behave like unique data | repeated data | [8] Abstract
We find that with constrained data for a fixed compute budget, training with up to 4 epochs of repeated data yields negligible changes to loss compared to having unique data. |
| 0.095 | exponent of L(D) | [1] App. A, key parameters
alpha_D=0.095 & D_c=5.4×10^13 tokens |
| fitting the full L(N, D) | why the joint fit differs | [1] §4.2
as here we are fitting the full L(N,D) rather than just L(N, ∞) or L(∞, D). |
| about 0.05 in the loss | run-to-run variation, learning-rate figure | [1] App. D.6, Fig. caption
Run-to-run variation is at the level of ~0.05 in the loss, so averaging multiple runs is necessary to validate performance changes smaller than this level. |
| 5B to over 400B tokens | the same range, in the introduction | [2] §1
and trained on 5B to over 400B tokens |
| 67.5% | MMLU in the abstract | [2] Abstract
Chinchilla reaches a state-of-the-art average accuracy of 67.5% on the MMLU benchmark |
| Approach 3's estimates inconsistent with Approaches 1 and 2 | Besiroglu et al.'s finding | [13] Abstract
we find that their reported estimates are inconsistent with their first two estimation methods, fail at fitting the extracted data, and report implausibly narrow confidence intervals |
| 6.3×10²³ | recounted Gopher compute | [2] App. F
we use a slightly more accurate calculation giving a slightly different value (6.3 × 10^23 compared to 5.76 × 10^23) |
| most Kaplan runs under 100M | the second difference | [2] §2
in contrast the majority of runs in Kaplan et al. (2020) are significantly smaller---many being less than 100M parameters. |
| 0.74 to 0.78 against 0.73 | the simulated local exponent | [4] §3, Step 4
which are close to the Kaplan coefficient of 0.73. |
| only locally | the relation is not a power law | [4] §3, Step 3
This shows that in general the relationship between N^*_∖E & C_∖E is not a power law. |
| five models, 0.8M to 4.6M, BookCorpus | their tiny models | [4] §4, Experiment 1
We trained five models of sizes, N_T in [ 0.8M, 1.6M, 2.1M, 3.3M, 4.6M] on the BookCorpus dataset. |
| 0.49 with total parameters | their tiny-scale result | [4] §4, Result 1
Ours, N_T & C_T & 0.49 & 0.51 |
| 0.74 with non-embedding parameters | their tiny-scale result, non-embedding | [4] §4, Table 1
Ours, N_∖E & C_∖E & 0.74 & 0.26 |
| total parameters and compute | their recommendation | [4] Abstract
These findings lead us to recommend that future scaling studies use total parameters and compute. |
| 0.88 | Kaplan's unadjusted exponent | [5] §3.1
we obtain a compute-optimal exponent a and power law fits close to the power law 1.6e9 (C / 8.64e19)^0.88 obtained by Kaplan et al. (2020) |
| three factors | Porian et al.'s explanation | [5] Abstract
identifying three factors causing the difference: last layer computational cost, warmup duration, and scale-dependent optimizer tuning. |
| 3,000 steps at every scale | Kaplan's warmup is too long | [5] §3.3
this warmup period is too long: for smaller-scale models, the optimal number of tokens as a function of compute is less than or close to the number of warmup tokens, and therefore these models are suboptimally trained. |
| contra Hoffmann et al. | learning-rate decay is not essential | [5] Abstract
Counter to a hypothesis implied in Hoffmann et al. (2022), we find that careful learning rate decay is not essential for the validity of their scaling law. |
| 0.95 is suboptimal at 128 sequences or fewer | AdamW β₂ at small batch | [5] §1
we observe that setting the AdamW beta_2 parameter to be 0.95 is suboptimal at smaller batch sizes (128 and below) |