Prismatic World Model: Learning Compositional Dynamics for Planning in Hybrid Systems
TLDR
PRISM-WM decomposes hybrid dynamics into composable primitives using MoE, reducing rollout drift for planning in continuous control.
Reasoning
The paper addresses a key challenge in model-based planning for hybrid systems with a structured MoE architecture and latent orthogonalization. Strengths include a principled approach to mode decomposition and empirical gains on simulated benchmarks. Weaknesses are the lack of real-world experiments and reliance on simulated environments only.
Read-first score
Read-first score 54.9, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 45.
Field roles
Rank sensitivity
Stability: volatile; rank range: 388.
Keyword Scores
Deep Analysis
Innovations
- Decomposition of complex hybrid dynamics into composable primitives using a structured architecture
- Context-aware Mixture-of-Experts (MoE) framework with a gating mechanism for implicit physical mode identification
- Latent orthogonalization objective to ensure expert diversity and prevent mode collapse
Methodology
PRISM-WM employs a context-aware Mixture-of-Experts (MoE) framework where a gating mechanism implicitly identifies the current physical mode, and specialized experts predict the associated transition dynamics. A latent orthogonalization objective is introduced to maintain expert diversity and prevent mode collapse. The model is trained on continuous control benchmarks and evaluated as a substrate for trajectory optimization algorithms like TD-MPC.
Key Results
Experiments on high-dimensional humanoids and multi-task settings demonstrate that PRISM-WM reduces rollout drift and provides a high-fidelity substrate for trajectory optimization, indicating its potential as a foundational model for model-based agents.
Limitations
- Reliance on implicit mode identification may fail in highly ambiguous or unseen hybrid dynamics
- Performance on real-world robotic systems has not yet been demonstrated, as evaluation is limited to simulation benchmarks