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

从泛化角度探索主动3D目标检测

Exploring Active 3D Object Detection from a Generalization Perspective

The University of Queensland(昆士兰大学)

Yadan Luo, Zhuoxiao Chen, Zijian Wang, Xin Yu, Zi Huang, Mahsa Baktashmotlagh

摘要:

为了减轻基于LiDAR的3D目标检测中的高注释成本,主动学习是一种很有前途的解决方案,它学习仅选择一小部分未标记的数据进行注释,而不会影响模型性能。然而,我们的实证研究表明,主流的基于不确定性和基于多样性的主动学习策略在应用于3D检测任务时是无效的,因为它们无法平衡点云信息量和框级注释成本之间的权衡。为了克服这一限制,我们在我们的框架Crb中共同研究了用于点云采集的三个新标准-标签简洁性、特征代表性和几何平衡,其分层过滤掉冗余3D限界框标签、潜在特征和几何特征(例如,点云密度),并从未标记的样本池中选择具有较少对象的信息性样本进行注释。我们的理论分析表明,所提出的标准对齐的边缘分布的选择子集和先验分布的看不见的测试集,并最小化的泛化误差的上界。为了验证Crb的有效性和适用性,我们对KITTI和Waymo的两个基准3D目标检测数据集进行了广泛的实验,并检查了两个单阶段(即,第二)和两级3D检测器(即,Pv-rcnn)。实验证明,所提出的方法优于现有的主动学习策略,并实现完全监督的性能,需要$1\%$和$8\%$注释的边界框和点云,分别。源代码:https://github.com/Luoyadan/CRB-active-3Ddet。

英文摘要:

To alleviate the high annotation cost in LiDAR-based 3D object detection, active learning is a promising solution that learns to select only a small portion of unlabeled data to annotate, without compromising model performance. Our empirical study, however, suggests that mainstream uncertainty-based and diversity-based active learning policies are not effective when applied in the 3D detection task, as they fail to balance the trade-off between point cloud informativeness and box-level annotation costs. To overcome this limitation, we jointly investigate three novel criteria in our framework Crb for point cloud acquisition - label conciseness}, feature representativeness and geometric balance, which hierarchically filters out the point clouds of redundant 3D bounding box labels, latent features and geometric characteristics (e.g., point cloud density) from the unlabeled sample pool and greedily selects informative ones with fewer objects to annotate. Our theoretical analysis demonstrates that the proposed criteria align the marginal distributions of the selected subset and the prior distributions of the unseen test set, and minimizes the upper bound of the generalization error. To validate the effectiveness and applicability of Crb, we conduct extensive experiments on the two benchmark 3D object detection datasets of KITTI and Waymo and examine both one-stage (i.e., Second) and two-stage 3D detectors (i.e., Pv-rcnn). Experiments evidence that the proposed approach outperforms existing active learning strategies and achieves fully supervised performance requiring $1\%$ and $8\%$ annotations of bounding boxes and point clouds, respectively. Source code: https://github.com/Luoyadan/CRB-active-3Ddet.

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

↑