cs.AIAug 23, 2026

WAM-OPD: Joint Video-Action Supervision for World Action Model Post-Training with On-Policy Distillation

Authors: Liuhaichen Yang, Zhengyang Zhong, Hanshang Zhu, Ningwei Bai, Qichen Yin, Zhi Han, Jiarui Qin, Zhuang Jiang, +8 more

Organizations: University College London · Shanghai Jiao Tong University · The University of Manchester · Chinese Academy of Sciences, Nanjing Institute of Software Technology · Shenzhen University of Advanced Technology · University of Liverpool

Abstract

World Action Models (WAMs) generate both future video and robot actions, offering two connected outputs for post-training supervision. How can a pretrained WAM learn from a stronger Teacher on the histories it encounters during execution? We present WAM-OPD, which collects Student rollout histories and queries a Teacher for paired video and action targets. The Student learns from both targets while retaining its one-step video and action generation at deployment. Across 12 RoboTwin 2.0 tasks, WAM-OPD improves average success from 33.8% to 65.7%; across four real-robot tasks, it improves average success from 51.4% to 64.6%. With the collected Student histories held fixed, joint video-action supervision achieves the highest observed success on all three ablation tasks, while either modality alone also improves performance. A separate comparison with Teacher-generated histories finds task-dependent differences between the two history sources. These results demonstrate the value of paired video-action supervision for improving WAM policies without increasing their deployed sampling budget.

Figures & tables

Explore similar work

CardsList
  1. Vid2WAM: Distilling Video Diffusion Priors into World Action Models

    Aug 9, 2026Chenhao Qiu, Ruixiang Wang, Runyi Zhao +7Cross-Modal Knowledge DistillationWorld Model Learning

  2. RAP-WAM: Learning World-Action Models through Predictive Rehearsal and Action Perturbations

    Oct 12, 2026Yuanpeng He, Ceyao Zhang, Fangjing Li +8Visuomotor Policy LearningWorld Action Models