HaM-World: Soft-Hamiltonian World Models with Selective Memory for Planning
TLDR
HaM-World combines Hamiltonian dynamics with selective memory (Mamba) for stable long-horizon planning in model-based RL, achieving strong results on DMC tasks and OOD perturbations.
Reasoning
Strengths include novel integration of Hamiltonian mechanics and state-space memory to address instability in world model rollouts, with thorough empirical validation on multiple tasks and OOD conditions. Weaknesses are limited to simulated control tasks and lack of analysis on more complex or high-dimensional environments.
Read-first score
Read-first score 56.1, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 51.
Field roles
Rank sensitivity
Stability: volatile; rank range: 403.
Keyword Scores
Deep Analysis
Innovations
- Decomposition of latent state into canonical (q, p) subspace and context subspace c
- Use of Mamba selective state-space memory as history-conditioned input to latent dynamics
- Hamiltonian vector field with learnable residual/control dynamics for (q, p) evolution
- Soft-Hamiltonian dynamics design that bounds action-free Hamiltonian-energy drift
- Shared latent state for dynamics prediction, reward/value estimation, imagined rollouts, and CEM action search
Methodology
HaM-World decomposes the latent state into a canonical (q, p) subspace and a context subspace c, using Mamba selective state-space memory as history-conditioned input. The (q, p) evolves via an energy-derived Hamiltonian vector field plus learnable residual/control dynamics, while c captures semantic, dissipative, and non-conservative factors. The model is evaluated on four DeepMind Control Suite tasks, with baselines including a strong baseline model, using metrics such as Avg. AUC, long-horizon rollout error, MSE cells, and OOD return.
Key Results
HaM-World achieves the highest Avg. AUC (117.9, +9.5%), reduces long-horizon rollout error to 45% of a strong baseline, and wins 11/12 k in {3,5,7} MSE cells. Under 12 OOD perturbations, it achieves the highest return in every condition, with average OOD-return gains of 10.2% on Finger Spin and 13.6% on Reacher Easy.