arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.08760cs.CV

MIM4D:用于自动驾驶表示学习的多视角视频掩码建模

MIM4D: Masked Modeling with Multi-View Video for Autonomous Driving Representation Learning

School of EIC, Huazhong University of Science & Technology(华中科技大学电子信息与通信学院);Institute of Artificial Intelligence, Huazhong University of Science & Technology(华中科技大学人工智能研究院);Horizon Robotics

Jialv Zou, Bencheng Liao, Qian Zhang, Wenyu Liu, Xinggang Wang

AI总结:

提出MIM4D,一种基于双重掩码图像建模的多视角视频预训练方法,利用场景流和3D可微渲染学习时空表示,在nuScenes上显著提升BEV分割、3D检测和HD地图构建性能。

摘要:

从大规模多视角视频数据中学习鲁棒且可扩展的视觉表示,仍然是计算机视觉和自动驾驶领域的一项挑战。现有的预训练方法要么依赖带有3D标注的昂贵监督学习,限制了可扩展性,要么专注于单帧或单目输入,忽略了时间信息。我们提出了MIM4D,一种基于双重掩码图像建模(MIM)的新型预训练范式。MIM4D通过对掩码多视角视频输入进行训练,利用空间和时间关系。它使用连续场景流构建伪3D特征,并将其投影到2D平面以进行监督。为了解决缺乏密集3D监督的问题,MIM4D采用3D体素可微渲染重建像素,以学习几何表示。我们证明,MIM4D在nuScenes数据集上实现了自动驾驶视觉表示学习的最先进性能。它在多个下游任务上显著改进了现有方法,包括BEV分割(8.7% IoU)、3D目标检测(3.5% mAP)和HD地图构建(1.4% mAP)。我们的工作为自动驾驶中大规模学习表示提供了新的选择。代码和模型已在https://github.com/hustvl/MIM4D发布。

英文摘要:

Learning robust and scalable visual representations from massive multi-view video data remains a challenge in computer vision and autonomous driving. Existing pre-training methods either rely on expensive supervised learning with 3D annotations, limiting the scalability, or focus on single-frame or monocular inputs, neglecting the temporal information. We propose MIM4D, a novel pre-training paradigm based on dual masked image modeling (MIM). MIM4D leverages both spatial and temporal relations by training on masked multi-view video inputs. It constructs pseudo-3D features using continuous scene flow and projects them onto 2D plane for supervision. To address the lack of dense 3D supervision, MIM4D reconstruct pixels by employing 3D volumetric differentiable rendering to learn geometric representations. We demonstrate that MIM4D achieves state-of-the-art performance on the nuScenes dataset for visual representation learning in autonomous driving. It significantly improves existing methods on multiple downstream tasks, including BEV segmentation (8.7% IoU), 3D object detection (3.5% mAP), and HD map construction (1.4% mAP). Our work offers a new choice for learning representation at scale in autonomous driving. Code and models are released at https://github.com/hustvl/MIM4D

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑