DepthWorld: 3D World Model for Robot Manipulation
cs.RO, cs.AI, cs.CV
Submitted: 2026-10-06
Updated: 2026-10-06
Comments: Accepted at the Conference on Robot Learning (CoRL) 2026. Project page: https://www.jaibardhan.com/depthworld. 32 pages including supplementary material, 15 figures, 7 tables
Project page: https://www.jaibardhan.com/depthworld
License: http://creativecommons.org/licenses/by/4.0/
The gist: World models offer a data-driven alternative to traditional simulators for robotics, with applications spanning policy evaluation, improvement, and planning.
Terminology
Abstract
World models offer a data-driven alternative to traditional simulators for robotics, with applications spanning policy evaluation, improvement, and planning. All of these uses depend on faithful 3D geometry, yet current video-based world models are trained on RGB alone and produce rollouts that look correct frame-by-frame but do not compose into a consistent 3D world. Closing this gap requires progress on two fronts: large-scale 3D supervision for manipulation, and an architecture that can absorb it without disturbing strong pretrained video priors. We introduce a calibration pipeline that combines learned stereo depth with a joint factor graph, pooling all episodes collected from the same physical robot to recover its shared kinematic parameters alongside per-scene extrinsics. Applied to the DROID dataset, this yields DROID-3D, a calibrated 3D dataset providing dense metric depth and recalibrated multi-view extrinsics (achieving <0.7 px reprojection error on 90% of episodes for external cameras). We then train DepthWorld, a Stable Video Diffusion-based world model that jointly predicts multi-view RGB and depth via spatial latent tiling, leaving the pretrained Variational Autoencoder (VAE) unchanged. Depth supervision improves RGB prediction itself by +1.48 dB PSNR over an identical RGB-only baseline at equal training budget, while simultaneously yielding accurate metric depth for downstream geometric reasoning.
Sources
- Ctrl-World: A Controllable Generative World Model for Robot Manipulation
- Persistent Robot World Models: Stabilizing Multi-Step Rollouts via Reinforcement Learning
- TesserAct: Learning 4D Embodied World Models
- GAIA-2: A Controllable Multi-View Generative World Model for Autonomous Driving
- $\pi^3$: Permutation-Equivariant Visual Geometry Learning
- Depth Anything 3: Recovering the Visual Space from Any Views
- MapAnything: Universal Feed-Forward Metric 3D Reconstruction
- ARKitScenes: A Diverse Real-World Dataset For 3D Indoor Scene Understanding Using Mobile RGB-D Data
- Habitat-Matterport 3D Dataset (HM3D): 1000 Large-scale 3D Environments for Embodied AI
- DROID: A Large-Scale In-The-Wild Robot Manipulation Dataset
- RoboWM-Bench: A Benchmark for Evaluating World Models in Robotic Manipulation
- Stereo Magnification: Learning View Synthesis using Multiplane Images
- Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasets
- RoDyn: Taming Interactive Robot-Dynamic 2.5D World Model for Robotic Manipulation
- RT-1: Robotics Transformer for Real-World Control at Scale
- RH20T: A Comprehensive Robotic Dataset for Learning Diverse Skills in One-Shot
- RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robots
- PointWorld: Scaling 3D World Models for In-The-Wild Robotic Manipulation
- RoMa v2: Harder Better Faster Denser Feature Matching
- LoMa: Local Feature Matching Revisited
Related papers
- FMT x: An Efficient and Asymptotically Optimal Extension of the Fast Marching Tree for Dynamic Replanning
- MPCFormer: A physics-informed data-driven approach for explainable socially-aware autonomous driving
- RoboLab: A High-Fidelity Simulation Benchmark for Analysis of Task Generalist Policies
- ARC: A Reasoning Recipe for Robot Foundation Models
- HRDexDB: A 4D Dexterous Grasping Dataset Across Human and Multiple Robot Embodiments
- APT: Action Expert Pretraining Improves Instruction Generalization of Vision-Language-Action Policies