arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2604.11007cs.CV

通过开放词汇图像分割的伪标签实现3D点云语义分割的数据高效方法

Data-Efficient Semantic Segmentation of 3D Point Clouds via Open-Vocabulary Image Segmentation-based Pseudo-Labeling

University of Yamanashi(山梨大学)

Takahiko Furuya

AI总结:

本文提出PLOVIS算法,利用开放词汇图像分割模型生成伪标签,解决3D点云分割中训练数据不足的问题,通过两阶段过滤和类平衡内存银行提升训练效果,实验表明在数据稀缺条件下优于现有方法。

摘要:

3D点云场景的语义分割是各种应用中的关键任务。在现实场景中,训练分割模型常面临三种数据不足:训练场景稀缺、点级标注稀缺以及缺乏用于重建点云的2D图像序列。现有数据高效算法通常只解决其中一两种挑战,未探索三者联合处理。本文提出一种专门针对三种数据不足的数据高效训练框架。我们的算法称为Point pseudo-Labeling via Open-Vocabulary Image Segmentation (PLOVIS),利用开放词汇图像分割(OVIS)模型作为伪标签生成器,补偿训练数据的不足。PLOVIS直接从训练3D点云生成2D图像进行伪标签生成,无需2D图像序列。为缓解伪标签中的固有噪声和类别不平衡,我们引入两阶段伪标签过滤结合类平衡内存银行以实现有效训练。两阶段过滤机制首先移除低置信度伪标签,然后丢弃可能错误的伪标签,从而提升伪标签质量。在四个基准数据集(ScanNet、S3DIS、Toronto3D和Semantic3D)上进行实验,实验在现实数据稀缺条件下(几十个训练3D场景,每个仅标注<100个3D点)表明,PLOVIS在数据稀缺条件下持续优于现有方法,包括标准微调策略和最先进的弱监督学习算法。代码将公开发布。

英文摘要:

Semantic segmentation of 3D point cloud scenes is a crucial task for various applications. In real-world scenarios, training segmentation models often faces three concurrent forms of data insufficiency: scarcity of training scenes, scarcity of point-level annotations, and absence of 2D image sequences from which point clouds were reconstructed. Existing data-efficient algorithms typically address only one or two of these challenges, leaving the joint treatment of all three unexplored. This paper proposes a data-efficient training framework specifically designed to address the three forms of data insufficiency. Our proposed algorithm, called Point pseudo-Labeling via Open-Vocabulary Image Segmentation (PLOVIS), leverages an Open-Vocabulary Image Segmentation (OVIS) model as a pseudo label generator to compensate for the lack of training data. PLOVIS creates 2D images for pseudo-labeling directly from training 3D point clouds, eliminating the need for 2D image sequences. To mitigate the inherent noise and class imbalance in pseudo labels, we introduce a two-stage filtering of pseudo labels combined with a class-balanced memory bank for effective training. The two-stage filtering mechanism first removes low-confidence pseudo labels, then discards likely incorrect pseudo labels, thereby enhancing the quality of pseudo labels. Experiments on four benchmark datasets, i.e., ScanNet, S3DIS, Toronto3D, and Semantic3D, under realistic data-scarce conditions (a few tens of training 3D scenes, each annotated with only <100 3D points) demonstrate that PLOVIS consistently outperforms existing methods including standard fine-tuning strategies and state-of-the-art weakly supervised learning algorithms. Code will be made publicly available.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑