cs.AIAug 23, 2026

Predicting Consequences and Reinforcing Navigation Policies with Latent World Models

Authors: Zengmao Wang, Wei Gao, Shuhan Shen

Organizations: School of Advanced Interdisciplinary Sciences, University of Chinese Academy of Sciences · Institute of Automation, Chinese Academy of Sciences · School of Artificial Intelligence, University of Chinese Academy of Sciences

Abstract

World models enable agents to reason about future outcomes and learn policies from their knowledge of state transition, but existing approaches primarily focus on reconstructing future observations or features, which introduces unnecessary complexity and limits their effectiveness for decision making. In this work, we propose a compatibility prediction Latent World Model (LWM) for robot navigation that predicts action-conditioned latent feature compatibility rather than reconstructing observations. Our key insight is that spatial proximity correlates with latent feature similarity, enabling action consequences to be evaluated directly in latent space. To support counterfactual training, our model leverages action sequences sampled across trajectories and learns to predict which sequences lead closer to the goal. Furthermore, we demonstrate how the learned world model can supervise policy learning from unlabeled video data and further improve policies through reinforcement learning entirely within the world model. This imagination-driven framework eliminates the need for action annotations and additional environment interaction. Extensive experiments on multiple real-world robot navigation datasets show that our approach significantly outperforms prior world model and imitation learning methods in prediction accuracy, policy learning, and real-world navigation performance. The code, pretrained models, and additional materials are available at https://wzm206.github.io/latent-world-model-nav.

Figures & tables

Appendix figures & tables4 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. LiteNWM: Efficient Latent World Models for Onboard Visual Navigation in the Wild

    Oct 8, 2026Linkai Liu, Yuntian Zhang, Zhenshan Bing +5Robot NavigationLatent World Models

  2. World-Action Models for Robot Learning and Control: A Survey

    Sep 13, 2026Zuxing Lu, Hongjia Zhai, Guanzhi Wang +13World Action ModelsAction-Conditioned World Models

  3. World Model for Robot Learning: A Comprehensive Survey

    Apr 30, 2026Bohan Hou, Gen Li, Jindou Jia +15World Model LearningWorld Models for Robotics