KerJEPA: Kernel Discrepancies for Euclidean Self-Supervised Learning

arXiv:2512.19605 · cs.LG, cs.CV · Submitted 2025-12-22 · Read on arXiv

cs.LG, cs.CV

Submitted: 2025-12-22

Updated: 2026-10-13

Code: https://github.com/fastai/imagenette

License: http://creativecommons.org/licenses/by/4.0/

The gist: Breakthroughs in self-supervised Joint-Embedding Predictive Architectures (JEPAs) have established that regularizing Euclidean representations toward isotropic Gaussian priors yields provable gains

Terminology

Abstract

Breakthroughs in self-supervised Joint-Embedding Predictive Architectures (JEPAs) have established that regularizing Euclidean representations toward isotropic Gaussian priors yields provable gains in training stability and downstream generalization. We introduce a new, flexible family of Euclidean joint embedding self-supervised learning algorithms with kernel-based regularizers. One instance of this family corresponds to the SIGReg Epps-Pulley regularizer which approximates a sliced Maximum Mean Discrepancy (MMD) with a Gaussian prior and Gaussian kernel. By expanding the class of viable kernels and priors and computing the closed-form high-dimensional limit of sliced discrepancies, we develop alternative methods with improved training stability and design flexibility and leverage those techniques to train strong vision learners at scale.

Sources

Related papers