mlsanityData to 7 Oct 2026

HaM-World: Soft-Hamiltonian World Models with Selective Memory for Planning

arXiv 2605.05951 ↗First posted 7 May 2026cs.AIcited by 1

On the map: World models → Latent predictive world models

also in 1 more
Abstract (arXiv)

World models support model-based planning through learned latent dynamics, but imagined rollouts can become unstable as the planning horizon grows or the dynamics distribution shifts. We propose HaM-World, a structured world model that combines history-conditioned selective memory with a Soft-Hamiltonian latent dynamics prior. The latent state is decomposed into a canonical (q,p) subspace and a context subspace c. Mamba selective state-space memory summarizes past observations and actions and conditions the same latent transition used for prediction, reward and value estimation, imagined rollouts, and cross-entropy method planning. The (q,p) subspace follows an energy-derived Hamiltonian vector field augmented with learnable residual and control dynamics, while c represents semantic, dissipative, and other non-conservative factors. On six DeepMind Control Suite tasks, HaM-World ranks first on four tasks and second on two, achieving the highest average AUC on the four-task core suite (117.9, 9.5% above TD-MPC2). Within the short-to-medium horizons used by the planner, it reduces imagined-rollout error to 45% of a strong baseline and wins 11 of 12 rollout-MSE cells for horizons k in {3,5,7}. At longer open-loop horizons, its error grows faster and is overtaken between k=7 and k=20; we therefore do not claim uniform long-horizon stability. Under 12 out-of-distribution perturbations involving dynamics shifts, action delay, and observation masking, it achieves the highest absolute return in every condition, with average gains of 10.2% on Finger Spin and 13.6% on Reacher Easy. Ablations show that memory accounts for the larger share of the observed gains, while Soft-Hamiltonian geometry provides smaller but consistent complementary improvements.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 50 of its 57 references that are in the map. First posted on arXiv 7 May 2026.

Builds on

50 of 57 references.

Closest ancestors · 4

Same direction, newest first

  1. 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb
  2. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan
  3. 2022Temporal Difference Learning for Model Predictive Control9 Mar
  4. 2018Learning Latent Dynamics for Planning from Pixels12 Nov

Other references · 46

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,635
  3. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  4. 2018Spectral Normalization for Generative Adversarial Networkscited by 1,618
  5. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
  6. 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,327
  7. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,283
  8. 2020Mastering Atari with Discrete World Modelscited by 990
  9. 2018DeepMind Control Suitecited by 935
  10. 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
  11. 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
  12. 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 738
34 more
  1. 2024Genie: Generative Interactive Environmentscited by 706
  2. 2019Hamiltonian Neural Networkscited by 702
  3. 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 482
  4. 2023Learning Interactive Real-World Simulatorscited by 440
  5. 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 424
  6. 2020Planning to Explore via Self-Supervised World Modelscited by 365
  7. 2022Transformers are Sample-Efficient World Modelscited by 317
  8. 2024Diffusion for World Modeling: Visual Details Matter in Ataricited by 285
  9. 2019Symplectic ODE-Net: Learning Hamiltonian Dynamics with Controlcited by 183
  10. 2024Understanding World or Predicting Future? A Comprehensive Survey of World Modelscited by 161
  11. 2024iVideoGPT: Interactive VideoGPTs are Scalable World Modelscited by 136
  12. 2023Structured State Space Models for In-Context Reinforcement Learningcited by 109
  13. 2023STORM: Efficient Stochastic Transformer based World Models for Reinforcement Learningcited by 88
  14. 2021DreamerPro: Reconstruction-Free Model-Based Reinforcement Learning with Prototypical Representationscited by 73
  15. 2020Dreaming: Model-based Reinforcement Learning by Latent Imagination without Reconstructioncited by 66
  16. 2022Denoised MDPs: Learning World Models Better Than the World Itselfcited by 66
  17. 2025RoboScape: Physics-informed Embodied World Modelcited by 38
  18. 2024GenRL: Multimodal-foundation world models for generalization in embodied agentscited by 29
  19. 2023Learning World Models with Identifiable Factorizationcited by 23
  20. 2024Connecting Joint-Embedding Predictive Architecture with Contrastive Self-supervised Learningcited by 19
  21. 2024PWM: Policy Learning with Multi-Task World Modelscited by 17
  22. 2025TD-JEPA: Latent-predictive Representations for Zero-Shot Reinforcement Learningcited by 17
  23. 2024Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RLcited by 15
  24. 2025Stable Port-Hamiltonian Neural Networkscited by 13
  25. 2025Bootstrapped Model Predictive Controlcited by 13
  26. 2025DMWM: Dual-Mind World Model with Long-Term Imaginationcited by 10
  27. 2025Accelerating Model-Based Reinforcement Learning with State-Space World Modelscited by 10
  28. 2024Operator World Models for Reinforcement Learningcited by 7
  29. 2025SOMBRL: Scalable and Optimistic Model-Based RLcited by 7
  30. 2024Policy-shaped prediction: avoiding distractions in model-based reinforcement learningcited by 6
  31. 2025Disentangled World Models: Learning to Transfer Semantic Knowledge from Distracting Videos for Reinforcement Learningcited by 6
  32. 2025Dynamics-Aligned Latent Imagination in Contextual World Models for Zero-Shot Generalizationcited by 5
  33. 2025EDELINE: Enhancing Memory in Diffusion-based World Models via Linear-Time Sequence Modelingcited by 3
  34. 2025DyMoDreamer: World Modeling with Dynamic Modulationcited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Beyond a single latent space: a dual-latent world model for long-horizon planning
  2. 2026SAGE: Subgoal-Conditioned Action Generation for Latent World Model Planning
  3. 2026Hierarchical Planning with Latent World Models
  4. 2026How Long, Not How Close: A Learned Temporal Metric for Planning in Latent World Models
  5. 2026Beyond the Next Step: Variable-Length Latent World Models for Long-Horizon Planning
  6. 2026The Planning Limits of Latent World Models
  7. 2026Latent Geometry Beyond Search: Amortizing Planning in World Models
  8. 2020Evolutionary Planning in Latent Space