When Low Prediction Error Misleads Planning: Diagnosing Representation, Dynamics, and Decision Failures in Latent World Models
cs.LG
Submitted: 2026-10-04
Updated: 2026-10-04
Comments: 40 pages, 7 figures
License: http://creativecommons.org/publicdomain/zero/1.0/
The gist: The component that dominates a latent world model's prediction error need not be the one whose repair most improves action selection.
Terminology
Abstract
The component that dominates a latent world model's prediction error need not be the one whose repair most improves action selection. We show this by comparing action sequences from identical physical starts and separating endpoint error into a candidate-pool center and action-relative responses. Across four model families and four tasks, a confirmation pool of 256 new starts per task and 300 shared candidates per start shows that center error dominates MSE in 14/16 model-task cells. Yet in six of these cells, an oracle that corrects only the action-relative responses yields better physical rank correlation and top-30 elite quality than one that corrects only the center, while leaving more latent MSE (family-wise corrected intervals). The preference differs across the evaluated settings: a separate LeWorldModel (LeWM) study that executes oracle-selected actions favors center repair on PushT and on Reacher with a render-matched goal. Matched-candidate tests localize ordering loss: for LeWM, encoding realized endpoints raises physical Spearman from 0.464 to 0.975 on that Reacher setting and from 0.193 to 0.631 on PushT (64 starts per task), while Cube's encoded-goal cost remains uninformative. A 72-run objective study improves selected response diagnostics, while incremental closed-loop planning gains remain unconfirmed. These results separate error magnitude from the decision effects of oracle correction and motivate evaluating representation, prediction, and planning as separate stages.
Sources
- Temporal-Distance JEPA: Plan-Aware Representation Learning for Latent World Model Predictive Control
- Revisiting Feature Prediction for Learning Visual Representations from Video
- Latent World Models with Monotone Planning Costs for Image-Goal Navigation
- ATM: Why Latent World Models Can Fail to Plan
- Value-guided action planning with JEPA world models
- Towards Unraveling and Improving Generalization in World Models
- ActSWM: Action-Sensitive World Models for Long-Horizon Planning in Open-World Games
- World Models
- Flow-JEPA: Robust Latent Dynamics for JEPA World Models via Flow Matching
- Koopman Dreamer: Spectrally Constrained Latent Dynamics for Stable World-Model Imagination
- Dueling World Models: Advantage-Style Action Channels for Common-Mode Distractor Rejection
- Discriminative World Models for Web Agents
- Action from Adjacent Set in Physical Space Outperforms the Best Prediction in World Models
- World Model Control by Trajectory Reachability Metrics
- Semigroup-JEPA: Latent Dynamics Consistency for Zero-Shot Physics Generalization
- LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels
- Latent Energy Action Planning with World Models
- Overcoming Statistical Bias in Action-Controllable World Models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks