SAGE: Subgoal-Conditioned Action Generation for Latent World Model Planning
TLDR
SAGE improves long-horizon planning in latent world models by using subgoal-conditioned action generation, achieving higher success rates on PushT and OGBench Cube.
Reasoning
The paper introduces a novel prior-conditioned planner that decomposes goals into subgoals to guide action generation, effectively addressing the proposal quality issue in long-horizon planning. However, the evaluation is limited to two simulated benchmarks (PushT and OGBench Cube), lacking real-world or diverse task validation, and the method's generalizability remains unclear.
Read-first score
Read-first score 38.1, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 39.
Field roles
Rank sensitivity
Stability: volatile; rank range: 213.
Keyword Scores
Deep Analysis
Innovations
- Prior-conditioned planner that replaces random proposal initialization with structured guidance via a goal-conditioned generator predicting the next reachable latent subgoal for a specified duration.
- Use of subgoals at varying durations as priors to capture semantic information across temporal scales, balancing fine-grained local control with long-horizon progress.
- Conditioning action sequence generation on predicted subgoals before evaluation and refinement by a frozen world model.
Methodology
SAGE is a prior-conditioned planner for latent world models. At each planning stage, a goal-conditioned generator predicts reachable latent subgoals of varying durations; these subgoals condition the generation of candidate action sequences. A frozen world model then evaluates and refines these proposals before execution.
Key Results
On PushT and OGBench Cube with target offset 150, SAGE improves long-horizon success rates from 12.7% to 64.7% and from 26.7% to 67.3%, respectively, while preserving strong short-horizon performance.