PriFit: Learning to Fit Primitives Improves Few Shot Point Cloud Segmentation
Not on the map. Why?
Abstract (arXiv)
We present PriFit, a semi-supervised approach for label-efficient learning of 3D point cloud segmentation networks. PriFit combines geometric primitive fitting with point-based representation learning. Its key idea is to learn point representations whose clustering reveals shape regions that can be approximated well by basic geometric primitives, such as cuboids and ellipsoids. The learned point representations can then be re-used in existing network architectures for 3D point cloud segmentation, and improves their performance in the few-shot setting. According to our experiments on the widely used ShapeNet and PartNet benchmarks, PriFit outperforms several state-of-the-art methods in this setting, suggesting that decomposability into primitives is a useful prior for learning representations predictive of semantic parts. We present a number of ablative experiments varying the choice of geometric primitives and downstream tasks to demonstrate the effectiveness of the method.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 33 of its 66 references that are in the map. First posted on arXiv 27 Dec 2021.
Builds on
33 of 66 references.
References · 33
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,248
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,897
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,223
- 2018Occupancy Networks: Learning 3D Reconstruction in Function Spacecited by 2,041
- 2018AtlasNet: A Papier-Mach'e Approach to Learning 3D Surface Generationcited by 539
- 2018PartNet: A Large-scale Benchmark for Fine-grained and Hierarchical Part-level 3D Object Understandingcited by 526
- 2020PointContrast: Unsupervised Pre-training for 3D Point Cloud Understandingcited by 504
- 2019PointFlow: 3D Point Cloud Generation with Continuous Normalizing Flowscited by 488
- 2019Learning Shape Templates with Structured Implicit Functionscited by 242
- 2018SO-Net: Self-Organizing Network for Point Cloud Analysiscited by 210
- 2016Learning Shape Abstractions by Assembling Volumetric Primitivescited by 181
21 more
- 2019CvxNet: Learnable Convex Decompositioncited by 174
- 2019Superquadrics Revisited: Learning 3D Shape Parsing beyond Cuboidscited by 124
- 2015Gradient Estimation Using Stochastic Computation Graphscited by 121
- 20183D Point Capsule Networkscited by 106
- 2020ParSeNet: A Parametric Surface Fitting Network for 3D Point Cloudscited by 96
- 2018Multiresolution Tree Networks for 3D Point Cloud Processingcited by 91
- 2018Supervised Fitting of Geometric Primitives to 3D Point Cloudscited by 91
- 2019BAE-NET: Branched Autoencoder for Shape Co-Segmentationcited by 89
- 2019Unsupervised Multi-Task Feature Learning on Point Cloudscited by 89
- 2020Learning Unsupervised Hierarchical Part Decomposition of 3D Objects from a Single RGB Imagecited by 72
- 2017Recurrent Pixel Embedding for Instance Groupingcited by 58
- 2021Unsupervised Learning for Cuboid Shape Abstraction via Joint Segmentation from Point Cloudscited by 44
- 2020Learning to Group: A Bottom-Up Framework for 3D Part Discovery in Unseen Categoriescited by 30
- 2020Label-Efficient Learning on Point Clouds using Approximate Convex Decompositionscited by 27
- 2020Learning Generative Models of Shape Handlescited by 25
- 2020Joint Supervised and Self-Supervised Learning for 3D Real-World Challengescited by 21
- 2020Neural Star Domain as Primitive Representationcited by 17
- 2019MortonNet: Self-Supervised Learning of Local Features in 3D Point Cloudscited by 9
- 2019Learning Point Embeddings from Shape Repositories for Few-Shot Segmentationcited by 5
- 2017Tags2Parts: Discovering Semantic Regions from Shape Tagscited by 4
- 2013Bayesian Inference in Model-Based Machine Visioncited by 1
Built on by
Citing papers per year: 2022: 2; 2023: 3; 2024: 1.
6 papers.
- 2022PartSLIP: Low-Shot Part Segmentation for 3D Point Clouds via Pretrained Image-Language Modelscited by 118
- 2023Marching-Primitives: Shape Abstraction from Signed Distance Functioncited by 19
- 2023Iterative Superquadric Recomposition of 3D Objects from Multiple Viewscited by 13
- 2022MvDeCor: Multi-view Dense Correspondence Learning for Fine-grained 3D Segmentationcited by 10
- 2024SweepNet: Unsupervised Learning Shape Abstraction via Neural Sweeperscited by 5
- 2023Recognising geometric primitives in 3D point clouds of mechanical CAD objectscited by 4
- 2024SweepNet: Unsupervised Learning Shape Abstraction via Neural Sweepers8 Jul 2024
- 2023Iterative Superquadric Recomposition of 3D Objects from Multiple Views5 Sep 2023
- 2023Marching-Primitives: Shape Abstraction from Signed Distance Function23 Mar 2023
- 2023Recognising geometric primitives in 3D point clouds of mechanical CAD objects11 Jan 2023
- 2022PartSLIP: Low-Shot Part Segmentation for 3D Point Clouds via Pretrained Image-Language Models3 Dec 2022
- 2022MvDeCor: Multi-view Dense Correspondence Learning for Fine-grained 3D Segmentation18 Aug 2022