ResWAM: Learning What Changes, Not What Stays, for World-Action Models

Chongyu Fan1, Gaowen Liu2, Ramana Rao Kompella2, Jason Corso3, Sijia Liu1,4

1Michigan State University · 2Cisco · 3University of Michigan · 4IBM Research

00 · Abstract

Abstract

World-action models (WAMs) augment vision-language-action policies (VLAs) with future prediction so that the policy can anticipate how its actions will affect the physical world. Existing WAMs supervise complete future observations. However, observations typically change little over an action chunk, so the future loss is dominated by copying static background content rather than learning task-relevant robot and object motion, limiting policy learning. Moreover, existing WAMs rely on large video generation models or separate world models, incurring heavy training and inference compute that hinders closed-loop deployment. To improve policy performance, we propose visual residual learning, supervising only the feature change from the current frame to the future frame in a frozen DINOv3 space. Compared with complete future features, this residual learning target yields a higher feature rank with richer and more diverse representations of scene changes, and spatially concentrates on the robot arm and manipulated objects rather than the background. To further improve computational efficiency, we present ResWAM, a residual world-action model with 2.3B parameters in total, including a lightweight 0.5B dual-branch DiT that jointly trains an action branch and a residual branch via conditional flow matching. Without robot pretraining (i.e., pretraining on external robot action trajectories beyond the benchmark demonstrations), ResWAM achieves top average success rates on LIBERO-Plus (81.1%), RoboTwin 2.0 (93.8%), and RoboDojo (9.40% success, 13.41 score) at the lowest training (2.7–4.0 TFLOPs) and inference compute (1.0–1.4 TFLOPs), and leads real-robot Open, Put, and Stack (80.0% average, +10.0% over π0.5) at 92.7 ms latency.

01 · Contributions

Contributions

  1. 1

    Supervision-wise, to strengthen the supervision signal for higher performance, we propose visual residual learning. We show that this signal exhibits higher feature rank than full future features and concentrates on the robot and manipulated objects rather than static backgrounds, providing a more focused and expressive supervision signal.

  2. 2

    Architecture-wise, to improve computational efficiency, we design ResWAM, a residual world-action model with 2.3B parameters in total, including a lightweight 0.5B dual-branch DiT that jointly denoises actions and frozen DINOv3 visual residuals via conditional flow matching in place of video generation models or separate world models.

  3. 3

    Experiment-wise, without robot pretraining, ResWAM achieves the highest average success rates on LIBERO-Plus (81.1%), RoboTwin 2.0 (93.8%), and RoboDojo (9.40% success, 13.41 score) at the lowest training (2.7–4.0 TFLOPs) and inference compute (1.0–1.4 TFLOPs), and leads real-world FR3 tasks (80.0% average, +10.0% over π0.5) at 92.7 ms latency.

02 · Visual residual learning

What Is the Residual?

We extract visual patch tokens with a frozen DINOv3 encoder.

The visual residual target is the feature change from time t to time t+H.

Δt = E(ot+H) − E(ot)
Predictions of VLAs, WAMs, and ResWAM

02 · Visual residual learning

Two findings about the residual

1. The residual concentrates on the robot and the manipulated objects.

LIBERO

put both moka pots on the stove

open the top drawer and put the bowl inside

pick up the black bowl in the top drawer of the wooden cabinet and place it on the plate

RoboTwin 2.0

Place the light blue bowl with darkened rim onto the light blue bowl with darkened rim to form a stack.

Pass the red block to the right arm and set it down.

Locate the medium gray microwave with top vents and open it using the left arm.

2. The residual has higher effective rank and stable rank.

Let JX be the token-centered feature matrix, where J = IN − (1/N)11⊤, and let pi = σi2 / Σjσj2.

Effective rank

reff(X) = exp(−Σi pi log pi)

Stable rank

rstable(X) = (Σiσi2) / σ12

Higher values mean that spectral energy spans more feature directions of scene change instead of copying static background modes. Both measures are scale-invariant.

Effective and stable rank comparison between future and residual DINOv3 tokens
Figure 1. Effective rank (left) and stable rank (right) of the future DINOv3 tokens Zt+H and the residual Δt, averaged over all windows of LIBERO and RoboTwin 2.0.

03 · Model architecture

ResWAM at a glance

ResWAM architecture overview with VLM, frozen DINOv3 encoder, and coupled Action DiT and Residual DiT branches
Figure 2. ResWAM architecture overview. A VLM encodes the current images ot and instruction c. The action and residual branches denoise the predicted action chunk Ât and residual Δ̂t by cross-attending to the VLM output and the current DINOv3 tokens Zt, exchanging information via joint attention under flow-matching losses La and Lr.
  • 2.3B parameters in total: a lightweight 0.5B dual-branch DiT (action branch and residual branch) attached directly to a pretrained VLM (RynnBrain-1.1-2B) and a frozen DINOv3 ViT-B/16 encoder (2×2-pooled to N = 64 tokens per view).
  • Training via conditional flow matching: the future frame ot+H is used only during training to form the frozen DINOv3 residual target Δt = Zt+H − Zt (+2.9% over direct future prediction on RoboTwin 2.0 at identical compute).
  • Single-pass latent inference: at test time, only ot, proprioceptive state qt, and instruction c are required, integrating actions and compact residual tokens over 10 Euler steps without pixel reconstruction or 3D video VAEs.

04 · Experiments

Headline: best average among compared methods

LIBERO-Plus success versus latency, RoboTwin 2.0, and RoboDojo average success comparison

81.1%LIBERO-Plus (10,030 episodes)+1.6 pts over VLA-JEPA (79.5%)

93.8%RoboTwin 2.0 (50 tasks)+1.6 pts over LingBot-VA (92.2%)

9.40% / 13.41RoboDojo (SR / Score)+2.49 pts SR over π0.5 (6.91%)

80.0%Real Robot FR3 (92.7 ms)+10.0 pts over π0.5 (70.0%)

2.3B parameters · 2.7–4.0 TFLOPs Train
1.0–1.4 TFLOPs Infer. · 92.2–99.7 ms · w/o Robot Pretrain

04 · Experiments · Table 1

LIBERO-Plus: zero-shot out-of-distribution generalization

Without robot pretraining and with 2.3B parameters, ResWAM achieves an 81.1% overall success rate across all 10,030 perturbed episodes at the lowest training compute (2.7 TFLOPs), inference compute (1.0 TFLOPs), and latency (92.2 ms), outperforming VLA-JEPA (79.5%, 5.7 / 1.2 TFLOPs, 192.3 ms) and LaWAM (76.6%, 3.2 / 1.2 TFLOPs, 198.1 ms).

Efficiency and success rates (%) on LIBERO-Plus across the seven perturbation dimensions and their overall average (Avg.) over all 10,030 perturbed episodes. All policies are trained on standard LIBERO demonstrations (without exposure to LIBERO-Plus perturbations). Robot Pretrain indicates whether a policy uses external robot action datasets beyond the benchmark demonstrations (✓) or is trained solely on benchmark demonstrations (✗). Size is the number of parameters in billions loaded at inference. Efficiency comprises per-sample training (Train) and per-chunk inference (Infer.) compute in TFLOPs, and per-chunk inference Latency in ms on one NVIDIA RTX A6000. The best result in each column is in bold and the second best is underlined.

Method Robot
Pretrain
Size Efficiency Success Rate (%)
TrainInfer.Latency CameraRobotLanguageLightBackgroundNoiseLayoutAvg.
Vision-language-action models (VLAs)
OpenVLA ✓ 7.5 13.05.0235.10.83.523.08.134.815.228.515.6
Mantis ✓ 5.8 19.35.6311.139.215.741.862.528.945.945.139.8
UniVLA ✓ 7.6 9.44.5166.31.846.269.669.081.021.231.942.9
π0 ✓ 3.2 11.84.4112.213.86.058.885.081.479.068.953.6
π0.5 ✓ 3.4 13.54.7120.143.249.471.192.886.248.079.465.8
OpenVLA-OFT ✓ 7.7 18.48.9132.456.431.979.588.793.375.874.269.6
GR00T-N1.6 ✓ 3.3 3.22.3244.964.438.483.994.993.784.474.974.9
VLA-JEPA ✓ 2.8 5.71.2192.363.367.185.495.693.666.385.179.5
World-action models (WAMs)
Motus ✓ 14.4 16.076.12769.517.943.374.154.744.926.252.044.0
LingBot-VA ✓ 11.5 14.264.62903.740.983.086.482.353.164.476.269.5
LaWAM ✓ 2.6 3.21.2198.137.271.687.193.594.169.483.376.6
Fast-WAM ✗ 12.4 12.03.7142.816.444.568.978.253.737.760.751.5
Cosmos-Policy ✗ 2.1 29.376.81123.969.742.486.390.977.575.665.271.5
ResWAM (Ours) ✗ 2.3 2.71.092.252.371.089.997.798.385.683.281.1

Bold: best in column. Underline: second best. Train and Infer. are in TFLOPs; Latency is in ms on one NVIDIA RTX A6000.

04 · Experiments · Table 2

RoboTwin 2.0: 50 bimanual tasks

ResWAM achieves 93.6% under clean conditions and 94.0% under domain-randomized conditions (93.8% average) at the lowest training compute (4.0 TFLOPs), inference compute (1.4 TFLOPs), and latency (99.4 ms), ranking first in every column without robot pretraining.

Efficiency and success rates (%) on RoboTwin 2.0 (50 tasks, 100 trials per task) under clean and randomized (Random) conditions. Notation as in Table 1.

Method Robot
Pretrain
Size Efficiency Success Rate (%)
TrainInfer.Latency CleanRandomAvg.
Vision-language-action models (VLAs)
π0 ✓ 3.2 11.84.4111.965.958.462.2
π0.5 ✓ 3.4 13.65.0124.782.776.879.8
World-action models (WAMs)
Motus ✓ 14.4 16.176.62902.588.787.087.8
LaWAM ✓ 2.6 4.11.5220.992.689.891.2
LingBot-VA ✓ 11.5 23.1114.74473.792.991.692.2
Fast-WAM ✗ 12.4 14.64.0148.891.991.891.8
ResWAM (Ours) ✗ 2.3 4.01.499.493.694.093.8

04 · Experiments · Table 3

RoboDojo: 42 bimanual tasks across five capability dimensions

Without robot pretraining, ResWAM achieves 9.40% average success rate and 13.41 progress score alongside the lowest training compute (3.9 TFLOPs), inference compute (1.4 TFLOPs), and latency (99.7 ms), outperforming π0.5 (6.91% / 11.41, 13.6 / 5.0 TFLOPs, 125.4 ms) and nearly tripling GigaWorld-Policy (3.27% / 6.20, 34.6 / 2.5 TFLOPs, 666.3 ms).

Efficiency, success rates (SR, %), and stage-level progress scores (Score) on the RoboDojo simulation benchmark across the five capability dimensions (42 tasks, 54 configurations, 2,100 episodes) and their macro-average (Avg.). Notation as in Table 1.

Method Robot
Pretrain
Size Efficiency Generalization Precision Long-Horizon Memory Open Avg.
TrainInfer.Latency SR Score SR Score SR Score SR Score SR Score SR Score
Vision-language-action models (VLAs)
OpenVLA-OFT ✓ 7.7 35.7 17.2 237.7 0.00 0.04 0.00 0.20 0.00 0.70 0.00 0.00 0.08 0.08 0.02 0.21
InternVLA-A1 ✓ 3.2 9.0 2.2 422.5 1.83 2.87 0.92 3.00 1.17 4.79 1.33 1.58 0.17 0.17 1.08 2.48
GR00T-N1.6 ✓ 3.3 6.1 3.6 259.8 1.22 2.16 0.67 2.54 3.58 8.30 0.89 1.06 0.17 0.18 1.31 2.85
π0 ✓ 3.2 11.8 4.4 112.5 2.56 3.94 0.75 3.56 2.00 6.19 2.11 3.47 0.25 0.25 1.53 3.48
LingBot-VLA ✓ 4.2 6.7 2.6 740.4 4.28 6.71 1.83 5.33 5.25 10.89 2.78 3.82 0.67 0.72 2.96 5.50
π0.5 ✓ 3.4 13.6 5.0 125.4 8.17 13.37 5.50 12.40 14.67 23.54 4.56 5.78 1.67 1.98 6.91 11.41
StarVLA-α ✗ 4.6 7.0 2.6 192.8 2.33 3.93 4.33 9.90 6.50 14.15 2.44 3.34 0.58 0.68 3.24 6.40
World-action models (WAMs)
LDA-1B ✓ 5.4 5.9 3.7 358.2 0.17 0.71 0.50 3.21 0.08 1.92 1.78 2.08 0.00 0.00 0.51 1.58
Fast-WAM ✗ 12.4 14.6 4.0 148.3 1.11 2.34 0.00 1.96 5.17 9.14 3.44 3.55 0.42 0.42 2.03 3.48
AHA-WAM ✗ 13.7 33.8 3.2 637.0 3.28 5.79 2.42 5.86 2.67 8.61 2.78 2.97 0.83 0.88 2.39 4.82
GigaWorld-Policy ✗ 12.4 34.6 2.5 666.3 2.89 5.34 1.83 6.15 8.92 15.51 2.22 3.46 0.50 0.54 3.27 6.20
ResWAM (Ours) ✗ 2.3 3.9 1.4 99.7 6.50 9.12 14.75 21.31 15.75 25.52 9.00 9.97 1.00 1.14 9.40 13.41

04 · Experiments · Table 4

Real-World Manipulation on Franka Research 3

ResWAM achieves the highest success rate on all three tabletop tasks (Open 90.0%, Put 96.7%, Stack 53.3%) with the lowest training compute (2.8 TFLOPs), inference compute (1.0 TFLOPs), and latency (92.7 ms), averaging 80.0% across 30 trials per task (+10.0% over π0.5 at 70.0%, 5.0 TFLOPs, and 120.4 ms).

Efficiency and success rates (%) on the three real-world manipulation tasks (30 trials per task) on a Franka Research 3 (FR3) arm. Notation as in Table 1.

Method Robot
Pretrain
Size Efficiency Success Rate (%)
TrainInfer.Latency OpenPutStackAvg.
Vision-language-action models (VLAs)
GR00T-N1.6 ✓ 3.3 3.22.3245.560.076.723.353.3
π0.5 ✓ 3.4 13.65.0120.480.086.743.370.0
World-action models (WAMs)
LaWAM ✓ 2.6 3.21.2198.880.090.036.768.9
Fast-WAM ✗ 12.4 12.03.7143.656.763.330.050.0
ResWAM (Ours) ✗ 2.3 2.81.092.790.096.753.380.0
Three prompt-only real-robot manipulation demonstrations

05 · Citation

Citation

BibTeX

@article{fan2026reswam,
  title={ResWAM: Learning What Changes, Not What Stays, for World-Action Models},
  author={Fan, Chongyu and Liu, Gaowen and Kompella, Ramana Rao and Corso, Jason and Liu, Sijia},
  journal={arXiv preprint},
  year={2026}
}