Robot-Factored World Models via Robot Rendering

· Source: cs.CV updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Expert, extended

Summary

Robot-factored world models are proposed to enhance action-conditioned video prediction in robotics by externalizing robot-specific factors. This method first translates action commands into a "nominal trajectory" via the robot's controller and kinematics, which is then rendered as explicit robot geometry, including URDF mesh RGB and end-effector depth. This rendered geometry, combined with camera-aware static RGB/depth context, forms a consistent visual interface for the world model, allowing it to focus solely on scene response to robot motion. Evaluated on 41,642 DROID clips and 9,380 RoboCasa-GR1 clips, the rendered interface quantitatively outperforms vector-conditioned baselines like Ctrl-World and AdaLN state vector in PSNR, SSIM, and LPIPS. The approach also demonstrates generalization to unseen robot embodiments and facilitates generating robot manipulation videos from retargeted human demonstrations using datasets like DexYCB.

Key takeaway

For robotics engineers developing video world models, consider adopting a robot-factored visual interface. This approach, which renders nominal robot trajectories as explicit geometry and depth, significantly improves model performance and generalization across embodiments. You should integrate URDF-based rendering and end-effector depth to provide direct pixel-space action evidence, enabling your models to better predict scene interactions and adapt to new robot configurations or human demonstrations.

Key insights

Factoring robot-specific action realization and rendering into a visual interface improves robot world model performance and generalization.

Principles

Method

Actions are rolled into a nominal trajectory via robot kinematics, then rendered as URDF mesh RGB and end-effector depth. These, with static scene RGB/depth, condition a latent video diffusion model to predict scene response.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Robotics Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.