Point Transformer

Self-attention networks have revolutionized natural language processing and are making impressive strides in image analysis tasks such as image classification and object detection. Inspired by this success, we investigate the application of self-attention networks to 3D point cloud processing. We design self-attention layers for point clouds and use these to construct self-attention networks for tasks such as semantic scene segmentation, object part segmentation, and object classification. Our Point Transformer design improves upon prior work across domains and tasks. For example, on the challenging S3DIS dataset for large-scale semantic scene segmentation, the Point Transformer attains an mIoU of 70.4% on Area 5, outperforming the strongest prior model by 3.3 absolute percentage points and crossing the 70% mIoU threshold for the first time.

PDF Abstract ICCV 2021 PDF ICCV 2021 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
3D Point Cloud Classification ModelNet40 PointTransformer Overall Accuracy 93.7 # 40
Mean Accuracy 90.6 # 25
Point Cloud Segmentation PointCloud-C PointTransformers mean Corruption Error (mCE) 1.049 # 7
3D Semantic Segmentation S3DIS PointTransformer mIoU (6-Fold) 73.5 # 4
mIoU (Area-5) 70.4 # 4
Semantic Segmentation S3DIS PointCNN Mean IoU 65.4 # 35
Number of params N/A # 1
Semantic Segmentation S3DIS KPConv Mean IoU 70.6 # 22
Number of params 14.1M # 46
Params (M) 14.1 # 6
Semantic Segmentation S3DIS PointTransformer Mean IoU 73.5 # 16
mAcc 81.9 # 16
oAcc 90.2 # 11
Number of params 7.8M # 43
Params (M) 7.8 # 10
Semantic Segmentation S3DIS PointNet Mean IoU 47.6 # 50
Number of params N/A # 1
Semantic Segmentation S3DIS SPGraph Mean IoU 62.1 # 40
Number of params N/A # 1
Semantic Segmentation S3DIS Area5 PointCNN mIoU 57.3 # 45
Number of params N/A # 2
Semantic Segmentation S3DIS Area5 PointTransformer mIoU 70.4 # 24
oAcc 90.8 # 17
mAcc 76.5 # 21
Number of params 7.8M # 47
Semantic Segmentation S3DIS Area5 PointNet mIoU 41.1 # 48
Number of params N/A # 2
3D Part Segmentation ShapeNet-Part PointTransformer Class Average IoU 83.7 # 17
Instance Average IoU 86.6 # 14
3D Semantic Segmentation STPLS3D Point transformer mIOU 47.64 # 3

Methods