∇AI Research Lab
AI Research LabPaper reading companion

World models · OpenReview 2022 · not an experiment paper

JEPA: train the world model to predict $E(y)$, not $y$

Lede

A generative world model is trained to reconstruct the next observation $y$ itself — if $y$ is an image, that means pixels. Most of those pixels are useless for a grasp (texture, specular noise). LeCun’s 2022 position paper says the world model should instead encode $x$ and $y$ and train a predictor so $P(E(x), z) \approx E(y)$. The loss never sees pixels of $y$. He calls that a Joint Embedding Predictive Architecture. I-JEPA, V-JEPA, and V-JEPA 2 are the papers that later train it.

← All paper guides

Reach for a mug: what you predict, and what a generative model predicts

You decide to pick up a mug. The facts that matter are pose and contact: the mug stays put, the handle is on the left, the counter is solid. A video generator / MAE is trained to output the next RGB frame (or its pixels). You are not. A JEPA is trained to output a vector $E(y)$ from which those facts can still be read, and from which unpredictable texture can be dropped. That is the claim of this position paper.

01

You

You encode the scene as a handful of facts (counter, mug, kettle pose). You do not store a pixel buffer.

The paper

Perception module. Embed the sensory stream into a representation $s_t$ that already threw away unpredictable clutter.

02

You

You remember: I just walked through a doorway; the mug was on the left. You do not replay the last minute of video.

The paper

Short-term memory. A running latent of recent $s_t$, not a pixel buffer.

03

You

You predict the next state of the mug, not a new photograph of the kitchen.

The paper

World model = JEPA. Encode the scene as $E(x)$, encode a later or hidden part as $E(y)$, train a predictor so $P(E(x), z) \approx E(y)$. The loss never looks at pixels of $y$. $z$ says which hidden part / which future you are asking about.

04

You

You feel a cost: do not burn your hand; do not knock the cat. Nobody labelled those as rewards in this kitchen.

The paper

Cost / energy module with intrinsic drives (energy, uncertainty, discomfort) plus a few hardwired terms. Behaviour is not “maximise a scalar reward the experimenter wrote.”

05

You

You pick an action by mentally trying a few reaches until the predicted cost is low, then you move.

The paper

Actor proposes actions; planning is search in latent space through the world model. Configurator sets which level of the hierarchy is running (grasp vs walk-to-counter).

The mapping is the architecture: perception → memory → JEPA world model ($P(E(x),z)\approx E(y)$) → cost → actor. I-JEPA trains that predictor on still images. V-JEPA trains it on video. V-JEPA 2 adds the actor’s planning loop on a Franka.

The paper, in order — every section

Open OpenReview PDF BZ5a1r-kVsf (62 pages, version 0.9.2). This is a position paper. There are no ImageNet tables. Do not look for them.

Abstract

Three questions: how could machines learn as efficiently as humans and animals? How could they reason and plan? How could they learn representations of percepts and action plans at multiple levels of abstraction, so they can predict and plan at multiple time horizons?

The proposed answer is an architecture plus training paradigms: a configurable predictive world model, behaviour driven by intrinsic motivation, and hierarchical joint-embedding architectures trained with self-supervised learning.

Keywords he lists so you know the vocabulary pile: energy-based models, world models, joint embedding architecture, intrinsic motivation, SSL.

§1 Prologue

He says this is not a technical nor scholarly paper in the traditional sense. It is a vision for machines that learn more like animals, that can reason and plan, whose behaviour is driven by intrinsic objectives rather than hard-wired programs, external supervision, or external rewards. Read that sentence twice: the later I-JEPA paper is allowed to have no labels and no SimCLR colour jitter because this prologue already forbade those as the path.

Stated contributions (early in the PDF)

  1. A cognitive architecture for autonomous intelligence (the six modules).
  2. JEPA and Hierarchical JEPA — a non-generative architecture for predictive world models that learn a hierarchy of representations (his Figures 12 and 15).
  3. A non-contrastive SSL paradigm that produces representations that are simultaneously informative and predictable (his Figure 13; the VICReg family is the working example).

Why not generative models / LLMs as the world model

A generative world model (MAE on images, a video generator, an LLM that emits tokens) is trained to reconstruct the observation $y$ itself. If $y$ is an image, that means pixels. Most of those pixels are unpredictable — leaves in wind, steam, sensor noise — so the model spends capacity on texture it cannot use for planning. LeCun’s claim is that animals (and a useful robot) do not do that. A JEPA predicts $E(y)$, a vector that is allowed to drop anything the encoder decides is noise. Planning then searches in that vector space, not in image space. That is the technical content of the position paper; I-JEPA §2 draws it as three energy-based diagrams.

LLMs, in this document, are useful but are not a world model of the physical world: they predict tokens, not latent states of a kitchen. Do not over-read a 2022 position paper as a 2026 LLM takedown. The engineering content is “predict $E(y)$ from $E(x)$, conditioned on $z$.”

Energy-based models (the maths he actually uses)

Compatible pair $(x,y)$ should have low energy $F(x,y)$; incompatible pairs high energy. JEA (joint embedding architecture, SimCLR/BYOL/VICReg): $F = D(E_x(x), E_y(y))$. Generative: $F = D(y, \mathrm{Dec}(E(x), z))$. JEPA: $F = D(E_y(y), P(E_x(x), z))$. Same energy idea, loss in latent space, extra variable $z$ that tells the predictor which $y$ to aim at (mask location, future time, action).

The six-module agent

Configurator, perception, world model, cost, actor, short-term memory. Perception and world model are JEPAs (possibly stacked). The configurator modulates them so the same hardware can plan a grasp at 200 ms and a walk across the room at 20 s. Cost is not a sparse external reward; it is a differentiable energy the actor minimises by thinking ahead through the world model.

Hierarchical JEPA

Low level: fast, concrete, maybe closer to pixels. High level: slow, abstract, good for long-horizon planning. Each level predicts at its own timescale. This is the piece none of I-JEPA / V-JEPA / V-JEPA 2 fully trains. They are single-level JEPAs. When someone says “LeCun’s JEPA,” check whether they mean this hierarchy or the single predictor in I-JEPA Figure 3.

Training / collapse

Joint embedding methods collapse if you only pull positives together. Contrastive negatives, VICReg variance-covariance, clustering, or an EMA teacher (BYOL) are the collapse brakes. I-JEPA and V-JEPA pick the EMA teacher. The position paper prefers non-contrastive (no need to define “incompatible” images).

What the paper does not claim

No new ImageNet number. No trained weights. No proof that JEPA beats MAE. No robot demo. Those arrive in 2023–2025. If you only need the trained image model, skip to I-JEPA.

The engineering tension

SSL in vision in 2022 is a fork: (A) invariance methods that need a hand-written list of “this crop is the same image,” which do not travel to video or audio cleanly; (B) MAE-style pixel reconstruction, which travels but learns a lot of texture. JEPA is the bet that you can keep MAE’s masking (so no SimCLR recipe) and still get semantic features by moving the loss into embedding space. In the kitchen: you want a vector that still says “mug on the left,” not a reconstruction of the steam.

Naive world model

Decode the next RGB frame. Capacity goes into leaves and specularities. Planning a grasp means generating a video. Expensive, and most of the video is irrelevant.

JEPA world model

Encode $x$ and $y$. Predict $E(y)$ from $E(x)$ given $z$. The encoder may delete steam. Planning is search in $E(\cdot)$, which is what V-JEPA 2-AC later does on a Franka.

Prerequisites

  1. 01

    What a representation is for

    If “embedding” is still a vibe, the rest of JEPA is word salad. You need: a vector that is useful for later tasks, not a compressed JPEG.

  2. 02

    Self-supervised learning, two families

    Invariance (SimCLR / DINO / VICReg) versus reconstruction (MAE / BERT). JEPA is a third cartoon that looks like MAE but loses like DINO.

  3. 03

    World models as “imagine then act”

    Ha & Schmidhuber 2018 is the named ancestor for “train a model of the world, then plan inside it.” JEPA is that idea with the decoder ripped out.

Why this video: the prerequisite, not this paper. What an embedding layer even is.
Ha & Schmidhuber lineage: VAE + RNN world model. JEPA’s move is to drop the pixel decoder and predict $E(y)$ instead.

Knowledge graph

energy-based models │ ├─ JEA (SimCLR / VICReg / DINO) — invariant views ├─ generative (MAE / BERT) — reconstruct y └─ JEPA — predict E(y) | E(x), z │ ├─ I-JEPA images, 2023 ├─ V-JEPA video, 2024 └─ V-JEPA 2 video + robot planning, 2025
EBM / VICReg / MAE
↓
perception JEPA world model cost
↓
six-module agent (this paper)

Architecture

Configuratorwhich timescale / which expert
↓ modulates
Perceptions_t = E(obs)
Short-term memoryrecent s
↓
World model (JEPA)ŝ' = P(E(x), z)
Cost / energyintrinsic + a few hardwired
↓
Actorsearch actions that lower predicted cost

LeCun’s agent. Only the world-model box is what I-JEPA trains. V-JEPA 2-AC is the first paper in this family that actually closes the actor loop on hardware.

xcontext (visible)
ytarget (hidden / future)
zmask / time / action
↓
E_x(x)
E_y(y) often EMA teacher
↓
P(E_x(x), z) ≈ E_y(y)loss in embedding space

The JEPA cartoon every later paper redraws. $z$ is what makes this predictive rather than “make two views identical.”

Results, honestly

Fact from the paperWhat it means
Zero trained models, zero tablesDo not cite this PDF for an ImageNet number.
JEPA defined as $P(E(x),z)\approx E(y)$This is the sentence I-JEPA implements with multi-block masking.
Hierarchical JEPA is proposed, not trainedStill mostly open. V-JEPA 2 has two stages (video then actions), not a trained hierarchy of timescales.
Behaviour from intrinsic cost, not RL rewardV-JEPA 2-AC still plans toward an image goal, not a learned intrinsic cost. The actor in this PDF is ahead of the engineering.
If a blog says “JEPA achieves 81% on ImageNet,” they mean I-JEPA or V-JEPA. This document cannot achieve anything. It is a drawing.

Worked intuition

Take a $224\times 224$ kitchen photo. Split it into $16\times 16$ ViT patches ($14\times 14 = 196$ tokens). Hide a $6\times 6$ block covering the mug (36 tokens). Context encoder sees the remaining tokens → a set of $d$-dimensional vectors. Predictor, told the $(x,y)$ positions of the missing block via $z$, emits 36 vectors. Target encoder (EMA) sees the full image and you read off those 36 locations. L2 or L1 between predicted and target vectors. Nobody asked you to paint mug pixels. If the 36 vectors mean “ceramic cylinder, handle on the left,” the loss is already happy — which is what you wanted in the kitchen.

Caveats

  • Position paper. Citing it as empirical evidence is a mistake.
  • The six-module agent is a research programme, not a released stack.
  • Collapse prevention is hand-waved toward VICReg / EMA; the later papers pick EMA and an asymmetric predictor.
  • OpenReview, not arXiv. Link the PDF id BZ5a1r-kVsf, not a random mirror.

Study plan

60–75 minutes: kitchen case (10) → abstract + JEPA cartoon vs MAE vs SimCLR (20) → six modules (15) → hierarchical JEPA and “what is not claimed” (10) → start I-JEPA Figure 3 (15). Do not read all 62 pages of speculation on consciousness; the load-bearing pages are the energy cartoons and the module diagram.

Primary sources

BibTeX
@article{lecun2022path,
  title={A Path Towards Autonomous Machine Intelligence},
  author={LeCun, Yann},
  journal={OpenReview},
  year={2022},
  note={Version 0.9.2, 2022-06-27, id BZ5a1r-kVsf}
}