Compute-Optimal Training Couples Model, Data, and Budget
Use a simplified compute ledger to compare parameter-token allocations, then retain data quality and empirical optimization boundaries.
A parameter count alone does not specify a training run. Under a fixed compute budget, a larger model usually permits fewer training tokens, while a smaller model permits more. The allocation can change the final loss.
Use a Simple Budget Model
For a dense Transformer training estimate, write
where is the parameter count, is the number of training tokens, and collects architecture and accounting constants. If , then both allocations satisfy the same simplified budget:
| Allocation | |||
|---|---|---|---|
| A | 100 | 12 | 1200 |
| B | 40 | 30 | 1200 |
The budget equation alone cannot tell which obtains lower loss. That requires an empirical relation involving model size, data, and the tested regime.
Keep Tokens and Data Quality Visible
Training tokens are not interchangeable units of information. Distribution, duplication, filtering, contamination, curriculum, modality, and tokenizer all matter. Repeating a small dataset is not automatically equivalent to observing new samples.
Q1. Complete a compute allocation
Under and the approximation , a model has . How many training tokens fit the simplified budget?
Compute it first, then check your number.
Hint
Solution
Compare Studies without Blending Them
Hoffmann et al. (2022) re-examined compute-optimal allocation using experiments and analyses that differed from earlier scaling work. Preserve each study's models, data, compute definitions, fitted range, and objective. “More parameters is always better” and “one fixed token-to-parameter ratio is universal” are both too strong.