Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces H-JEPA, a hierarchical extension of Joint Embedding Predictive Architectures (JEPAs) that learns multiple action-conditioned latent world models operating at different temporal scales. By enabling top-down planning where higher levels set subgoals for lower levels, H-JEPA effectively decomposes complex, long-horizon tasks into manageable temporal abstractions. This hierarchical approach significantly improves planning performance across simulated navigation and manipulation tasks, achieving a 73% success rate on Visual AntMaze compared to 18% for a flat JEPA, while also reducing planner compute.
Hierarchical JEPAs unlock efficient long-horizon visual planning by learning distinct latent spaces for different temporal scales, dramatically boosting success rates and reducing compute.
Long-horizon planning with latent world models requires reasoning across timescales and levels of abstraction. Existing task-agnostic JEPA world models predict and plan at a single timescale or with multiple horizons in one shared latent space. We introduce H-JEPA, an end-to-end recipe for training a hierarchy of action-conditioned JEPAs in which each level predicts farther ahead in its own learned latent space. Planning proceeds top-down: the top level optimizes progress toward the goal, and each level's predictions become subgoals for the planner below it. When factors in the data evolve at separated timescales, higher levels discard fast, unpredictable detail and retain slower task-relevant state. Across four simulated navigation and manipulation environments, hierarchical planning improves over a flat JEPA; on Visual AntMaze, a three-level hierarchy raises success from 18% to 73% using less planner compute. Ablations attribute these gains to both temporal decomposition and higher-level goal representations. With inverse-dynamics supervision, the approach extends to diverse real-robot videos from DROID, where hierarchy improves offline planning fidelity at lower planner compute.
Meta AI