Branch-JEPA: Finite-Support Predictive Distributions for JEPA World Models

Zhi Song, Ximing Xing, Zhenchao Tang +9

Joint-embedding predictive architectures (JEPAs) learn dynamics by predicting future observations in representation space. Yet most JEPA world models return one latent successor, e…