Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristics

About

Learning manipulable representations of the world and its dynamics is central to AI. Joint-Embedding Predictive Architectures (JEPAs) offer a promising blueprint, but lack of practical guidance and theory has led to ad-hoc R&D. We present a comprehensive theory of JEPAs and instantiate it in {\bf LeJEPA}, a lean, scalable, and theoretically grounded training objective. First, we identify the isotropic Gaussian as the optimal distribution that JEPAs' embeddings should follow to minimize downstream prediction risk. Second, we introduce a novel objective--{\bf Sketched Isotropic Gaussian Regularization} (SIGReg)--to constrain embeddings to reach that ideal distribution. Combining the JEPA predictive loss with SIGReg yields LeJEPA with numerous theoretical and practical benefits: (i) single trade-off hyperparameter, (ii) linear time and memory complexity, (iii) stability across hyper-parameters, architectures (ResNets, ViTs, ConvNets) and domains, (iv) heuristics-free, e.g., no stop-gradient, no teacher-student, no hyper-parameter schedulers, and (v) distributed training-friendly implementation requiring only $\approx$50 lines of code. Our empirical validation covers 10+ datasets, 60+ architectures, all with varying scales and domains. As an example, using imagenet-1k for pretraining and linear evaluation with frozen backbone, LeJEPA reaches 79\% with a ViT-H/14. We hope that the simplicity and theory-friendly ecosystem offered by LeJEPA will reestablish self-supervised pre-training as a core pillar of AI research (\href{https://github.com/rbalestr-lab/lejepa}{GitHub repo}).

Randall Balestriero, Yann LeCun• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationFood-101--
542
Image ClassificationCIFAR-100--
302
Image ClassificationImageNet
Top-1 Acc79
65
Image ClassificationSTL-10
Top-1 Acc80.73
24
Image ClassificationImageNet-1000
Top-1 Accuracy53.5
16
Semantic segmentationWaymo
mIoU0.261
14
Object DetectionWaymo
mAP XY19.3
7
Image ClassificationCIFAR-100
Top-1 Accuracy70.2
3
Showing 8 of 8 rows

Other info

Follow for update