AD-WM: Action-Discriminative World Models for Counterfactual Model Predictive Control

arXiv 2026 37.4 method

TLDR

Introduces AD-WM, an action-discriminative joint-embedding world model for counterfactual MPC, improving planning success in simulation and real robot transfer by preserving action information.

Reasoning

The paper presents a strong methodological contribution with action-discriminative regularization and demonstrates clear empirical gains across simulation and real-world Franka transfer. Weaknesses include limited visible evidence on generative or video capabilities, and the auxiliary heads are discarded, so broader generality is not established from the abstract alone.

Read-first score

Read-first score 37.4, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 38.

Recency 5%

100

Uses a gentle age decay so recent papers surface without erasing older foundations. 2026

Methodology quality 15%

Screens visible abstract and analysis fields for experiment, dataset, baseline, metric, and limitation evidence. markers=baseline,result

Topical relevance 25%

54.3

Uses existing LLM keyword relevance scores normalized to 0-100. world model,world simulator,generative world model,interactive world model,video world model,world dynamics prediction,model-based reinforcement learning world model

Reproducibility 15%

Screens links and visible text for paper, code, dataset, artifact, and repository signals. pdf=True; code=False; dataset=False; markers=code,github,reproduce

Graph prestige 15%

5.7

PageRank-style prestige over the local citation graph built from available OpenAlex references. local_pagerank=0.0016

Citation impact 15%

Uses OpenAlex-shaped citation metadata as a bibliometric attention signal, separate from paper quality. cited_by_count=0

Citation velocity 10%

Citation velocity estimates citations per publication-year to reduce old-paper bias. velocity=0.00

Field roles

FrontierMethodology anchor

Rank sensitivity

Stability: volatile; rank range: 517.

Keyword Scores

world model

world dynamics prediction

world simulator

interactive world model

model-based reinforcement learning world model

generative world model

video world model

Deep Analysis

Innovations

Introduces AD-WM, an action-discriminative joint-embedding world model for counterfactual model predictive control.
Combines residual latent dynamics with predictor-level action-recovery regularization.
Uses inverse dynamics and a normalized recovery objective motivated by conditional mutual information to preserve action information in planning transitions.
Auxiliary action-recovery heads are discarded at test time, leaving MPC unchanged.

Methodology

AD-WM is a latent world model with residual latent dynamics and auxiliary action-recovery heads trained using inverse dynamics and a normalized recovery objective. The auxiliary heads are removed at test time. It is evaluated on OGBench-Cube, five simulation environments against a matched LeWM/reproduced baseline, and zero-shot transfer to a Franka setup with a frozen V-JEPA 2 encoder and matched DROID post-training.

Key Results

AD-WM improves OGBench-Cube hard-start success from 3.7% to 52.0% over a matched LeWM baseline and improves mean success over the reproduced baseline in four of five simulation environments. CEM-aligned elite regret tracks closed-loop success more closely than factual prediction error or whole-bank action ranking, and zero-shot Franka basic pick-and-place success increases from 42.2% to 71.1%.

Limitations

The reported improvement in simulation is not universal: mean success improved in four of five simulation environments.
The zero-shot transfer result is shown only for basic pick-and-place on the Franka setup.
The abstract does not state other explicit limitations or failure modes.

Resources

Paper