Ankit & Francois on Why AGI Needs a World Model, Not More Data
World models — learning to predict the next state of the world — may be the route to sample-efficient AI, especially in robotics where brute-force reinforcement learning is intractable.
Sample Efficiency Is the Real Wall
The frontier problem is not raw knowledge but how few examples it takes to learn something new — humans need a handful, models need tens of thousands.
One of the biggest open problems in AI right now is how to solve sample efficiency.
A Perfect World Model Needs Zero Samples
A perfect world model is the zero-sample limit of learning — the more exact your model of what happens next, the fewer real-world trials you need, all the way down to none.
Can you imagine if like we needed to collect 1 million training examples of like us shooting spaceships to the moon to like know how to do it
A World Model Is Just the Transition Function
Strip away the mystique and a world model is just the map from a state and an action to the next state — and once you have that map, steering the system becomes a control problem you compute rather than learn.
So we can solve this in closed form basically like you know we can because we have this world model of Newtonian physics we can say at every step exactly how this drone should fly so that it lands on the appropriate thing under a set of constraints like max thrust available gravity
The Dividing Line Is Whether You Can Take a Gradient
Add one other agent whose next move you can't differentiate through and closed-form control collapses into the messy, expensive world of reinforcement learning.
I have to resort to this awful area called reinforcement learning
AlphaGo Doesn't Scale to the Real World
Tree search is affordable only while the action space stays tiny — it goes from 24,000 model calls per Go move to a hopeless number once the world is as wide as a robot arm.
So that be 60 million uh invocations of the model. So that better be a small model, right?
Behavior Cloning Is Next-Token Prediction, With Actions
Take the tokenizer head off a language model, bolt on an action space, and copying a human driver becomes the same next-token game that made LLMs work — if you can find the actions.
in many ways, it's the closest thing that just looks like the next token prediction from LLM that seems to scale pretty well with natural language
Learn the World From Video, Then Dream the Training Data
The breakthrough is to decouple the two things we have in unequal supply: learn the world from abundant video, spend your scarce action labels only on conditioning, then train the policy on synthetic rollouts.
the first paper to mine diamonds in Minecraft. I'm not a big Minecraft player, but apparently that's extremely difficult. That's like next level difficulty. And it did it all on synthetic data, which is kind of crazy
JEPA Predicts in Latent Space — If It Doesn't Collapse First
The current frontier predicts a compressed latent of the next state instead of every pixel — far cheaper, but it invites a sharp failure where the model cheats by predicting nothing.
now this doesn't work this collapses hard and so what happens is basically just if you if you just predict zero