JEPA (Joint Embedding Predictive Architecture) does predictive self-supervised learning in latent space: a context encoder
g takes the current latent
zt and action
at and predicts the target representation
f(zt+1), where the target encoder
f is updated by EMA (exponential moving average). The training loss is:
ℓ=∥g(zt,at)−f(zt+1)∥2
i.e. mean squared error between the predicted and target representations. The EMA target plus the asymmetric structure (no gradient to
f) prevents representational collapse, so no negative samples are needed.