English

ST-WAM: Semantic-Temporal World Action Model for Robust Manipulation under Visual Distribution Shifts

Robotics 2026-07-31 v1 Computer Vision and Pattern Recognition

Abstract

World Action Models (WAMs) have emerged as a promising paradigm by jointly modeling robot actions and future visual dynamics. However, their reliance on pixel-generative future supervision can entangle action-relevant state transitions with task-irrelevant visual content, limiting robustness under visual distribution shifts. We identify Training-Distribution Hallucination, a recurring phenomenon in which futures conditioned on visually shifted observations hallucinate training-domain content rather than remain faithful to the current scene. A controlled frame-triplet diagnosis further shows that DINOv3 features remain more stable across visual shifts while better preserving task-state distinctions than Wan-VAE latents. Rather than correcting the predicted futures, we propose Semantic-Temporal WAM (ST-WAM) to improve action robustness by using DINOv3 as a shared semantic representation for future prediction and history retrieval while retaining fine-grained VAE dynamics. Its Dual-Space Future Experts (DSFE) jointly predict future VAE latents and DINO features, while Current-Anchored Intent Retrieval (CAIR) retrieves task-relevant evidence from recent DINO history under the current visual-language context. ST-WAM is trained end-to-end without additional embodied pretraining or task-specific annotations, and requires no explicit future generation at inference. It achieves 98.7% on LIBERO and 92.8% on RoboTwin 2.0; more importantly, compared with Fast-WAM, it improves zero-shot LIBERO-Plus performance by 21.3 percentage points and more than doubles real-world success under visual shifts from 25.8% to 61.5%. These results demonstrate that semantic-temporal modeling effectively complements pixel-generative dynamics for robust manipulation.

Cite

@article{arxiv.2607.28993,
  title  = {ST-WAM: Semantic-Temporal World Action Model for Robust Manipulation under Visual Distribution Shifts},
  author = {Mingxin Wang and Bin Hu and Bin Qian and Kaitao Jiang and Haoning Wu and Feng Yan and Bowen Jing and Ruiyang Hao and Enyi Wang and Kangning Niu and Yandan Yang and Mu Xu and Yan Wang and Houde Liu and Tianlun Li},
  journal= {arXiv preprint arXiv:2607.28993},
  year   = {2026}
}

Comments

9 pages, 5 figures