用SAM增强3D语义分割伪标签,提升跨域适应性能
SAM-guided Pseudo Label Enhancement for Multi-modal 3D Semantic Segmentation
- 利用SAM图像掩码为3D点云提供互补先验,生成更可靠的伪标签
- 通过几何感知传播算法,使伪标签覆盖更多点且减少误标
- 在多个数据集上显著提升伪标签质量与领域适应效果
多模态3D语义分割对自动驾驶和虚拟现实等应用至关重要。为实现在真实场景中的有效部署,需采用跨域适应技术弥合训练数据与真实数据间的差距。近年来,基于伪标签的自训练已成为该任务主流方法,但生成可靠伪标签需严格约束,常导致伪标签稀疏,阻碍性能提升。本文提出一种图像引导的伪标签增强方法,利用分割一切模型(SAM)的2D先验知识,在配对图像掩码基础上,将对应3D点聚集至同一对象区域,并分两步优化伪标签:首先通过多数投票确定掩码类别,并施加多种约束剔除不可靠标签;其次引入几何感知渐进传播(GAPP),在保留结构一致性的同时,将标签传播至掩码内所有3D点,避免因2D-3D错位引发的异常点。在多个数据集与跨域场景下的实验表明,本方法显著增加了高质量伪标签数量,优于基线方法。
原文摘要 · Abstract (English)
Multi-modal 3D semantic segmentation is vital for applications such as autonomous driving and virtual reality (VR). To effectively deploy these models in real-world scenarios, it is essential to employ cross-domain adaptation techniques that bridge the gap between training data and real-world data. Recently, self-training with pseudo-labels has emerged as a predominant method for cross-domain adaptation in multi-modal 3D semantic segmentation. However, generating reliable pseudo-labels necessitates stringent constraints, which often result in sparse pseudo-labels after pruning. This sparsity can potentially hinder performance improvement during the adaptation process. We propose an image-guided pseudo-label enhancement approach that leverages the complementary 2D prior knowledge from the Segment Anything Model (SAM) to introduce more reliable pseudo-labels, thereby boosting domain adaptation performance. Specifically, given a 3D point cloud and the SAM masks from its paired image data, we collect all 3D points covered by each SAM mask that potentially belong to the same object. Then our method refines the pseudo-labels within each SAM mask in two steps. First, we determine the class label for each mask using majority voting and employ various constraints to filter out unreliable mask labels. Next, we introduce Geometry-Aware Progressive Propagation (GAPP) which propagates the mask label to all 3D points within the SAM mask while avoiding outliers caused by 2D-3D misalignment. Experiments conducted across multiple datasets and domain adaptation scenarios demonstrate that our proposed method significantly increases the quantity of high-quality pseudo-labels and enhances the adaptation performance over baseline methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。