i
Research
Review · 2025-09-05

VLWM predicts the future in language instead of pixels

VLWM predicts the future in language instead of pixels

When we ask a machine to help us cook dinner or swap a SIM card, it has to do more than recognize the objects in frame — it has to picture how the world will change from one step to the next. Most systems today see pixels and answer in short phrases, and long-horizon planning still does not work. The VLWM (Vision Language World Model) team proposes a different route: describe the future in words and operate on meanings rather than frames. Their work is about fitting vision, language and reasoning together so that agents move toward a goal with confidence.

VLWM overview: (a) a JEPA world model that predicts abstract future states instead of noisy, high-volume raw observations; (b) from video context it targets a structured text description of the unobserved future — a goal plus interleaved actions (A) and state changes (ΔS), extracted automatically; (c) it proposes candidate goals and interprets them against the current initial state and the expected final state; it supports both a fast reactive System-1 plan and reflective System-2 reasoning based on cost minimization.

The core idea is simple: instead of expensively generating future frames, the world model works in language space. It watches a video, formulates a reachable goal, then predicts a trajectory — actions alternating with state changes. A plan like that is compact, transparent and easy for other language models to check. It is also cheaper to compute: text scales better than arrays of pixels.

How do you pull that off on real video that is long and noisy? The authors compress the raw stream into a caption tree — a hierarchical representation in which a long clip is broken into semantic segments with short descriptions. That cuts the data volume sharply and frees up capacity for reasoning about the future. A large language model then extracts four key components from those captions using a self-refine scheme: the goal, its interpretation in terms of the initial and the desired state, the executable steps, and how the world changes after each step. Iterative refinement strips out inaccuracies and keeps the plan coherent.

The structure of the caption tree (bottom) extracted from the video (top). Each block corresponds to one video caption.

Then the VLWM world model itself comes into play. It is trained to predict a sequence of pairs: an action and a state change. That enables two modes. The fast, reactive one simply generates the whole plan and executes it. The more deliberate one enumerates several candidates, rolls the future forward inside the model, scores how close the predicted states come to the goal, and picks the path with the lowest cost. Cost is computed by a separate critic: a small language model trained on its own to rank good continuations against bad ones.

System-2 planning in VLWM: (a) the critic is trained self-supervised, lowering cost for valid progress and raising it for irrelevant distractors or shuffled steps; (b) VLWM generates candidate action sequences and simulates the future state transitions, the critic scores the trajectories against the goal, and the planner selects the plan with the lowest cost.

The results are credible. On the Visual Planning for Assistance benchmark VLWM sets a new best across every major metric, with the 8B model beating larger systems. In the PlannerArena human comparison, people pick plans produced with reflective search noticeably more often: System-2 mode earns roughly 27% more Elo than fast decoding. That is a sign the internal search and the cost scoring really do help avoid mistakes and absurd steps.

An illustration of the PlannerArena annotation interface.

The critic is not a supporting detail. On goal-completion detection it comfortably beats standard semantic-similarity models, and on the WorldPrediction-PP procedural planning task it sets a new accuracy record. What matters most is a good reward function.

Cost curves scored by different critic models. Each plot shows 3K curves over goal-completion detection trajectories, where every trajectory consists of the gold reference plan (0%–100%) followed by distractor steps (100%–200%). Red dots mark the minimum-cost steps (the moments of goal completion). VLWM-Critic pinpoints goal completion at about 100% of plan length, while the baselines behave suboptimally or noisily.
WorldPrediction-PP results: VLWM-critic-1B set a new SoTA with 45.4% accuracy.

How all of this scales matters too. Training ran on hundreds of thousands of instructional and egocentric (first-person) videos, and the caption tree squeezed terabytes down to hundreds of megabytes of text. The approach fits naturally into the ecosystem of large language and multimodal models, so perception or reasoning can be improved later without rebuilding the whole system.

There are limits. Language abstraction sometimes loses fine spatial detail, and quality depends heavily on the captions and the extracted plans being correct. Actor and critic have to be balanced: search that is too aggressive slows the response down, search that is too fast costs reliability. On the whole, though, it reads as a sober trade-off between speed and deliberation.

VLWM shows that predicting the future does not have to happen in pixels. Describe the world in words, encode its state as a chain of actions and changes, and perception, reasoning and planning come together. It is a step away from imitation and toward agents that can think, test hypotheses inside their own world model and choose a sensible path — even in long, tangled scenarios.

AI paper breakdowns

Every day we read the new AI papers and retell what matters in plain language — no hype, no filler. If you want to see where AI agents are heading before everyone else, subscribe.

New breakdowns every day.

On Telegram