LeWAM: A JEPA World Action Model with Diffusion-Steering-Based MPC
World action models (WAMs) predict actions and future observations, typically from a reconstruction-based representation that carries noisy, redundant information which can complicate downstream predictions. We introduce LeWAM, a bidirectional transformer for forward, backward, inverse dynamics and policy prediction, on a decoder-free JEPA latent trained end-to-end through all four modes. We see…
We haven't written up this one. arXiv cs.AI has the full story — the link below goes straight to it.
