Beyond a single latent space: a dual-latent world model for long-horizon planning
cs.AI
Submitted: 2026-09-29
Updated: 2026-09-29
Comments: 31 pages, 22 figures, 9 tables. Main text: 9 pages
Code: https://github.com/DeLin1001/Dual-WM-Official
License: http://creativecommons.org/licenses/by/4.0/
The gist: Latent world models often struggle with long-horizon planning despite accurate short-term predictions.
Terminology
Abstract
Latent world models often struggle with long-horizon planning despite accurate short-term predictions. Recursive rollouts accumulate errors, while distance concentration in high-dimensional latent spaces can weaken goal discrimination. We introduce the Dual-Latent World Model (Dual-WM), which separates local execution and long-range planning through distinct state representations and dynamics models. The low-level model predicts action-conditioned transitions, while the high-level model uses learned macro-actions to plan over longer temporal spans. We also propose Long-Horizon Representation Learning with Weighted Rollout (LoRe), which supervises self-generated predictions at both levels. An analysis of recursive error propagation motivates exponential horizon weights with separate decay rates for the two temporal scales. During planning, the high-level model generates latent subgoals that the low-level model refines into actions for precise execution. We evaluate from-scratch Dual-WM on five goal-conditioned visual control tasks against the task-wise strongest baselines without actor-guided proposals. At goal offsets of 50 and 100 environment steps, mean success increases from 75.9% to 84.4% and from 61.4% to 69.5%, respectively. At offset 100, Dual-WM outperforms these baselines on all five tasks and improves mean success over LeWM by 30.8 percentage points. Ablations and supporting analyses provide evidence of more informative representations for goal evaluation and greater consistency under recursive prediction. These results highlight the value of separating temporal roles and training across multiple horizons for reliable latent planning. Our core implementation is available at https://github.com/DeLin1001/Dual-WM-Official.
Sources
- Diagnosing JEPA World Models with Action-Conditioned Predictive Consistency
- The SIGReg Objective as Variational Free Energy: A Theoretical Active-Inference Account of JEPA World Models
- Temporal-Distance JEPA: Plan-Aware Representation Learning for Latent World Model Predictive Control
- Latent Goal Prediction from Language for Model-Based Planning
- Mind the Gap: Promises and Pitfalls of Hierarchical Planning in LeWorldModel
- Mitigating Compounding Error via Video Representation Regularization
- SAGE: Subgoal-Conditioned Action Generation for Latent World Model Planning
- A Generalization Theory for JEPA-Based World Models
- Beyond the Next Step: Variable-Length Latent World Models for Long-Horizon Planning
- IMWM: Intuition Models Complement World Models for Latent Planning
- Fast LeWorldModel
- Mastering Diverse Domains through World Models
- Reperesentation Geometry Matters for Planning with JEPA World Models
- Flow-JEPA: Robust Latent Dynamics for JEPA World Models via Flow Matching
- When Does LeJEPA Learn a World Model?
- Predictive but Not Plannable: RC-aux for Latent World Models
- ODEWorld: A Continuous Predictive Architecture via Physical-Time Flow
- ProWorld: Progress-Aware Hyperbolic World Models for Long-Horizon Visual Goal Reaching
- LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems