用联合嵌入预测架构,让激光雷达目标检测预训练更高效准确
Self-Supervised Representation Learning with Joint Embedding Predictive Architecture for Automotive LiDAR Object Detection
- 不生成也不对比,直接预测鸟瞰图嵌入来学习场景特征
- 在多个数据集上提升检测性能,且训练节省1.9到2.7倍显存时间
- 适合需要高效自监督预训练的自动驾驶感知研究者
近年来,基于大量未标注数据的自监督表示学习被探索为自动驾驶的预训练方法。然而,直接应用主流对比或生成方法效果不佳,甚至引发负迁移。本文提出AD-L-JEPA,一种基于联合嵌入预测架构(JEPA)的新型自监督预训练框架,专用于车载激光雷达3D目标检测。不同于现有方法,AD-L-JEPA既非生成也非对比型。它不显式重建缺失区域,而是预测鸟瞰图(Bird's-Eye-View)嵌入以捕捉驾驶场景多样性。此外,通过显式方差正则化避免表示坍塌,无需手动构造对比对。实验表明,在KITTI3D、Waymo和ONCE数据集上,下游3D目标检测任务性能持续提升;相比先进方法Occupancy-MAE,GPU耗时减少1.9–2.7倍,显存占用降低2.8–4倍。尤其在最大数据集ONCE上,使用10万帧预训练获得1.61 mAP提升,优于所有其他在10万或50万帧上预训练的方法;使用50万帧预训练则获得2.98 mAP提升,优于所有在50万或100万帧上预训练的方法。AD-L-JEPA是首个应用于自动驾驶的JEPA类预训练方法,兼具更高质量、更快速度与更低显存开销。代码即将开源。
原文摘要 · Abstract (English)
Recently, self-supervised representation learning relying on vast amounts of unlabeled data has been explored as a pre-training method for autonomous driving. However, directly applying popular contrastive or generative methods to this problem is insufficient and may even lead to negative transfer. In this paper, we present AD-L-JEPA, a novel self-supervised pre-training framework with a joint embedding predictive architecture (JEPA) for automotive LiDAR object detection. Unlike existing methods, AD-L-JEPA is neither generative nor contrastive. Instead of explicitly generating masked regions, our method predicts Bird's-Eye-View embeddings to capture the diverse nature of driving scenes. Furthermore, our approach eliminates the need to manually form contrastive pairs by employing explicit variance regularization to avoid representation collapse. Experimental results demonstrate consistent improvements on the LiDAR 3D object detection downstream task across the KITTI3D, Waymo, and ONCE datasets, while reducing GPU hours by 1.9x-2.7x and GPU memory by 2.8x-4x compared with the state-of-the-art method Occupancy-MAE. Notably, on the largest ONCE dataset, pre-training on 100K frames yields a 1.61 mAP gain, better than all other methods pre-trained on either 100K or 500K frames, and pre-training on 500K frames yields a 2.98 mAP gain, better than all other methods pre-trained on either 500K or 1M frames. AD-L-JEPA constitutes the first JEPA-based pre-training method for autonomous driving. It offers better quality, faster, and more GPU-memory-efficient self-supervised representation learning. The source code of AD-L-JEPA is ready to be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。