Depth-Regularized JEPA World Models Learn More Transferable Representations from Real Outdoor Robot Data
On the map: World models → Latent predictive world models
Abstract (arXiv)
World models, especially based on JEPA architectures, have been shown to learn robust dynamics of various environments. However, learning from visually complex real-world data remains a challenge, especially in unpredictable outdoor environments. We introduce depth as a geometric prior during training in learning more robust latent dynamics directly from robot video data and handling visual complexity. This combines depth supervision with an isotropy-inducing latent regularizer (SIGReg), maximizing task-agnostic latent diversity while constraining how that diversity is organized, with the combined objective targeting the highest-entropy representation consistent with scene geometry. To satisfy this greater complexity without increasing inference time, we also add training-only overparameterization. Training an 18M-parameter model on video from a real agricultural robot, we evaluate with frozen-representation visual odometry probes, predictor-based surprise detection, and multi-step latent rollout fidelity. Compared to the baseline LeWM, our method lowers visual odometry probe error by 33%, substantially increases surprise-score separation both in-domain and on the out-of-domain TartanGround benchmark, and improves multi-step rollout fidelity under domain shift, with gains that grow with rollout horizon. Notably, we also see improvements in surprise-score separation on physics understanding that is not directly tied to 3D geometry, such as lighting and shadows. These results show that a lightweight training-time geometric prior makes a compact JEPA world model more useful and more transferable on real outdoor data with strong underlying representations, without adding inference overhead. Our work suggests that depth as a physically grounded prior can enhance world model generalization on a variety of tasks.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 20 of its 21 references that are in the map. First posted on arXiv 15 Jul 2026.
Builds on
20 of 21 references.
Closest ancestors · 6
Same direction, newest first
- 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels13 Mar
- 2025seq-JEPA: Autoregressive Predictive Learning of Invariant-Equivariant World Models6 May
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planning7 Nov
- 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan
- 2018Learning Latent Dynamics for Planning from Pixels12 Nov
Other references · 14
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2018World Modelscited by 1,513
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 809
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 522
- 2021Understanding self-supervised Learning Dynamics without Contrastive Pairscited by 256
- 2024Enhancing End-to-End Autonomous Driving with Latent World Modelcited by 166
- 2020MDP Homomorphic Networks: Group Symmetries in Reinforcement Learningcited by 142
- 2022MobileOne: An Improved One millisecond Mobile Backbonecited by 109
- 2025World4Drive: End-to-End Autonomous Driving via Intention-aware Physical Latent World Modelcited by 95
- 2025Learning from Reward-Free Offline Data: A Case for Planning with Latent Dynamics Modelscited by 89
2 more
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Invariance Co-training for Robot Visual Generalization
- 2026Robot-Factored World Models via Robot Rendering
- 2026DepthWorld: 3D World Model for Robot Manipulation
- 2026Temporally Centered SIGReg Improves LeWorldModel Representations for Robot Policy Learning
- 2026Is Forward Prediction Enough? Physical State Grounding for JEPA World Models
- 2026J-LAW: Joint Localization and Action-Conditioned World Modeling via Coupled Latent Factor Graphs
- 2026Point Tracking Improves World Action Models
- 2026Frozen in a Frame: The Velocity Blind Spot in JEPA World Models