ProWorld: Progress-Aware Hyperbolic World Models for Long-Horizon Visual Goal Reaching
On the map: World models → Latent predictive world models
Abstract (arXiv)
JEPA-style visual world models offer an effective paradigm for visual goal planning by predicting future latent representations. Existing methods typically learn local transition consistency through next-step representation prediction. However, in long-horizon tasks, accurate local prediction alone need not ensure sustained progress toward the goal. First, multi-step rollouts can remain locally plausible while drifting away from goal-relevant trajectories. Second, locally similar future states can correspond to substantially different long-term progress, making them difficult to distinguish in a latent space optimized mainly for local consistency. To address these challenges, we introduce goal-conditioned progress order, a relative ordering of states according to how they advance toward a given goal. This order exhibits an asymmetric, coarse-to-fine structure: early states retain broader future possibilities, while later states concentrate on more specific goal-relevant regions. Such a structure is well suited to hyperbolic geometry. Motivated by this observation, we propose ProWorld, a progress-aware hyperbolic visual world model. ProWorld leverages goal-conditioned progress order to organize visual latent-space dynamics, maintains directional progress within trajectories via hyperbolic entailment learning, and mitigates progress ambiguity among locally similar future states via hyperbolic future discrimination. Furthermore, we design a progress-aware planning objective that scores candidate rollouts by jointly considering proximity to the goal and sustained progress across intermediate states. Experiments on four visual goal-reaching tasks demonstrate that ProWorld achieves an average absolute success-rate gain of 9.67 over LeWM. The code will be released after the paper is accepted.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 26 of its 37 references that are in the map. First posted on arXiv 3 Aug 2026.
Builds on
26 of 37 references.
Closest ancestors · 6
Same direction, newest first
- 2026Sub-JEPA: Subspace Gaussian Regularization for Stable End-to-End World Models10 May
- 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels13 Mar
- 2026Causal-JEPA: Learning World Models through Object-Level Latent Masking11 Feb
- 2026A Lightweight Library for Energy-Based Joint-Embedding Predictive Architectures3 Feb
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planning7 Nov
- 2018Learning Latent Dynamics for Planning from Pixels12 Nov
Other references · 20
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 785
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 773
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
- 2016Deep Visual Foresight for Planning Robot Motioncited by 285
- 2015Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Imagescited by 274
- 2018Learning Continuous Hierarchies in the Lorentz Model of Hyperbolic Geometrycited by 265
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 254
- 2024OGBench: Benchmarking Offline Goal-Conditioned RLcited by 238
- 2019Hyperbolic Image Embeddingscited by 221
- 2019Search on the Replay Buffer: Bridging Planning and Reinforcement Learningcited by 194
8 more
- 2022Hyperbolic Vision Transformers: Combining Improvements in Metric Learningcited by 111
- 2022Hyperbolic Image Segmentationcited by 102
- 2020Objective Mismatch in Model-based Reinforcement Learningcited by 96
- 2020Long-Horizon Visual Planning with Goal-Conditioned Hierarchical Predictorscited by 62
- 2020Goal-Aware Prediction: Learning to Model What Matterscited by 45
- 2025HMamba: Hyperbolic Mamba for Sequential Recommendationcited by 11
- 2025Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learningcited by 4
- 2022Bending the Future: Autoregressive Modeling of Temporal Knowledge Graphs in Curvature-Variable Hyperbolic Spacescited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Beyond a single latent space: a dual-latent world model for long-horizon planning29 Sep 2026
- 2026Memory Has Geometry: Non-Uniform Geometric Memory for Long-Horizon Personalized AI16 Sep 2026
- 2026Finite-Sample Metric Non-Collapse for Geometrically Supervised Latent World Models in Control7 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026GeoWorld: Geometric World Models
- 2026FIRM-WM: State-factorized factual-interventional recurrent modeling for reward-free visual planning
- 2026SkipWorld: Latent Skip Planning with Distilled World Models for Efficient Visual Navigation
- 2026World2VLM: Distilling World Model Imagination into VLMs for Dynamic Spatial Reasoning
- 2026VisualPatchWorld: Code World Models as Latent Structured Representations for Planning
- 2025StateSpaceDiffuser: Bringing Long Context to Diffusion World Models
- 2026FoMoVLA: Bridging Visual Foresight and Motion Guidance for Vision-Language-Action Models
- 2026WALT: Learning World-Model-Aligned Latent Trajectories for Autonomous Driving