Decoupled Local Aggregation for Point Cloud Learning

31 Aug 2023  ·  Binjie Chen, Yunzhou Xia, Yu Zang, Cheng Wang, Jonathan Li ·

The unstructured nature of point clouds demands that local aggregation be adaptive to different local structures. Previous methods meet this by explicitly embedding spatial relations into each aggregation process. Although this coupled approach has been shown effective in generating clear semantics, aggregation can be greatly slowed down due to repeated relation learning and redundant computation to mix directional and point features. In this work, we propose to decouple the explicit modelling of spatial relations from local aggregation. We theoretically prove that basic neighbor pooling operations can too function without loss of clarity in feature fusion, so long as essential spatial information has been encoded in point features. As an instantiation of decoupled local aggregation, we present DeLA, a lightweight point network, where in each learning stage relative spatial encodings are first formed, and only pointwise convolutions plus edge max-pooling are used for local aggregation then. Further, a regularization term is employed to reduce potential ambiguity through the prediction of relative coordinates. Conceptually simple though, experimental results on five classic benchmarks demonstrate that DeLA achieves state-of-the-art performance with reduced or comparable latency. Specifically, DeLA achieves over 90\% overall accuracy on ScanObjectNN and 74\% mIoU on S3DIS Area 5. Our code is available at https://github.com/Matrix-ASC/DeLA .

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
3D Point Cloud Classification ModelNet40 DeLA Overall Accuracy 94.0 # 22
Mean Accuracy 92.2 # 3
Number of params 5.3M # 99
FLOPs 1.44G # 1
Semantic Segmentation S3DIS Area5 DeLA mIoU 74.1 # 4
oAcc 92.2 # 3
mAcc 80.0 # 3
Number of params 7.0M # 46
Semantic Segmentation ScanNet DeLA val mIoU 75.9 # 8
3D Point Cloud Classification ScanObjectNN DeLA Overall Accuracy 90.4 # 10
Mean Accuracy 89.3 # 3
Number of params 5.3M # 58
FLOPs 1.5G # 1

Methods


No methods listed for this paper. Add relevant methods here