HiLo-WAMDecoupling Long-Horizon Task Evolution and Local Dynamics in World Action Models

Xingyu Wang1,2Yanjiang Guo3Zhijing Ding1Jiaxin Fang4,1Zeyu Lai5,1Jiayu Dong1Weikang Wan1,5Jianyu Chen3Yuxiang Gao1,†Mengdi Xu3,†

1 CoCoMatrix2 Institute of Automation, Chinese Academy of Sciences3 Tsinghua University4 Fudan University5 University of California, San Diego

† Corresponding authors

TL;DR

HiLo-WAM decouples long-horizon task evolution from local visual dynamics. A sparse stream provides historical context and coarse future evolution; a dense stream captures fine-grained robot–object interactions. Together, they guide action generation; progressive inference reduces redundant computation.

HiLo-WAM overview: sparse long-horizon task evolution and dense local dynamics, simulation results, and real-robot manipulation.
Sparse task evolution + dense local dynamics → coordinated action generation.

Abstract

World Action Models (WAMs) improve robotic manipulation by jointly modeling visual dynamics and robot actions. However, most existing WAMs model dense short-horizon visual dynamics, capturing local interactions but providing limited long-range task context. Extending prediction horizons increases uncertainty and computation while entangling task-level evolution with fine-grained local dynamics, obscuring their dependencies and distinct roles in action generation. We propose HiLo-WAM, which decouples long-horizon task evolution from local visual dynamics while jointly modeling their interaction with robot actions. A sparse long-horizon visual stream captures historical task context and future evolution, while a dense local dynamics visual stream models short-horizon robot–object interactions. The two streams operate at different temporal scales within a unified dual-DiT architecture and guide actions through structured cross-stream attention. Progressive inference further reduces unnecessary computation through stream-wise partial denoising, joint denoising, and dynamic skipping. Experiments demonstrate strong long-horizon manipulation performance: HiLo-WAM achieves a 93.70% average success rate on RoboTwin 2.0 and a 17.43 progress score on RoboDojo long-horizon tasks, with robust in-domain and OOD performance on the Fourier GR-3 robot. Ablations confirm the complementary contributions of long-horizon task evolution and local interaction dynamics, while temporal-scale analyses highlight the importance of balancing temporal coverage and resolution. Progressive inference further achieves a 2.93× speedup while improving performance.

Method

Two temporal scales, one coordinated policy

Sample one video at two temporal scales, then jointly refine visual predictions and actions.

Multi-scale visual streams and joint progressive denoising: noise enters below the DiTs and predictions leave above.

One video, sparse and dense views

······ t−mkt−ktt+1t+2t+kt+Tt+nk··············· OUTPUT Shared Video DiTAction DiT Long-horizon visual streamCached Local dynamics visual streamCached Action streamReady Structured cross-stream attention cache cache INPUT HistoryNoise εHCurrentNoise εLAction noise εA nH stepsnL stepsnA steps

Swipe diagram ↔

Visual streams stop and cache before action denoising finishes.

nH ≤ nL ≤ nA
Rollout frames illustrate predictions; actions reuse cached visual latents.

01 / Overall results

Stronger manipulation across horizons

Best RoboTwin success and RoboDojo progress among the compared methods.

RoboTwin 2.0

50 bimanual tasks

93.70%

+1.85 percentage points over Fast-WAM

Fast-WAM91.85
AHA-WAM92.80
HiLo-WAM93.70

Average success rate (%) · axis starts at 90

RoboDojo

8 long-horizon tasks

17.43 progress

+8.29 progress points over Fast-WAM

Fast-WAM9.14
StarVLA-OFT14.28
HiLo-WAM17.43

Full-task success: 6.50% (tied highest).

02 / Ablation studies

Long-horizon context complements local dynamics

RoboTwin 2.0 · Average success rate (%)
Direct history¹Future evolutionLocal dynamicsAvg. SR ↑
90.58
93.22
93.52
93.70

¹ Masked direct history still conditions the long-horizon stream’s future prediction.

+3.12 pp

Full model over local dynamics alone

90.58% → 93.70%

03 / Temporal scale

Balance temporal coverage and sampling resolution

RoboTwin success rates at strides 1, 2, 4, and 8
RoboTwin 2.0 · fixed visual-token budget.

Best tested stride k = 4 · 93.7%

04 / Inference efficiency

2.93× faster, with success preserved

380.02 ms

Per action chunk

93.54% → 93.70%RoboTwin success

Action-chunk latency · Lower is better

  1. Baseline1114.46 ms
  2. + Partial denoising723.65 ms
  3. + Joint denoising619.85 ms
  4. + Dynamic skipping380.02 ms

Test setup. RoboTwin 2.0 · single NVIDIA RTX PRO 5000 GPU.

Timing scope. Steady-state, end-to-end latency per action-chunk inference call, mainly covering Video VAE processing, Video DiT denoising, and Action DiT denoising.

Prompt embeddings are computed once at startup and cached. Speedup is relative to HiLo-WAM’s sequential baseline.

Real-robot demos

Long-horizon real-robot manipulation

Three long-horizon bimanual tasks on Fourier GR-3, requiring consistent progress across multiple assembly or disassembly steps.

Higher average progress and success under scene shifts.

π₀.₅Fast-WAMHiLo-WAM

Average task progress

5.9
5.1
6.9
ID
5.4
2.6
6.0
OOD

Average success rate (%)

23%
20%
33%
ID
27%
0%
30%
OOD

Mean over three tasks · 10 trials per task. OOD: lighting, background, and clutter shifts.

Build House

π₀.₅
Fast-WAM
HiLo-WAM

Disassemble House

π₀.₅
Fast-WAM
HiLo-WAM

Disassemble Heart

π₀.₅
Fast-WAM
HiLo-WAM

Figure