Article
Yang Likun 4 Years-old concept comes to fruition: H-JEPA AntMaze success rate rises from 18%to 73%Beating AI news reports that AMI Labs founded by Yang Likun, together with multiple universities, has announced H-JEPA, a world model capable of hierarchically predicting the future and planning actions. The high level handles long-term goals, while the low level handles specific actions. The paper, code, and pretrained weights have all been made public.
H-JEPA extends the approach of JEPA. It first converts images into abstract environmental features, then combines them with actions to predict future states, without having to generate video frame by frame. However, long-term planning and immediate control require different information. For example, when a quadruped robot navigates a maze, route selection mainly depends on position and corridors, whereas taking steps requires body posture. A single-layer model uses the same features for both tasks, making farther-ahead planning increasingly difficult.
H-JEPA combines multiple JEPA into different planning levels. The high level predicts farther into the future, passes intermediate goals to lower levels, and progressively refines them into specific actions. Unlike HWM which previously shared the same features across all levels, H-JEPA allows each level to learn its own environmental features. In maze experiments, the high level retained positional information while gradually ignoring short-term details such as leg posture. Yang Likun proposed a similar concept as early as 2022 years ago, and the team has now provided an end-to-end trainable implementation.
In the Visual AntMaze simulated maze, the three-level H-JEPA achieved a success rate of 73.3%, compared with 18.0%for the single-layer LeWorldModel, while requiring less planning computation. However, more planning levels are not necessarily better. In the simulated object-pushing Push-T task, the three-level model actually underperformed the two-level model. The team also conducted offline tests using the real-robot video dataset DROID, and its predicted action trajectories were closer to expert demonstrations than those of the single-layer model.
Original article link https://m.theblockbeats.info/flash/371155