Hierarchical Planning with Latent World Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhang, Wancong, Terver, Basile, Zholus, Artem, Chitnis, Soham, Sutaria, Harsh, Assran, Mido, Balestriero, Randall, Bar, Amir, Bardes, Adrien, LeCun, Yann, Ballas, Nicolas
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915913529294848
author Zhang, Wancong
Terver, Basile
Zholus, Artem
Chitnis, Soham
Sutaria, Harsh
Assran, Mido
Balestriero, Randall
Bar, Amir
Bardes, Adrien
LeCun, Yann
Ballas, Nicolas
author_facet Zhang, Wancong
Terver, Basile
Zholus, Artem
Chitnis, Soham
Sutaria, Harsh
Assran, Mido
Balestriero, Randall
Bar, Amir
Bardes, Adrien
LeCun, Yann
Ballas, Nicolas
contents Model predictive control (MPC) with learned world models has emerged as a promising paradigm for embodied control, particularly for its ability to generalize zero-shot when deployed in new environments. However, learned world models often struggle with long-horizon control due to the accumulation of prediction errors and the exponentially growing search space. In this work, we address these challenges by learning latent world models at multiple temporal scales and performing hierarchical planning across these scales, enabling long-horizon reasoning while substantially reducing inference-time planning complexity. Our approach serves as a modular planning abstraction that applies across diverse latent world-model architectures and domains. We demonstrate that this hierarchical approach enables zero-shot control on real-world non-greedy robotic tasks, achieving a 70% success rate on pick-&-place using only a final goal specification, compared to 0% for a single-level world model. In addition, across physics-based simulated environments including push manipulation and maze navigation, hierarchical planning achieves higher success while requiring up to 4x less planning-time compute.
format Preprint
id arxiv_https___arxiv_org_abs_2604_03208
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Hierarchical Planning with Latent World Models
Zhang, Wancong
Terver, Basile
Zholus, Artem
Chitnis, Soham
Sutaria, Harsh
Assran, Mido
Balestriero, Randall
Bar, Amir
Bardes, Adrien
LeCun, Yann
Ballas, Nicolas
Machine Learning
Model predictive control (MPC) with learned world models has emerged as a promising paradigm for embodied control, particularly for its ability to generalize zero-shot when deployed in new environments. However, learned world models often struggle with long-horizon control due to the accumulation of prediction errors and the exponentially growing search space. In this work, we address these challenges by learning latent world models at multiple temporal scales and performing hierarchical planning across these scales, enabling long-horizon reasoning while substantially reducing inference-time planning complexity. Our approach serves as a modular planning abstraction that applies across diverse latent world-model architectures and domains. We demonstrate that this hierarchical approach enables zero-shot control on real-world non-greedy robotic tasks, achieving a 70% success rate on pick-&-place using only a final goal specification, compared to 0% for a single-level world model. In addition, across physics-based simulated environments including push manipulation and maze navigation, hierarchical planning achieves higher success while requiring up to 4x less planning-time compute.
title Hierarchical Planning with Latent World Models
topic Machine Learning
url https://arxiv.org/abs/2604.03208