Source-linked AI summary

{\Delta}WAM: Distilling Action Tangent Fields into World Action Models

Ke Wu, Hanwen Huang, Bo Gu, Kaizhao Zhang, Xiangting Meng, Yupeng Zheng, Zijun Xu, Jieru Zhao, Wenchao Ding

arXiv:2610.09734v1cs.RO

TL;DR

World Action Models can devote prediction capacity to persistent appearance rather than action-dependent dynamics. ∆WAM preserves a recoverable future representation and uses local action probes to emphasize responsive directions; it improves robustness across benchmarks, though camera and layout shifts remain challenging.

  • Problem

    Future-video supervision can be dominated by persistent appearance and scene content rather than action-dependent changes, while motion-based targets may miss action sensitivity and discard useful cues.

    Source: §1 · p. 2
  • Method

    ∆WAM represents future dynamics with a recoverable Residual-VAE target and uses perturbed-action ACWM predictions to estimate and emphasize locally action-responsive directions.

    Source: §1 · p. 2
  • Results

    Across LIBERO-Plus, RoboTwin, and RoboTwin 2.0-Plus, robustness improves; on LIBERO-Plus, aggregate success is 87.8% versus 51.5% for Fast-WAM.

    Source: §1 · p. 2, §5 · p. 9
  • Takeaways & Limitations

    Effective WAM supervision can retain the full future while emphasizing locally action-responsive directions, with one VideoDiT pass at inference.

    Source: §5 · p. 9

    Large camera and layout changes remain challenging.

    Source: §4.1 · p. 7, §5 · p. 9

Abstract

from arXiv · show

World Action Models (WAM) improve robot policies by augmenting sparse action supervision with dense future prediction. However, much of the predictable future is dominated by appearance and scene persistence rather than action-dependent dynamics. We observe that several recent WAM designs, including optical flow, motion-centric representations, and latent actions, can be understood from a common perspective in which world supervision becomes more efficient as it contains a higher proportion of action-relevant variation. Based on this insight, we introduce Action Tangent Fields, which reformulate world supervision through a local Taylor expansion of how actions induce changes in future dynamics. We represent future dynamics in Residual-VAE space, where the future latent remains recoverable from the current latent and its residual, and use a strong action-conditioned world model (ACWM) to probe the local correspondence between action variations and residual-world variations. This local first-order structure is distilled into the WAM to guide its denoising supervision toward dynamics that are more tightly coupled to action, rather than merely predictable from appearance. Across LIBERO-Plus, RoboTwin, and RoboTwin2.0-Plus, our method consistently improves robustness to lighting, background, camera, layout, and other environmental perturbations. Despite using no large-scale embodied pretraining, it achieves stronger robustness under several distribution shifts than pretrained policies. We further distill multi-step VideoDiT denoising into a single step for efficient inference. Our results suggest that effective WAM supervision should remain information-rich while concentrating its predictive capacity on the directions along which actions change the future.

1 INTRODUCTION

World supervision can spend capacity on persistent appearance rather than action-dependent dynamics. ∆WAM retains an information-rich future target while emphasizing directions that respond to nearby actions.

  • Future-video prediction provides dense supervision, but persistent backgrounds, camera views, and untouched objects can dominate loss without teaching action consequences.Source: §1 · p. 2
  • Motion-centric targets reduce static-appearance supervision but can mistake lighting or distractor motion for action dependence and discard contact-relevant appearance cues.Source: §1 · p. 2
  • The Residual-VAE target subtracts a current-frame-derived reference from the future latent, preserving exact recovery while making scene departure explicit.Source: §1 · p. 2
  • Action Tangent Fields estimate locally action-responsive residual directions from perturbed actions and upweight those directions while retaining the demonstrated future as target.Source: §1 · p. 2
  • The method mid-trains Wan on robot video, then post-trains a Fast-WAM-style policy with cached ATFs, keeping the ACWM out of policy training and inference.Source: §1 · p. 2
  • ∆WAM reaches 87.8% vs 51.5% aggregate success on LIBERO-Plus and 79.3% vs 72.7% on RoboTwin 2.0-Plus, using one VideoDiT evaluation at inference.Source: §1 · p. 2

2 RELATED WORK

Prior WAMs vary in how they generate or represent future observations, with target choice affecting how capacity is allocated. ∆WAM uses an ACWM as a local differential probe rather than planning with generated rollouts or supervising on synthetic futures.

2.1 WORLD ACTION MODELS

2.2 ACTION-CONDITIONED WORLD MODELS

  • Action-conditioned world models broaden control interfaces through action tokens, pixel-space flow, or rendered action observations (Li et al., 2026a; Zou et al., 2026).Source: §2.2 · p. 3
  • Unlike rollout-oriented simulation methods, ∆WAM perturbs demonstrated actions, transfers local action–world geometry from an ACWM, and keeps real demonstrations as prediction targets.Source: §2.2 · p. 3

3 METHOD

∆WAM trains a VideoDiT–ActionDiT policy on real action–future pairs, transferring locally action-responsive directions from an embodiment-specific ACWM into world-error weighting. It predicts future latents relative to the observed frame and computes world features with one VideoDiT pass at inference.

3 METHOD

  • ∆WAM trains a VideoDiT–ActionDiT policy on real action–future pairs, transferring locally action-responsive directions from an embodiment-specific ACWM into world-error weighting.Source: §3 · p. 3

3.1 RESIDUAL VIDEO MID-TRAINING

  • The first-frame anchor plus the future residual exactly recovers the absolute VAE latent, while centering prediction on scene changes such as robot motion, object displacement, and contact.This preserves the VAE representation, not necessarily the original RGB pixels.Source: §3.1 · pp. 3–4

3.2 ACTION TANGENT FIELDS FROM LOCAL DYNAMICS

  • An embodiment-specific ACWM probes matched-condition nearby actions; normalized finite differences span sampled action-responsive residual directions, which are cached instead of estimating the full Jacobian.The cached basis means the ACWM is not needed during policy training or inference.Source: §3.2 · p. 5

3.3 ATF-GUIDED WORLD-ACTION POST-TRAINING

  • Post-training retains the real demonstrated residual target and adds weight to errors projected onto sampled action-responsive directions, while ordinary residual loss covers every latent element.The added projection term is zero when no sampled directions are available.Source: §3.3 · p. 6

3.4 SINGLE-PASS WORLD CONDITIONING AT INFERENCE

  • At inference, one VideoDiT pass produces world tokens that are reused while ActionDiT integrates the action chunk through four descending Euler steps.The image encoding and VideoDiT features remain fixed across those steps.Source: §3.4 · p. 6

4 EXPERIMENTS

The experiments evaluate ∆WAM across LIBERO-Plus and RoboTwin perturbation benchmarks and real-robot tasks. Results show strong aggregate and cross-embodiment performance, alongside specific weaknesses and ablation trade-offs.

4.1 LIBERO-PLUS

  • 87.8% aggregate success on LIBERO-Plus is 2.1 points above the strongest baseline, π0.5; original-setting success is 98.2%, near the best reported 98.6%.The aggregate is pooled across perturbation instances.Source: Table 1 · p. 7, §4.1 · p. 7
  • LIBERO-Plus leads under robot-initialization shifts (96.2%, 6.5 points above the best competitor) and language shifts (91.2%), but camera success (60.9%) trails the best by 14.9 points.Lighting reaches 86.8% and layout 58.4%; background performance is 94.1%, although π0.5 and X-VLA remain ahead.Source: §4.1 · p. 7

4.2 ROBOTWIN AND ROBOTWIN 2.0-PLUS

  • 92.8% on Clean and 92.1% on Randomized yield the best RoboTwin average, 92.4%; this is within 0.1 points of pretrained LingBot-VA on Clean and 0.6 points ahead on Randomized.The Clean–Randomized gap is 0.7 points.Source: §4.2 · p. 7, Table 2 · p. 8
  • 79.3% aggregate success on RoboTwin 2.0-Plus is 5.1 points above LingBot-VA; the method leads on language (88.8%), lighting (91.4%), and layout (91.3%), but camera success is 19.1%.It also scores 92.2% in the original setting; background and observation-noise results trail the best by 0.1 and 2.1 points.Source: Table 2 · p. 8, §4.2 · p. 8

4.3 REAL-WORLD EXPERIMENTS ACROSS EMBODIMENTS

  • On NERO, ∆WAM averages 0.79 on test-tube insertion versus 0.58 for π0.5 and 0.54 on bowl stacking versus 0.33 for both baselines; on Tianji it leads every task average.Its largest Tianji margin is on bimanual cooking, where baselines score at most 0.02; Table 3 averages three trials per task and condition.Source: §4.3 · p. 8, Table 3 · p. 9

4.4 ABLATION STUDY

  • 87.8% average success is reached with one or two VideoDiT denoising steps, versus 88.1% with four, so the method uses one step.Inference time rises from 0.43 s at one step to 1.57 s at four; the three extra steps buy 0.3 points at 1.14 s additional latency.Source: Figure 4 · p. 9, §4.4 · p. 9
  • Success rises from 79.0% at K = 0 to 85.1% at K = 4 and 87.8% at K = 8, motivating the use of eight local action perturbations.K = 0 retains the real action and residual targets but omits the ATF loss.Source: Figure 4 · p. 9, §4.4 · p. 9

5 CONCLUSION

∆WAM predicts future video dynamics and their variation near the demonstrated action, improving robustness across three benchmarks while large camera and layout changes remain challenging.

5 CONCLUSION

  • Robustness gains hold on LIBERO-Plus, RoboTwin, and RoboTwin 2.0-Plus, though large camera and layout changes remain challenging.Source: §5 · p. 9
  • Residual-VAE prediction makes temporal change explicit while retaining the full future latent.Source: §5 · p. 9
  • The ATF loss upweights errors along directions responsive to nearby actions while keeping the real future as the target; one VideoDiT pass suffices at inference.Source: §5 · p. 9
Loading 2610.09734v1…