Remembering What Matters for Long-Horizon Robot World Models
Abstract
Autoregressive robot video world models based on discrete visual tokens remain unreliable over long rollout horizons. Retaining the entire generated history does not necessarily improve long-horizon prediction quality. We investigate a possible memory-role conflict in Full Context decoding: persistent scene information, recent visual–action states, and stale generated history compete for attention despite their different predictive roles. This perspective motivates (Receding Context with Anchor Preservation), a training-free context-scheduling method for long-horizon robot video world models. keeps the pretrained tokenizer, world model, and visual decoder frozen. At inference time, it preserves the initial scene anchor and a contiguous window of recent complete visual–action blocks while evicting older generated history. The schedule preserves within-block structure and temporal order, requires no additional modules or parameter updates, and bounds active-context growth independently of rollout length. We evaluate on RT-1, BridgeV2, LIBERO-90, and CALVIN using frozen autoregressive world models. Across the evaluated environments and rollout horizons, improves visual fidelity, action-conditioned consistency, and robot structure preservation relative to the corresponding baselines. On BridgeV2, it reduces LPIPS by . In the system benchmark at , it reduces the logical KV footprint by and p50 latency by . Attention analyses show increased attention to retained recent states, while history interventions associate larger amounts of stale context with higher prediction error. These observations are consistent with the memory-role-conflict hypothesis, although they do not isolate it from context-length and position-handling effects. Our results support inference-time context scheduling as a practical approach to improving long-horizon robot video prediction and computational efficiency without retraining.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.