Representation World Model: Learning States, Transition and Executable Plans in Representation
TLDR
Proposes Representation World Model that learns states, transitions, and executable plans directly in representation space, enabling planning via latent path construction without rollouts or search.
Reasoning
The paper presents a novel approach that integrates planning into representation geometry, avoiding recursive rollouts and action-space search, which is a strong conceptual contribution. However, the abstract provides limited details on experimental settings, baselines, and limitations, making full assessment difficult.
Read-first score
Read-first score 33, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 34.
Field roles
Rank sensitivity
Stability: volatile; rank range: 252.
Keyword Scores
Deep Analysis
Innovations
- Introduces the Representation World Model (RWM) that learns states, transitions, and executable plans directly in representation space.
- Incorporates planning directly into the learned representation geometry instead of using explicit dynamics models with search, optimization, or policy-based prediction.
- Uses inverse-dynamics supervision locally along latent paths constructed from endpoint representations to learn geometry that preserves task-relevant state and transition information.
- Enables inference-time planning by directly constructing a latent path between current and goal representations and recovering actions via inverse dynamics, without recursive rollouts or action-space search.
Methodology
RWM learns representation geometry by applying inverse-dynamics supervision locally along latent paths constructed from endpoint representations, requiring these paths to preserve task-relevant state and transition information. At inference, planning directly constructs a latent path between current and goal representations, and inverse dynamics recovers the corresponding actions without recursive rollouts or action-space search. The approach is evaluated on continuous-control benchmarks and robotic manipulation tasks.
Key Results
Experiments on continuous-control benchmarks demonstrate RWM's effectiveness for direct planning, and robotic manipulation results show potential for extension to more complex embodied control tasks. The authors conclude that planning directly in representation space is a promising alternative to conventional world-model planning.