PILOT: Guiding Long-Horizon World-Model Planning for Robot Manipulation with Visual Subgoals and Policy Priors
Organizations: The authors are with the Department of Computer Science and Engineering, University of Minnesota, Minneapolis, MN 55455, USA.
Abstract
Latent world-model planning has shown promising results in goal-conditioned robotic manipulation, but remains challenging for long-horizon planning due to limited temporal guidance and inefficient action search. A distant visual goal provides little guidance for preparatory actions, while uninformed action sampling makes local optimization inefficient. We propose PILOT (Policy-Initialized Local Optimization with Temporal subgoals), which uses a video model to generate intermediate visual subgoals and a learned policy to initialize the cross-entropy method (CEM). CEM refines candidate action sequences toward each subgoal using an action-conditioned world model in a receding-horizon control loop. In our primary real-robot evaluation across four UR5e manipulation tasks, PILOT achieves 68.8% success, compared with 55.0% without policy initialization, 13.8% for policy-only execution, and 0.0% for direct final-goal planning. Simulation results show that these gains persist over longer horizons: PILOT exceeds direct execution of the same policy by 30 percentage points on long Push-T tasks and by 22 percentage points on LIBERO-10. Policy initialization also accelerates local optimization: on Push-T, CEM matches the cost reached by generic initialization after 10 iterations in only 4 iterations and reduces the final planning cost by 24.1% under the same sampling budget. These results show that combining temporal subgoals with policy-guided action search improves both long-horizon task execution and local planning efficiency. Videos are available on the webpage: https://pilot-planner.github.io/.
Figures & tables
| Method | Cup | Pencil Case | Bowl-Box | Bowl-Bowl | Avg. (95% CI) |
|---|---|---|---|---|---|
| Flat WM | 0/20 | 0/20 | 0/20 | 0/20 | 0.0% (0.0–4.6) |
| Policy only | 4/20 | 3/20 | 2/20 | 2/20 | 13.8% (7.9–23.0) |
| PILOT w/o prior | 8/20 | 5/20 | 14/20 | 17/20 | 55.0% (44.1–65.4) |
| PILOT | 11/20 | 13/20 | 13/20 | 18/20 | 68.8% (57.9–77.8) |
| Method | Short | Medium | Long |
|---|---|---|---|
| Flat WM | 33.3% | 0.0% | 0.0% |
| Flat WM + prior | 46.7% | 0.0% | 0.0% |
| Policy only (5 demos) | 0.0% | 0.0% | 0.0% |
| Policy only (10 demos) | 13.3% | 3.3% | 6.7% |
| Policy only (20 demos) | 26.7% | 16.7% | 10.0% |
| Policy only (30 demos) | 40.0% | 33.3% | 23.3% |
Long-Horizon Robotic Manipulation
Long-horizon robotic manipulation studies how robots plan, remember, and recover across extended sequences of physical tasks.
See all 431 papers