arXiv AI
4d ago

Rollback: Experience Backtracking in World Models for Policy Improvement

The paper introduces Rollback, a method that uses action‑conditioned world models to backtrack to critical decision points, branch into alternative actions, and compare outcomes to attribute success to individual decisions. By turning world models into counterfactual engines, Rollback provides fine‑grained, action‑level supervision without a separate value model and integrates distributed world‑model search with asynchronous real‑robot execution. Experiments on block extraction, ball tossing, and tabletop curling show that Rollback improves success rates by over 10% compared to the strongest baseline and surpasses human performance on two tasks within eight hours of training.

By Jiayi Luo, Yang Zhang, Liliang Chen, Nan Wang, Chen Gao, Boxiang Qiu, Zhuoxuan Li, Baorui Peng, Chuankang Li, Yuxiang Yan, Si Liu, Qingyun Sun, Shuicheng Yan, Maoqing Yao, Jianxin Li, Guanghui Ren
arXiv AI
Jun 30

WoVR: World Models as Reliable Simulators for Post-Training VLA Policies with RL

arXiv:2602. 13977v2 Announce Type: replace-cross Abstract: Reinforcement learning (RL) promises to unlock capabilities beyond imitation learning for Vision--Language--Action (VLA) models, but its requirement for massive real-world interaction prevents direct deployment on physical robots.

By Zhennan Jiang, Shangqing Zhou, Yutong Jiang, Zefang Huang, Mingjie Wei, Yuhui Chen, Tianxing Zhou, Zhen Guo, Hao Lin, Quanlu Zhang, Yu Wang, Haoran Li, Chao Yu, Dongbin Zhao
arXiv Computer Vision
Aug 26

Do Robotic World Models Really Follow Actions? Diagnosing and Aligning Action-Conditioned Generation for Policy Learning

arXiv:2608.24885v1 Announce Type: cross Abstract: Action-conditioned world models are increasingly used as learned simulators for policy evaluation and improvement, yet their effectiveness rests on a...

By Sixiang Chen, Jiaming Liu, Jixian Wu, Yichen Guo, Tinghao Wang, Siyuan Qian, Hao Chen, Jiajun Cao, Jian Tang, Shanghang Zhang
arXiv AI
Oct 1

RoboCoach: World Models as Active Coaches for Compositional Robot Skills

RoboCoach introduces a world-model-guided coaching framework that uses imagined failures to direct demonstration requests and expert updates for robot manipulation tasks. The system, called RIDI, runs reusable skill experts within a shared action-conditioned world model and records the first failing subtask to decide which demonstrations to acquire and which adapters to refine. Experiments on simulation suites and real robots show that with only 150 additional subtask demonstrations, success rates increase dramatically, and the coached experts transfer effectively to unseen task compositions.

By Jiajun Liu, Yifan Chen, Yichao Liu, Jiayi Zhang, Ruoqu Chen, Shaoxuan Xie, Guocai Yao, Mengdi Xu, Sen Cui, Changshui Zhang