Compute-Optimal Allocation Between Model Size and Data Under a Parametric Scaling Law: Closed-Form Solution, Convexity Proof, Misallocation Penalty, and Sensitivity to the Fitted Constants

Under the parametric loss L(N,D) = E + A/N^α + B/D^β with training compute C = 6ND, I derive the compute-optimal model size and token count in closed form: N* = G(C/6)^{β/(α+β)} with G = (αA/(βB))^{1/(α+β)}. I prove that the constrained problem is strictly convex in log N, so the solution is unique, and derive a quadratic approximation to the loss from misallocation. Two published fits are evaluated: the rounded constants of Hoffmann et al. (2022) and the re-estimate of Besiroglu et al. (2024), both read directly from the papers' text. They imply different allocation rules. At the Gopher budget of 5.76 × 10²³ FLOPs, the first gives N* = 32 B parameters and 93 tokens per parameter; the second gives N* = 72 B and 18 tokens per parameter. The ratio drifts with compute in opposite directions under the two fits (50 → 153 against 22 → 16 tokens per parameter over 10²¹–10²⁶ FLOPs). Misallocation is cheap near the optimum: a model 2× too large raises loss by about 0.005 nats, and offsetting a 4× oversize needs 1.8–1.9× more compute. The prescriptive content of "compute-optimal" depends on second-decimal differences in fitted exponents.

Authors

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-10-18
DOI
https://doi.org/10.5281/zenodo.23450086
Primary Topic
Topic Modeling
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Compute-Optimal Allocation Between Model Size and Data Under a Parametric Scaling Law: Closed-Form Solution, Convexity Proof, Misallocation Penalty, and Sensitivity to the Fitted Constants

Manoj Kumar Aguru
Zenodo (CERN European Organization for Nuclear Research)
Topic Modeling
preprint

Compute-Optimal Allocation Between Model Size and Data Under a Parametric Scaling Law: Closed-Form Solution, Convexity Proof, Misallocation Penalty, and Sensitivity to the Fitted Constants

Manoj Kumar Aguru
preprint en

Abstract

Under the parametric loss L(N,D) = E + A/N^α + B/D^β with training compute C = 6ND, I derive the compute-optimal model size and token count in closed form: N* = G(C/6)^{β/(α+β)} with G = (αA/(βB))^{1/(α+β)}. I prove that the constrained problem is strictly convex in log N, so the solution is unique, and derive a quadratic approximation to the loss from misallocation. Two published fits are evaluated: the rounded constants of Hoffmann et al. (2022) and the re-estimate of Besiroglu et al. (2024), both read directly from the papers' text. They imply different allocation rules. At the Gopher budget of 5.76 × 10²³ FLOPs, the first gives N* = 32 B parameters and 93 tokens per parameter; the second gives N* = 72 B and 18 tokens per parameter. The ratio drifts with compute in opposite directions under the two fits (50 → 153 against 22 → 16 tokens per parameter over 10²¹–10²⁶ FLOPs). Misallocation is cheap near the optimum: a model 2× too large raises loss by about 0.005 nats, and offsetting a 4× oversize needs 1.8–1.9× more compute. The prescriptive content of "compute-optimal" depends on second-decimal differences in fitted exponents.

Zenodo (CERN European Organization for Nuclear Research)
Topic Modeling
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.