PrimeSim: A Generalist World Model Simulator from Large-Scale Cross-Embodiment Demonstrations
AuthorsTravis Davies, Wentao Ma, Lihao Guo, Tengfei Ma, Xiaowen Chang, Jiafeng Xu, Shilong Wang, Yiqi Deng, Shaowen He, Xiaoqi Li, Qian Zhu, Ran Cheng, Hao Dong
Affiliations[
PrimeSim turns large-scale robot demonstrations into a fast, multiview simulator designed to predict how robots and objects will behave over long action sequences.
Key results
Training hours across 22 robot platforms.
FPS with the four-step sampler.
Maximum reported simulation duration in seconds.
PrimeSim’s overall score, the best reported.
Mean score across five diagnostic suites.
Policy success after 200 GRPO steps inside PrimeSim, up from 61.7%.
What the paper found
PrimeSim turns the Wan2.2-TI2V-5B video generator into an action-conditioned, multiview robot simulator. It midtrains on 30,110 hours of demonstrations across 22 robot platforms, using a shared 31-dimensional action representation, frame-aligned dual-path conditioning, and view-aware interleaved attention to link commanded motion to generated frames while coordinating camera views. Posttraining combines causal generation, four-step consistency distillation, self-rollout training with asymmetric distribution matching, and hierarchical KV-cache memory to limit drift during extended rollouts. PrimeSim runs at 10.1 FPS and produces coherent simulations up to 105 seconds. It scores 6.09 on EWMBench and leads four of five WorldSimProbe suites, with a mean score of 66.6; fine-grained local action calibration remains a weakness. Its simulator-estimated task success correlates with LIBERO results at r = 0.99, and after 200 GRPO steps inside PrimeSim, policy success in LIBERO rises from 61.7% to 73.2%. The paper also compares against systems including Qwen-RobotWorld and OpenAI’s Sora 2, though those instruction- or text-conditioned results are not directly comparable to action-conditioned evaluation.
Original abstract
A robot world model that serves as a simulator must do more than generate plausible videos: it must faithfully follow commanded actions, maintain consistency across camera views, and sustain coherent long-horizon rollouts. We present PrimeSim, an action-conditioned, multiview world model simulator that addresses these challenges through progressive midtraining and posttraining. Starting from a pretrained video generator, we midtrain on 30,110 hours of robot demonstrations spanning 22 platforms. To improve action following, we introduce a unified cross-embodiment action representation and frame-aligned dual-path conditioning to strengthen action-frame correspondence. To maintain multiview consistency, we develop view-aware interleaved attention with explicit camera identities, enabling cross-view information exchange while preserving view-specific representations. We further introduce a fast-video, slow-memory posttraining framework that transforms the bidirectional video generator into an efficient autoregressive simulator. By combining fast causal generation, self-rollout training, and hierarchical temporal memory, PrimeSim mitigates error accumulation from self-generated histories and supports coherent long-horizon simulation. PrimeSim runs at 10.1 FPS, remains stable over long-horizon rollouts on rigid and deformable manipulation tasks, and transfers to held-out platforms within 100 to 200 gradient steps where the pretrained backbone does not. Against five open-source action-conditioned world models it leads four of the five WorldSimProbe suites and posts the best overall EWMBench score, while fine-grained local calibration still trails the field; its rollouts are faithful enough to support GRPO-based policy optimization. We release the model weights, inference code, and training code.
Read the original paperMore in World Models
Browse all 53 papers →World-in-Context: Interactive Video World Modeling with In-Context Conditioning
Harold Haodong Chen, Rui Peng, Rongjin Guo, Lingteng Qiu, Kejie Qiu, Chao Xu, Zilong Dong
World-in-Context makes video world models more controllable by letting them jointly interpret viewpoint, action, and camera instructions.
WorldFlow: Joint Video and 3D Motion Prediction via Large-Scale Pretraining
Yuhao Lu, Jiongxu Chen, Hexiang Wei, Jiajin Wu, Juelin Li, Mingjian Liang, Fan Yang, Zhuowei Liang, Zihao Liu, Zhennan Wang, Sheng Wang, Lin Shao
WorldFlow jointly predicts what the world will look like and how objects will move in 3D.
LaLa-World: Scaling Latent World Models with Natural Language Actions
Junha Song, Geonmo Gu, Jeongeun Park, Heejeong Nam, Lucas Maes, Byeongho Heo, Dongyoon Han, Sangdoo Yun
LaLa-World uses language to train a scalable latent model of how the world changes, then uses its imagined futures to plan robot actions.