用网格划分+集成模型不确定性,高效减少点云标注量。
Label-Efficient Point Cloud Segmentation with Active Learning
- 用2D网格将点云切分为可标注区域,简化处理流程。
- 在S3DIS等数据集上,性能媲美甚至超越复杂方法。
- 建议以标注区域面积为衡量标准,更符合点云特性。
3D点云语义分割面临高昂的标注成本。主动学习通过自动选择需标注数据,降低总标注量以达到满意性能。现有3D点云主动学习方法常依赖复杂的启发式策略,对点云进行可标注区域划分并选取最有益于网络训练的数据。本文提出一种新颖且易实现的区域划分策略:利用2D网格将点云划分为列。为确定下一步标注目标,采用网络集成估计输出不确定性。我们在S3DIS、Toronto-3D及弗莱堡城市大规模3D点云数据集(部分人工标注)上评估该方法。实验表明,本方法在所有数据集上的表现均与或优于现有复杂先进方法。此外,结果提示在点云场景中,标注区域面积比标注点数更能有效衡量主动学习算法性能。
原文摘要 · Abstract (English)
Semantic segmentation of 3D point cloud data often comes with high annotation costs. Active learning automates the process of selecting which data to annotate, reducing the total amount of annotation needed to achieve satisfactory performance. Recent approaches to active learning for 3D point clouds are often based on sophisticated heuristics for both, splitting point clouds into annotatable regions and selecting the most beneficial for further neural network training. In this work, we propose a novel and easy-to-implement strategy to separate the point cloud into annotatable regions. In our approach, we utilize a 2D grid to subdivide the point cloud into columns. To identify the next data to be annotated, we employ a network ensemble to estimate the uncertainty in the network output. We evaluate our method on the S3DIS dataset, the Toronto-3D dataset, and a large-scale urban 3D point cloud of the city of Freiburg, which we labeled in parts manually. The extensive evaluation shows that our method yields performance on par with, or even better than, complex state-of-the-art methods on all datasets. Furthermore, we provide results suggesting that in the context of point clouds the annotated area can be a more meaningful measure for active learning algorithms than the number of annotated points.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。