World Models

World models learn a latent state and transition model that supports prediction, planning, or control. In video understanding, this means representing how scenes evolve, not only recognizing a label. The topic overlaps with video representation, self-supervised video representation learning, and V-JEPA 2.

Encoder, dynamics, decoder

A compact world model has an encoder, dynamics model, and decoder or cost head:

Planning can happen in latent space by rolling out candidate actions and minimizing a cost. World models and JEPA focuses on the variant where prediction happens in representation space rather than pixel space.

flowchart LR
  Obs[Observation x_t] --> Encoder[Encoder E]
  Encoder --> Latent[Latent state z_t]
  Action[Action a_t] --> Dynamics[Dynamics F]
  Latent --> Dynamics
  Dynamics --> Next[Predicted next latent]
  Next --> Decoder[Decoder or cost head]

Worked rollout

With a simple latent transition

the rollout is:

steplatent state
0
1
2
3

The first coordinate grows because it receives times the second coordinate at each step; the second coordinate decays by a factor of . Real learned world models replace this hand-coded matrix with neural dynamics trained from video or interaction.

Caveats

A useful latent rollout is not automatically a faithful simulator. Evaluation must test counterfactuals, long-horizon error, uncertainty, and planning performance, not only next-step prediction. Video-only data may not identify action effects without interaction data.

References