PointCNN: Convolution On X-Transformed Points
Yangyan LiRui BuMingchao SunWei WuXinhan DiBaoquan Chen
Proposes PointCNN, a framework that generalizes standard convolution to irregular 3D point clouds by learning an $\mathcal{X}$-transformation matrix that simultaneously weights local features and permutes points into a canonical order.
Point clouds from 3D sensors are irregular and unordered, so standard convolution operators that work well on grid data such as images lose shape information and become sensitive to point ordering. This limits the direct use of proven CNN techniques for tasks such as object classification and scene segmentation that rely on 3D sensing.
The paper introduces PointCNN, a framework that generalizes CNNs to point clouds. Its central operator, X-Conv, learns a K-by-K transformation matrix from local point coordinates with a multilayer perceptron, then uses that matrix to weight and reorder neighboring features before a conventional convolution is applied. The operator is stacked hierarchically with down-sampling and optional dilation, producing networks that aggregate information from increasing receptive fields while remaining computationally practical.
Evaluations on ModelNet40, ScanNet, ShapeNet Parts, S3DIS and additional sketch and image-derived sets show PointCNN reaching or exceeding the accuracy of prior point-cloud networks, with overall accuracies of 92.5 percent on pre-aligned ModelNet40 and 86.14 percent part-averaged IoU on ShapeNet Parts. Ablation tests confirm that removing the learned transformation drops accuracy by roughly 1.5 points even when model depth and parameter count are matched. The same architecture also performs competitively on 2D sketches and MNIST points, though a larger gap remains on CIFAR10 points that lack clear shape structure. Runtime measurements indicate low parameter counts and inference times on the order of 0.012 seconds per batch for classification.
These results indicate that a learned, neighborhood-specific reordering step can restore the spatial-correlation benefits of convolution without forcing point clouds into dense grids. The approach therefore offers a practical route to hierarchical feature learning on raw 3D data for robotics, autonomous driving and indoor mapping applications.
Further gains may come from tighter integration with image-based CNNs on paired data and from refined sampling strategies that handle highly non-uniform point distributions. The current X-transformations achieve only partial permutation invariance, so additional theoretical analysis of the operator inside deep stacks would strengthen confidence in scaling the method. Code is released to support such extensions.
- Paper: PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation, Charles R. Qi et al. (2017). PointNet established the foundational architecture for directly processing unstructured 3D point clouds via permutation-invariant max-pooling, which PointCNN builds upon and generalizes.
- Paper: PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space, C. Qi et al. (2017). PointNet++ introduced hierarchical feature learning and local neighborhood processing for point clouds, directly inspiring PointCNN's localized convolution operators.
- Paper: Dynamic Graph CNN for Learning on Point Clouds, Yue Wang et al. (2018). Dynamic Graph CNN builds directly upon PointCNN by introducing dynamic edge convolutions that capture local geometric structures based on feature-space proximity rather than static transformations.
- Paper: PointRCNN: 3D Object Proposal Generation and Detection From Point Cloud, Shaoshuai Shi et al. (2019). PointRCNN extends point cloud feature learning frameworks like PointCNN into robust two-stage 3D object detection and proposal generation architectures for autonomous driving.
- Paper: KPConv: Flexible and Deformable Convolution for Point Clouds, Hugues Thomas et al. (2019). KPConv continues the progression of point cloud convolutions by introducing flexible and deformable kernel points that directly succeed PointCNN's transformation-based approach.
