提出新型点云自监督学习方法,解决激光雷达数据空洞难题
Multi-Scale Neighborhood Occupancy Masked Autoencoder for Self-Supervised Learning in LiDAR Point Clouds
- 仅在非掩码体素邻域重建占用信息,避免信息泄露
- 多尺度掩码生成捕捉不同大小物体特征,性能超越现有方法
- 适配主流3D模型,适用于语义分割与目标检测任务
掩码自编码器(MAE)在视觉等领域的自监督学习中表现优异,但激光雷达点云因三维空间大面积空洞,导致现有方法存在占用信息泄露和计算复杂度高的问题,实践中仅能使用2D鸟瞰图编码器进行预训练。本文提出新型邻域占用掩码自编码器(NOMAE),通过仅在非掩码体素邻域内进行占用重建,有效克服上述挑战。结合多尺度体素掩码与占用重建,并引入层级掩码生成策略,以捕捉点云中不同尺寸物体的特征。NOMAE具有高度灵活性,可直接用于现有3D架构的自监督学习。我们在nuScenes和Waymo Open数据集上对语义分割和3D目标检测任务进行了广泛评估,对比了判别式与生成式自监督方法。结果表明,NOMAE在多个基准测试中均达到新最佳性能。
原文摘要 · Abstract (English)
Masked autoencoders (MAE) have shown tremendous potential for self-supervised learning (SSL) in vision and beyond. However, point clouds from LiDARs used in automated driving are particularly challenging for MAEs since large areas of the 3D volume are empty. Consequently, existing work suffers from leaking occupancy information into the decoder and has significant computational complexity, thereby limiting the SSL pre-training to only 2D bird's eye view encoders in practice. In this work, we propose the novel neighborhood occupancy MAE (NOMAE) that overcomes the aforementioned challenges by employing masked occupancy reconstruction only in the neighborhood of non-masked voxels. We incorporate voxel masking and occupancy reconstruction at multiple scales with our proposed hierarchical mask generation technique to capture features of objects of different sizes in the point cloud. NOMAEs are extremely flexible and can be directly employed for SSL in existing 3D architectures. We perform extensive evaluations on the nuScenes and Waymo Open datasets for the downstream perception tasks of semantic segmentation and 3D object detection, comparing with both discriminative and generative SSL methods. The results demonstrate that NOMAE sets the new state-of-the-art on multiple benchmarks for multiple point cloud perception tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。