AD-L-JEPA: Self-Supervised Spatial World Models with Joint Embedding Predictive Architecture for Autonomous Driving with LiDAR Data
TLDR
AD-L-JEPA is a self-supervised JEPA-based pre-training framework for LiDAR 3D object detection that predicts Bird's-Eye-View embeddings, achieving state-of-the-art results with reduced compute.
Reasoning
Strengths include a novel non-generative, non-contrastive approach with strong empirical results on multiple large-scale datasets and significant compute savings. Weaknesses are its narrow focus on LiDAR object detection without general world modeling capabilities, and no discussion of limitations.
Read-first score
Read-first score 49.6, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 6.
Field roles
Rank sensitivity
Stability: volatile; rank range: 644.
Keyword Scores
Deep Analysis
Innovations
- First JEPA-based pre-training method for autonomous driving with LiDAR data.
- Predicts Bird's-Eye-View embeddings instead of explicitly generating masked regions.
- Uses explicit variance regularization to avoid representation collapse without manual contrastive pairs.
- Achieves better quality, faster training (1.9x-2.7x GPU hours reduction), and lower GPU memory (2.8x-4x reduction) compared to Occupancy-MAE.
Methodology
AD-L-JEPA is a self-supervised pre-training framework based on a joint embedding predictive architecture (JEPA) for LiDAR 3D object detection. It predicts Bird's-Eye-View embeddings from masked LiDAR data and employs explicit variance regularization to prevent representation collapse, avoiding both generative and contrastive approaches. The model is pre-trained on unlabeled LiDAR data and evaluated on downstream detection tasks using KITTI3D, Waymo, and ONCE datasets.
Key Results
On the ONCE dataset, pre-training on 100K frames yields a 1.61 mAP gain, outperforming all other methods pre-trained on up to 500K frames; pre-training on 500K frames yields a 2.98 mAP gain, outperforming all methods pre-trained on up to 1M frames. Additionally, AD-L-JEPA reduces GPU hours by 1.9x-2.7x and GPU memory by 2.8x-4x compared to Occupancy-MAE.