Search papers, labs, and topics across Lattice.
This paper introduces Qantara, an innovative Joint-Embedding Predictive Architecture (JEPA) that enables multi-paradigm control from a single model checkpoint by integrating a Brownian-bridge interpolant and noise-to-data flow matching. By allowing inference through latent planning, behavior-cloning action sampling, and inverse dynamics without retraining, Qantara significantly enhances flexibility in deployment scenarios. The approach achieves state-of-the-art performance on the LeWM control suite and OGBench-Cube, demonstrating a substantial improvement over existing methods, with a success rate of 91.2 SR and notable gains in performance metrics.
Qantara achieves a remarkable 91.2 SR on the LeWM control suite, redefining the capabilities of JEPA world models to operate across multiple inference paradigms without retraining.
Joint-Embedding Predictive Architectures (JEPAs) underpin a growing family of latent world models for control from raw pixels, but every existing JEPA world model commits at training time to a single inference paradigm: either trajectory optimisation in a learned dynamics model, or direct behaviour cloning. A single checkpoint that serves both would defer this choice to inference, when deployment constraints (rollout cost, observation accessibility) determine which path wins. We present Qantara, an end-to-end JEPA whose joint training objective pairs a Brownian-bridge interpolant between consecutive clean latents on the state axis with noise-to-data flow matching on the action axis. The same checkpoint serves three inference paradigms without retraining: latent planning, behaviour-cloning action sampling, and inverse dynamics, which we query through a video-inverse composition that first predicts the next latent without action conditioning, then extracts the action. Training concentrates mass on the edges of the (action-time, state-time) noise square, where inference queries the predictor: replacing it with uniform interior sampling drops Push-T planning from 90.1 to 53.3 SR at matched compute. On the LeWM control suite, Qantara reaches a 91.2 SR three-train-seed average and sets new SOTA on OGBench-Cube (+7.7 SR over DINO-WM, +19.7 over LeWM). From the same weights, the behaviour-cloning and video-inverse paths reach 82-83 SR on Push-T and 71-73 SR on Cube. These results move JEPA world models from single-paradigm planners to multi-paradigm controllers.
