Label-Efficient Point Cloud Segmentation with Active Learning
This work addresses annotation efficiency for 3D point cloud segmentation, offering a simpler alternative to complex methods, though it appears incremental in its approach.
The paper tackles the high annotation cost in 3D point cloud semantic segmentation by proposing an active learning method that uses a 2D grid for region subdivision and network ensemble for uncertainty estimation, achieving performance on par with or better than state-of-the-art methods on multiple datasets.
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.