A Simple and Efficient Point Cloud Sampling Strategy Based on Cluster Merging

Zhuhua Bai,Weiqing Li,Guolin Yang,Fantong Meng,Renke Kang,Zhigang Dong
DOI: https://doi.org/10.1109/icraic61978.2023.00022
2023-01-01
Abstract:With the continuous advancement of 3D vision, point clouds, as a significant representation form in 3D, have gained increasing attention. Point cloud downsampling is an important operation performed on point clouds in order to unify data size and reduce computational cost. We find that in point-based networks either traditional farthest point sampling (FPS) or task-aware learning-based sampling strategies, they often have high time complexity, becoming a bottleneck for efficiency. To address this problem, we design a cluster-based point token merging module (CPM), which is a novel and efficient taskaware sampling module with simple design that can be inserted into various point-based networks. It generates clusters of point tokens with different locations and sizes by progressively clustering them in the feature domain, and merges point tokens inside of the same cluster into a single one as the output, thus enabling downsampling. We conduct extensive experiments on traditional classification and segmentation tasks. Experiments demonstrate that our CPM sampling module can substantially improve the inference speed while maintaining accuracy. For example, when it is inserted into the Point Transformer model and tested on the ModelNet40 dataset, its accuracy can reach 94.4% (+ 1.2%), the inference speed is improved by 48.2%, and it is more robust to noise points. We will release the code after the paper is accepted for publication.
What problem does this paper attempt to address?