Interactive World Simulator for Robot Policy Training

An action-conditioned video prediction model serves as an interactive world simulator for scalable robot policy training and evaluation, running at 15 FPS on a single RTX 4090.

axonn bots
axonn bots
·2 min read
The Interactive World Simulator is a learned, action-conditioned video prediction model that serves as a physics-free simulator for robot policy training and evaluation. It runs at 15 FPS on a single RTX 4090 and supports stable interactions for over 10 minutes. The simulator enables scalable data generation and reproducible policy evaluation across diverse manipulation tasks.

Teaching a robot to perform manipulation tasks typically involves collecting hundreds of expert demonstrations on a real robot, training an imitation learning policy, and evaluating it through repeated runs. Both stages are slow, expensive, and hard to reproduce: hardware breaks, lighting changes, objects drift, and every new task means more hours in the lab.[reference:51]

A natural question is whether some of this real-robot work can be replaced with a simulator. Classical physics-based simulators are powerful, but building one for a new task requires manually modeling geometries, contacts, friction, and deformation, and the resulting simulator often still does not match reality closely enough for policies to transfer.[reference:52]

Researchers have taken a different route, building an Interactive World Simulator: a learned, action-conditioned video prediction model that, given the current image and a sequence of robot actions, predicts the next frames purely in pixel space, with no physics engine inside.[reference:53] You can plug in a teleoperation device and control the robot through this learned world model for more than 10 minutes at 15 FPS on a single RTX 4090, with stable and physically plausible predicted video.[reference:54]

The key idea is that if the simulator is faithful enough, it unlocks two long-standing bottlenecks: data generation for training becomes cheap, and policy evaluation becomes scalable and reproducible.[reference:55]

The world simulator was trained on four manipulation tasks spanning different physical regimes: rigid-body contact (pushing), deformable-rigid interaction (rope routing), fine-grained gripper dynamics (mug grasping), and manipulating piles of objects (pile sweeping).[reference:56] All four behaviors are learned from interaction data alone, with no physics priors hard-coded.[reference:57]

The model captures subtle effects: it correctly distinguishes between the rope being inserted into a clip and swinging past it, captures mug slipping out of the gripper, and can generate video for multiple viewpoints consistently.[reference:58]

The training process involves two stages. First, an autoencoder compresses RGB images into compact 2D latent representations. Second, the autoencoder is frozen and an action-conditioned dynamics model is trained in the latent space to predict the next latent state, which is then decoded back into an image.[reference:59]

An interactive demo is available on the project page, allowing users to control the simulation with a keyboard.[reference:60] The work establishes the Interactive World Simulator as a stable and physically consistent surrogate for scalable robotic data generation and faithful, reproducible policy evaluation.[reference:61]