Robust Few-Shot 3D Point Cloud Scene Segmentation

Huang Hao,Yuan Shuaihang,Wen Congcong,Hao Yu,Fang Yi
2024-01-01
Abstract:In the domain of 3D point cloud scene semantic segmentation, a preponderance of methodology predominantly adopts a fully supervised framework. Such paradigms exhibit an intrinsic dependency on extensive labeled datasets, presenting challenges in acquisition and exhibiting incapacity to segment novel classes, especially when the training data are contaminated by noisy samples. To addressing these limitations, this study introduces a novel meta-learning-based few-shot segmentation approach to robustly segment 3D point cloud scenes. Specifically, we first build a multi-prototype graph and then suppress noisy samples based on the graph structure. A subgraph voting scheme is proposed to conduct transductive semi-supervised learning to propagate labels. To optimize the graph structure to learn discriminative prototype features, we design a triplet contrastive loss to increase the compactness of the graph. We eveluate our method on two widely used 3D point cloud scene segmentation benchmarks within specific few-shot segmentation (i.e., 2/3-way 5-shot) settings. Experimental results demonstrate improvement over the compared baseline methods, illustrating the robustness of our method in few-shot 3D scene segmentation against noisy samples.
What problem does this paper attempt to address?