Qantara: Bridge-Flow Training for Multi-Paradigm JEPA Control
TLDR
Qantara introduces a joint training objective for JEPA world models enabling multiple inference paradigms from a single checkpoint, achieving SOTA on OGBench-Cube.
Reasoning
The paper presents a novel training method combining Brownian-bridge interpolant and flow matching, enabling latent planning, behavior cloning, and inverse dynamics from one checkpoint. Strengths include SOTA results and multi-paradigm flexibility; weaknesses include complexity and limited scope to latent world models.
Read-first score
Read-first score 45.9, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 54.
Field roles
Rank sensitivity
Stability: volatile; rank range: 464.
Keyword Scores
Deep Analysis
Innovations
- Single JEPA checkpoint serving three inference paradigms (latent planning, behaviour cloning, inverse dynamics) without retraining
- Joint training objective combining Brownian-bridge interpolant on state axis with noise-to-data flow matching on action axis
- Video-inverse composition for inverse dynamics: predict next latent without action conditioning, then extract action
- Training mass concentration on edges of (action-time, state-time) noise square to align with inference queries
Methodology
Qantara is an end-to-end JEPA trained with a joint objective that pairs a Brownian-bridge interpolant between consecutive clean latents on the state axis with noise-to-data flow matching on the action axis. Training concentrates probability mass on the edges of the (action-time, state-time) noise square, where inference queries the predictor. A single checkpoint supports latent planning, behaviour-cloning action sampling, and inverse dynamics via a video-inverse composition that first predicts the next latent without action conditioning and then extracts the action.
Key Results
On the LeWM control suite, Qantara achieves 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, behaviour-cloning and video-inverse paths reach 82-83 SR on Push-T and 71-73 SR on Cube. Uniform interior sampling during training reduces Push-T planning from 90.1 to 53.3 SR.
Limitations
- Performance is sensitive to training distribution: replacing edge-concentrated sampling with uniform interior sampling drastically reduces planning success rate (90.1 to 53.3 SR on Push-T).