arXiv:2507.06592cs.CV2025-07中稿 · publication in IEE…被引 4

针对点云分割中模糊区域标签不可靠问题,提出自适应对比学习方法。

Ambiguity-aware Point Cloud Segmentation by Adaptive Margin Contrastive Learning

论文配图:Ambiguity-aware Point Cloud Segmentation by Adaptive Margin Contrastive Learning
图 1 · 摘自论文原文
  • 根据每个点的模糊程度动态调整损失权重,避免对模糊点强约束
  • 在S3DIS和ScanNet上分别提升1.8%和1.2%的平均交并比
  • 适合处理标注模糊的3D场景分割任务,尤其对边界区域有效

本文提出一种自适应边缘对比学习方法(AMContrast3D),用于点云的3D语义分割。现有方法多采用统一惩罚目标,忽略过渡区域中因模糊性导致的特征不区分性。由于高度模糊点人类也难以区分,其人工标注可靠性低,对这些点施加硬约束会损害模型性能。为此,AMContrast3D将对比学习融入模糊度估计框架,根据各点模糊程度自适应设定目标。该方法在保证低模糊点准确性的前提下,允许高模糊点存在错误。此外,为克服推理时模糊度假设单一的问题,进一步提出AMContrast3D++,通过双分支联合训练,引入新型模糊度预测模块,从生成嵌入中同步学习模糊度,并设计掩码精炼机制,利用预测模糊度提升模糊嵌入的可靠性,从而增强分割性能与鲁棒性。在S3DIS和ScanNet两个3D室内场景数据集上的实验表明该方法有效。代码已开源。

原文摘要 · Abstract (English)

This paper proposes an adaptive margin contrastive learning method for 3D semantic segmentation on point clouds. Most existing methods use equally penalized objectives, which ignore the per-point ambiguities and less discriminated features stemming from transition regions. However, as highly ambiguous points may be indistinguishable even for humans, their manually annotated labels are less reliable, and hard constraints over these points would lead to sub-optimal models. To address this, we first design AMContrast3D, a method comprising contrastive learning into an ambiguity estimation framework, tailored to adaptive objectives for individual points based on ambiguity levels. As a result, our method promotes model training, which ensures the correctness of low-ambiguity points while allowing mistakes for high-ambiguity points. As ambiguities are formulated based on position discrepancies across labels, optimization during inference is constrained by the assumption that all unlabeled points are uniformly unambiguous, lacking ambiguity awareness. Inspired by the insight of joint training, we further propose AMContrast3D++ integrating with two branches trained in parallel, where a novel ambiguity prediction module concurrently learns point ambiguities from generated embeddings. To this end, we design a masked refinement mechanism that leverages predicted ambiguities to enable the ambiguous embeddings to be more reliable, thereby boosting segmentation performance and enhancing robustness. Experimental results on 3D indoor scene datasets, S3DIS and ScanNet, demonstrate the effectiveness of the proposed method. Code is available at https://github.com/YangChenApril/AMContrast3D.

点云分割对比学习模糊性建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。