Scale is an allocation problem

With one GPU-day, would you spend it on a larger model, more distinct tokens, or more updates over the same data?

Project 1 has three coupled quantities

  • N — learned parameters
  • D — training tokens processed
  • C — training compute

Project 1’s constraints, in one table

ConstraintValue
Design target32M learned parameters
Absolute eligibility boundary33.6M learned parameters (5% tolerance)
Compute allowance24 T4-equivalent hours
Training FLOP ceiling4.0×10¹⁷

Takeaway: every proposed run must fit all of these at once — a plan that saves parameters but blows the FLOP ceiling is not a legal plan.

Bigger is not automatically better

Three ways a plan can fail without touching the parameter boundary:

  1. too many parameters for the data on hand — capacity the corpus can’t fill
  2. many repeated passes over a narrow corpus — compute spent, little new information gained
  3. a tiny model trained long after useful validation improvement stopped — time spent past the point of return

Discuss: for each sketch, which resource — model, data, or compute — is actually the bottleneck?

Power-law intuition

L∞small-data regionuseful-scaling regionirreducible-error regionloss (log)scale (log)

The power law, written out

L(x) = a·x⁻ᵅ + L∞ — diminishing returns, and an approximately straight middle region on log–log axes.

Kaplan et al.: smooth scale relationships

  • reported smooth relationships between loss and N, D, C across many orders of magnitude
  • larger models are more sample-efficient — they reach a given loss with fewer tokens
  • weak sensitivity to architectural shape within the studied range
  • early stopping is compute-optimal: stop before full convergence on a fixed budget

Callout: the studied regimes are far above 32M. Read this as evidence of a shape, not a recipe for this course’s model.

Chinchilla changed the allocation

QuestionEarlier view (Kaplan-era)Chinchilla-era view
Under fixed compute, favour…more parametersparameters and tokens roughly together
Main risk—undertrained large models

The iso-compute valley

balanced allocationtoo small, too many tokenstoo large, too few tokensmodel size at fixed compute

Takeaway: an undertrained large model can lose to a smaller model trained on more tokens, under the same compute.

What transfers to 32M?

TransferableNot guaranteed at this scale
Identify which resource is the bottleneckThe published exponents
Compare configurations within a fixed budgetAn optimal token/parameter ratio
Allocate data with model sizeA specific batch size
Stop by evidence, not by habitAn exact loss value for this corpus

Decision 1: choose a plan

Three legal fictional plans, same maximum compute:

PlanParametersTokens
A18Mmore tokens
B31.8Mbalanced tokens
C33.5Mfewer tokens

Choose: not a winner — which uncertainty would you need to measure before picking one, given only this table?

A token is not a unit of value

SourceQuality signal
Clean, diverse storieshigh value per token
Duplicated storiesdiminishing value with repetition
OCR noiseactively harmful
Synthetic textbook-like storiescan raise value per token, not guaranteed

Quality, diversity, duplication, provenance, and domain fit all change what a token is worth.

Count data twice

Two ledgers, not one:

  1. unique source/token inventory — how much distinct material exists
  2. tokens processed during optimisation — how much the optimiser actually saw, including repeats

Multiple epochs increase the second without increasing the first — a report needs both numbers, or a duplication claim can’t be checked.

From batch to compute

  • sequence length × sequences per micro-batch × gradient accumulation → effective tokens per update
  • updates × effective tokens per update → total tokens processed

The repository is canonical for hardware conversion and FLOP estimation — the ledger relationship above is for reasoning about a plan, not for computing a number to report without running the tool.

Learning rate is a schedule, not one number

warm-uppeakdecay

Early instability calls for a short warm-up; later refinement calls for decay. An aggressive rate late in training can destroy a useful checkpoint — a lesson that carries forward directly into Projects 2 and 3’s post-training stages.

Stability signals

SignalWarning patternAmbiguous pattern
Train/validation lossvalidation rises while train fallsvalidation flattens early
Gradient normsudden spike or divergenceone isolated spike
Throughput/memorysteadily degrading throughputone slow step
Fixed-prompt samplesrepeated or degenerate textone attractive sample

A single spike, or one attractive sample, is not a conclusion on its own.

A pilot answers one uncertainty

Template: hypothesis → one changed factor → controlled variables → maximum spend → measurement window → stop rule → decision the result can change.

Contrast this with an undirected hyperparameter sweep, which spends budget without naming what it will decide.

Decision 2: continue, revise, or stop?

A fictional pilot: training loss falls, validation loss flattens, gradient norm is stable, projected runtime is 30 hours, samples repeat source phrases.

Choose an action, and name the evidence for it.

Your Project 1 run plan

Final checklist before spending the principal run:

  • legal preflight passed
  • clean, source-level split
  • unique and processed token counts recorded
  • baseline plus one controlled comparison

Your Project 1 run plan, continued

  • compute reservation held back
  • checkpoint interval set
  • fixed evaluation prompts ready
  • a stop rule and recovery margin

Next: bring this checklist to the Week 3 drop-in clinic.

Claims you may and may not make

SupportedUnsupported
“Under our fixed data and steps, A achieved lower validation loss than B.”“Width is generally better than depth.”
“Chinchilla proves our exact configuration is optimal.”
“Lower PPL guarantees better stories.”

Exit ticket

Submit: your chosen baseline · the largest unresolved uncertainty · the pilot that can resolve it · maximum hours/FLOPs reserved · the evidence that would trigger a change.

Project 1 due 21 March 2027, 23:59 AET.

Required reading: TinyStories, Sections 1–2

Figure and source register

Figure and source register, continued

  • Course budget values and ledger definitions: src/course-config.ts and the Project 1 assessment page, restated here rather than screenshotted, since no starter-repository run exists yet to capture.
  • The power-law, iso-compute-valley, and learning-rate-schedule diagrams in this deck are original schematics drawn for this course, illustrating the qualitative shape of the cited papers’ reported relationships — not reproductions of those papers’ own figures.