Should LLM teams train smaller models for longer?
When inference volume is factored into scaling laws, does the compute-optimal training recipe shift away from Chinchilla's recommendation toward smaller, longer-trained models? This matters because most teams face both training and inference costs.
The excerpt modifies the Chinchilla scaling laws — the empirical formulas that estimate how model quality changes with parameter count and training data — to account for the compute and dollar cost of inference, not just training. It argues that "LLM researchers expecting reasonably large inference demand (~1B requests) should train models smaller and longer than Chinchilla-optimal." The motivating observation: "the Llama 2 family of models were trained on 2 trillion tokens and the Llama 3 family of models were trained on 15 trillion tokens, which is far more data than the Chinchilla scaling laws would deem 'optimal.'" Since smaller models are cheaper to run, the extra training compute spent going past Chinchilla-optimal "pays off after enough inference requests."
The mechanism is a reformulation of the Chinchilla loss function L(N, Dtr) — parameterized by parameter count N and pre-training tokens Dtr — to solve jointly for the parameter and token counts that minimize combined training and inference compute (Sec. 2), and separately combined training and inference dollar cost (Sec. 6), given an assumed total inference demand Dinf (the sum of input and output tokens across all requests). The derivation assumes "that conditioned on pre-training loss, inference demand is independent of model size and token count" — i.e., models of equivalent quality but different sizes see the same request volume, so the only lever left is to shrink the served model and make up quality by training longer.
This modifies the same underlying scaling law as Can architecture choices improve inference efficiency without sacrificing accuracy?, but on a different axis. That note adds architectural variables (hidden size, MLP-to-attention ratio, GQA) as a lever that improves inference efficiency independently of parameter count or training compute; this excerpt instead treats the tokens-per-parameter ratio itself as the lever, holding architecture fixed, and ties the optimal ratio directly to projected inference volume. Read together, both reject the idea that a single training-compute-optimal model exists — the "optimal" model depends on how it will be architected and how much it will be asked to serve.
The excerpt's own inference-demand independence assumption is a simplification it flags, not a finding: Sec. 6's dollar-cost model "assume[s] MFU and cost per FLOP do not depend on model size, configuration, or sequence length," and leaves aside latency requirements. It also treats inference demand as a known, fixed input rather than something that itself depends on model price or latency in deployment — a feedback loop (cheaper, faster models may draw more usage) the excerpt does not model. The implication it does support, carefully stated: for any team that can forecast inference volume in the ~1B-request range, Chinchilla-optimal sizing is the wrong target, and the error compounds as projected demand grows.
Inquiring lines that read this note 6
This note is a source for these research framings, grouped by the broader line of inquiry each explores. Scan the bold lines of inquiry; follow any specific question forward.
What fundamentally drives AI inference pricing and cost decline patterns?- Does annualized capex or operating expense drive inference pricing?
- How does staff cost share differ between individual training runs and whole company budgets?
- Do smaller models match Chinchilla-optimal performance when trained much longer?
- What causes Chinchilla scaling curves to flatten with extreme training data?
- Do all models of equivalent quality see the same inference request volume?
- What happens to the optimal model size as projected inference volume grows?
Related concepts in this collection 2
This note in its neighbourhood — explore the map, then jump to a related concept in the list below.
Click a node to walk · click center to open · click Open in graph to see this note in the full knowledge graph
-
Can architecture choices improve inference efficiency without sacrificing accuracy?
Standard scaling laws optimize training efficiency but ignore inference cost. This explores whether architectural variables like hidden size and attention configuration can unlock inference gains without trading off model accuracy under fixed training budgets.
same move of modifying Chinchilla for inference cost, but via architecture choice rather than training-token ratio
-
Does model quality plateau when training on extreme token ratios?
Whether smaller language models trained on very high token-to-parameter ratios eventually hit a saturation point or continue improving indefinitely, and how this affects optimal training recipes.
the empirical validation of the smaller-longer recipe this note's formula recommends
Related papers in this collection 8
Papers most semantically related to this note, ranked by cosine similarity in the embedding space.
- Beyond Chinchilla-Optimal: Accounting for Inference in Language Model Scaling Laws
- Scaling Laws Meet Model Architecture: Toward Inference-Efficient LLMs
- A Survey on LLM Inference-Time Self-Improvement
- Adam's Law: Textual Frequency Law on Large Language Models
- The Unreasonable Ineffectiveness of the Deeper Layers
- ZebraLogic: On the Scaling Limits of LLMs for Logical Reasoning
- Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters
- Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention
Original note title
accounting for inference demand pushes the compute-optimal LLM training recipe toward smaller longer-trained models