arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2206.04028cs.CVcs.RO
收藏

CO³:协作无监督3D表示学习用于自动驾驶

CO^3: Cooperative Unsupervised 3D Representation Learning for Autonomous Driving

The University of Hong Kong(香港大学);Shanghai AI Laboratory(上海人工智能实验室);Huawei Noah’s Ark Lab(华为诺亚实验室);The Chinese University of Hong Kong(香港中文大学);Hong Kong University of Science and Technology(香港科技大学)

Runjian Chen, Yao Mu, Runsen Xu, Wenqi Shao, Chenhan Jiang, Hang Xu, Zhenguo Li, Ping Luo

AI总结:

CO³通过协作对比学习和上下文形状预测,实现无监督学习户外场景点云的3D表示,提升自动驾驶中的检测性能。

摘要:

无监督对比学习在室内场景点云中取得了巨大成功。然而,无监督学习户外场景点云仍然具有挑战性,因为先前的方法需要重建整个场景并捕获部分视图以实现对比目标。这在户外场景中不可行,因为存在移动物体、障碍物和传感器。在本文中,我们提出了CO³,即协作对比学习和上下文形状预测,以无监督的方式学习户外场景点云的3D表示。CO³相比现有方法有多个优点。(1) 它利用车辆侧和基础设施侧的LiDAR点云来构建差异足够但同时保持共同语义信息的视图,这比以前方法构建的视图更合适。(2) 除了对比目标外,提出了形状上下文预测作为预训练目标,为无监督3D点云表示学习带来更多的任务相关信息,这在将学习到的表示转移到下游检测任务时是有益的。(3) 与先前方法相比,CO³学习到的表示可以转移到由不同类型的LiDAR传感器收集的不同户外场景数据集。(4) 相比于现有方法,CO³在Once和KITTI数据集上将当前最先进的方法提高了高达2.58 mAP。我们相信CO³将有助于理解户外场景的LiDAR点云。

英文摘要:

Unsupervised contrastive learning for indoor-scene point clouds has achieved great successes. However, unsupervised learning point clouds in outdoor scenes remains challenging because previous methods need to reconstruct the whole scene and capture partial views for the contrastive objective. This is infeasible in outdoor scenes with moving objects, obstacles, and sensors. In this paper, we propose CO^3, namely Cooperative Contrastive Learning and Contextual Shape Prediction, to learn 3D representation for outdoor-scene point clouds in an unsupervised manner. CO^3 has several merits compared to existing methods. (1) It utilizes LiDAR point clouds from vehicle-side and infrastructure-side to build views that differ enough but meanwhile maintain common semantic information for contrastive learning, which are more appropriate than views built by previous methods. (2) Alongside the contrastive objective, shape context prediction is proposed as pre-training goal and brings more task-relevant information for unsupervised 3D point cloud representation learning, which are beneficial when transferring the learned representation to downstream detection tasks. (3) As compared to previous methods, representation learned by CO^3 is able to be transferred to different outdoor scene dataset collected by different type of LiDAR sensors. (4) CO^3 improves current state-of-the-art methods on both Once and KITTI datasets by up to 2.58 mAP. We believe CO^3 will facilitate understanding LiDAR point clouds in outdoor scene.

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

↑