LCPFormer: Towards Effective 3D Point Cloud Analysis via Local Context Propagation in Transformers

23 Oct 2022  ·  Zhuoxu Huang, Zhiyou Zhao, Banghuai Li, Jungong Han ·

Transformer with its underlying attention mechanism and the ability to capture long-range dependencies makes it become a natural choice for unordered point cloud data. However, separated local regions from the general sampling architecture corrupt the structural information of the instances, and the inherent relationships between adjacent local regions lack exploration, while local structural information is crucial in a transformer-based 3D point cloud model. Therefore, in this paper, we propose a novel module named Local Context Propagation (LCP) to exploit the message passing between neighboring local regions and make their representations more informative and discriminative. More specifically, we use the overlap points of adjacent local regions (which statistically show to be prevalent) as intermediaries, then re-weight the features of these shared points from different local regions before passing them to the next layers. Inserting the LCP module between two transformer layers results in a significant improvement in network expressiveness. Finally, we design a flexible LCPFormer architecture equipped with the LCP module. The proposed method is applicable to different tasks and outperforms various transformer-based methods in benchmarks including 3D shape classification and dense prediction tasks such as 3D object detection and semantic segmentation. Code will be released for reproduction.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
3D Point Cloud Classification ModelNet40 LCPFormer Overall Accuracy 93.6 # 43
Mean Accuracy 90.7 # 23
Semantic Segmentation S3DIS Area5 LCPFormer mIoU 70.2 # 25
oAcc 90.8 # 17
mAcc 76.8 # 20
Number of params N/A # 2
3D Semantic Segmentation SensatUrban LCPFormer mIoU 63.4 # 1
3D Object Detection SUN-RGBD val LCPFormer mAP@0.25 63.2 # 14
mAP@0.5 46.2 # 14

Methods


No methods listed for this paper. Add relevant methods here