JAMJust Action Models

Vision-language-action or world-action?

Just Action Models

Learn from the future. Never generate it in training.
Predict only actions at deployment.

01Method

VLA or WAM? Three choices, unbundled.

A VLA ignores the recorded future, a WAM generates it, and JAM predicts its features.

ABackbone

BLoss on the future

CInference path

Foresight loss

02Target

The best target lies ahead.

Future-only offsets beat the current only.

AArchitecture

BReadout offsets

CFuture anchor

Every evaluated target

03Backbones

Generation helps. Foresight helps more.

Six backbones, three families, three benchmarks: wherever generation is reported, foresight scores higher.

04Rollouts

One fails, one finishes.

Six pairs play in sync from the same initial state: same backbone, different training.

0:00.0
0:00.0
0:00.0

05Probe

Current features hint at what comes next.

Probes reconstruct the frame 16 steps ahead; with future targets, JAM (Ovis-U1) edges out ImageWAM.

0:00.0
How to read

06Attention

JAM attends more to the second mug.

Paper example on Ovis-U1: the instruction names two mugs, and ImageWAM concentrates on the held one.

0:00.0
How to read

07Results

Against VLAs and WAMs.

JAM next to other policies on LIBERO-Plus, LIBERO and RoboTwin 2.0.

Full table
Scroll for all columns

08Compute

Nothing to generate, less to compute.

Generation denoises future frames every update; cached foresight processes only the current frame.

AArchitecture

BForesight targets

09Ablations

Start from a pretrained backbone, keep the target frozen.

A randomly initialized target still works, and robustness gains span six of seven perturbation types.

Stronger when the world shifts

Each dot: a LIBERO-Plus task only one policy solved.

By category
Scroll for all columns

10Cite

Cite Just Action Models.