Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GEM-4D: Geometry-Enhanced Video World Models for Robot Manipulation

About

Video world models can generate realistic futures from a single instruction, but they often fail to track the same physical points consistently across time. As a result, the generated videos appear plausible, yet lack the physical grounding required for reliable action execution, such as robot manipulation. We present GEM-4D, a geometry-grounded video world model that resolves this limitation by injecting dense 4D correspondence supervision distilled from a pretrained geometry foundation model into the video generative backbone during training. This supervision enables the model to jointly capture appearance and geometric structure while retaining a single-stream architecture with no additional inference cost. We further introduce an inverse dynamics module that converts correspondence-consistent video rollouts into executable robot trajectories, enabling direct deployment in both real-world and simulated manipulation. GEM-4D achieves state-of-the-art performance on both video prediction and geometric consistency across both simulation and realistic scenarios and improves real-world manipulation success from 61% to 81%. Additional results are available at https://gem-4d.github.io/.

Kaichen Zhou, Yuzhen Chen, Fangneng Zhan, Hang Hua, Grace Chen, Xinhai Chang, Ao Qu, Yilun Du, Zhuang Liu, Paul Pu Liang, Mengyu Wang• 2026

Related benchmarks

TaskDatasetResultRank
4D scene generationDroid Realistic (test)
FVD31.82
5
4D scene generationRLBench Simulation (test)
FVD (Fréchet Video Distance)27.94
5
Embodied Action PlanningDroid Real-world Tasks
AUTOLab Score75
3
Embodied Action PlanningRLBench Simulated Tasks
Lift Numbered Block Success Rate78
2
Showing 4 of 4 rows

Other info

Follow for update