mlsanityData to 14 Oct 2026

Hamiltonian JEPA: Action-Conditioned World Models with an Inherited Control State

arXiv 2609.33497 ↗First posted 27 Sep 2026cs.LGcited by 0

On the map: World models → Latent predictive world models

Abstract (arXiv)

Planning from pixels needs more than a latent space that is stable and predictable. The state the planner scores must also be organized by how actions move the system. Joint-embedding predictive architectures (JEPAs) avoid pixel reconstruction by predicting future representations, but existing action-conditioned JEPAs ask one embedding to serve both perception and control. We introduce H-JEPA, which separates the two. A wide perceptual code is regularized toward a well-scaled isotropic geometry with a Bures-Wasserstein prior, and a fixed orthonormal slice of that code is the control state, which inherits the code's covariance without any objective of its own. The state evolves under phase-conditioned dissipative port-Hamiltonian dynamics whose input port has orthonormal columns. Port-inverse consistency (PIC) reads the executed action back through the transpose of that port. We show that this readout is exactly the rollout error projected onto the port directions, so PIC is a parameter-free reweighting of prediction error and not an auxiliary action decoder. Untying the readout from the port breaks this identity and loses half of the gain. H-JEPA matches or exceeds reconstruction-free baselines, including the action-decoding Delta-JEPA, on four pixel-based control benchmarks after at most 10 training epochs, and its largest gain is on OGB-Cube (91.9 against 79.3 percent). Ablations on PushT and OGB-Cube separate the contributions of the structured predictor, PIC, the prediction horizon, the state rank, and the anti-collapse prior.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 9 of its 48 references that are in the map. First posted on arXiv 27 Sep 2026.

Builds on

9 of 48 references.

Closest ancestors · 4

Same direction, newest first

  1. 2026Delta-JEPA: Learning Action-Sensitive World Models via Latent Difference Decoding30 Jun
  2. 2026Sensorimotor World Models: Perception for Action via Inverse Dynamics18 Jun
  3. 2026Sub-JEPA: Subspace Gaussian Regularization for Stable End-to-End World Models10 May
  4. 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels13 Mar

Other references · 5

  1. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  2. 2018DeepMind Control Suitecited by 956
  3. 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 254
  4. 2015All you need is a good initcited by 124
  5. 2026Rethinking Weight Tying: Pseudo-Inverse Tying for LM Stable Training and Updatescited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Diagnosing JEPA World Models with Action-Conditioned Predictive Consistency
  2. 2026Toward Physically Grounded JEPA World Models for Goal-Conditioned Robotic Planning
  3. 2026LRC-JEPA: Disentangling Dynamics and Residual Context for Efficient World Models
  4. 2026Qantara: Bridge-Flow Training for Multi-Paradigm JEPA Control
  5. 2026No Gaussian Required: Contrastive Inverse Dynamics for JEPA World Models
  6. 2026UWM-JEPA: Predictive World Models That Imagine in Belief Space
  7. 2026LeWAM: End-to-End World and Action Modeling with JEPAs
  8. 2026JEPA-x: Cross-Predictive Physics Grounding for Forecastable Latent Dynamics