Rafhael Oliveira · IA

The inversion that did not repeat: the world model's cost appears once the ruler stops swinging

Chapter 15 recorded that, with enough data, the world model started paying 3.38%. It was one seed. With more seeds and fixed validation, it costs in every round — and the cost grows with size.

1 min read· chapter: 10 min

Landed here from LinkedIn? This is chapter 19 of the experiment Antivision · Latent Rehearsal. It reads on its own; if you want the beginning, it is chapter 00.

I wrote in a chapter that my world model made the generator 3.38% better. With more seeds and a fixed ruler, it costs. The result came from a single round: the generator alone at 3.7503 loss, with the world model at 3.6236. With seeds 2 and 3, the map went back to costing. And while investigating why, I found the problem in the ruler: at every evaluation training drew a new validation sample, and the same nearly finished model scored 3.576 at one step and 3.719 at the end — a swing larger than the effect I was measuring. I fixed the validation batches and redid the whole pair: three mini seeds and two pequeno seeds. With the clean ruler, the seed spread fell from 2.5% to 0.87% of the loss, and the world model's cost showed up in every round: +1.77% on mini, +4.92% on pequeno.

The three points

  1. With seeds 2 and 3, the world model goes back to costing. The −3.38% came from a favourable seed, under a protocol where the drawn validation swung more than the effect itself: 3.576 at one step, 3.719 at the end of the same run.
  2. With fixed validation batches, the seed spread falls from 2.5% to 0.87% of the loss on mini, and to almost zero on pequeno. Nearly all the noise came from drawing the sample, not from initialization.
  3. With the clean ruler, the cost shows up in all five comparisons: +1.77% on mini (3 seeds) and +4.92% on pequeno (2 seeds). More budget does not flip the sign; the cost grows with the size of the model.

From the chapter, the beginning

Chapter 15 recorded that, with enough data, the world model stopped costing the generator and started paying 3.38%. It was a single seed. With more seeds, and then with a validation measure that stopped swinging, the cost shows up in every round — and it grows with the size of the model.

Chapter 15 trained mini on the whole corpus, with the Chinchilla budget (1,961 steps, 64 million tokens), in two ways: generator only, and generator with the hierarchical world model with EMA target — the recipe that had done best in chapter 13. The generator alone reached 3.7503 validation loss; with the world model, 3.6236. That is 3.38% better, and the chapter read the result as an inversion: in the collection's short training the map cost about 1%, and with enough data it would start paying.

The reading had a reasonable explanation — with little data, predicting the next state competes with the generator for the same capacity; with a lot, it acts as a regulariser — and it matched what one expects from a predictive architecture. But it came from one round on each side, with a single seed, in a regime where the seed spread had never been measured. Chapter 13 §12.3 measured that spread on the duel: about 0.84% of the loss. On the ladder, nobody had measured it.

Keep reading chapter 19 →Ler em portuguêsThe PDFs by e-mail →
What the chapter covers · 7 sections · 10 min
  1. What was written
  2. Two rounds
  3. Three seeds, with drawn validation
  4. Why one seed misled
  5. The same question, with a clean ruler
  6. What changes in the collection
  7. Limits of this measurement
All chapters of the experiment · Antivision · Latent Rehearsal
  1. 00 · Antivision: an ELLM, a language model that rehearses in latent space before speaking
  2. 01 · The custom tokenizer: why 16 thousand tokens suit this corpus better than 50 thousand, and what the whole table shows
  3. 02 · Two architectures, one file: the fair duel between 2019 and what came after
  4. 03 · The geometry of the embedding: training starts with a perfect space and ruins it
  5. 04 · The lineage of collapse: five answers from computer vision, each measured on text
  6. 05 · The rudder and the world model: three versions of the same question, each measured against the last
  7. 06 · Method and mistakes: the rules of a legitimate comparison and the catalogue of our own errors
  8. 07 · Scale and budget: what it costs to reach GPT-3, and why VRAM is almost never the limit
  9. 08 · The workbench: serve, x-ray and train from the browser
  10. 09 · The hierarchical architecture: the whole project aligned to the four patterns of the world model
  11. 10 · What JEPA is, according to LeCun: two talks read concept by concept, and where each one applies
  12. 11 · Memory and horizon: two claims from LeCun that became experiments, and both hold for text
  13. 12 · Learned costs and inference modes: the three searches and the two sources of energy, measured
  14. 13 · The corrections from LeCun's talks: failure, passage, flag, measurement
  15. 14 · Consulted memory: a bank of abstract states read by attention
  16. 15 · What it costs and what it delivers: us, GPT-1 and GPT-2 on the only ruler valid across tokenizers
  17. 16 · The minimum was off the list: the day I measured my own search by energy
  18. 17 · ELLM: the right name for a language model that predicts in latent space and decides by energy
  19. 18 · Searching better does not fix the energy: three searches, the same text
  20. 19 · The inversion that did not repeat: the world model's cost appears once the ruler stops swinging
  21. 20 · What changed on 17/09: two new chapters and two corrections

Before this: the GPT-2 course

Rafhael Oliveira
Rafhael Oliveira

Writes the GPT-2 course and runs the Antivision experiment, measuring every run before publishing the number. The LinkedIn posts start here and end in the chapters.

Follow on LinkedIn →