跳转至

GeoGuide: Hierarchical Geometric Guidance for Open-Vocabulary 3D Semantic Segmentation

会议: CVPR 2026
arXiv: 2603.26260
代码: 无
领域: 分割
关键词: 开放词表3D语义分割, 几何先验, 2D到3D蒸馏, 超点聚合, 实例级一致性

一句话总结

本文提出 GeoGuide,一个层次化几何引导的开放词表 3D 语义分割框架,通过基于不确定性的超点蒸馏、实例级掩码重建和跨实例关系一致性三个互补模块,利用预训练3D模型的几何先验来纠正 2D 到 3D 知识蒸馏中的几何偏差,在 ScanNet v2 上达到 64.8 mIoU 的 SOTA 性能。

研究背景与动机

  1. 领域现状:开放词表 3D 语义分割旨在分割训练集以外的任意类别。由于3D点-文本配对数据稀缺,主流方法通过从预训练的 2D 开放词表模型(如 CLIP、LSeg、OpenSeg)向 3D 模型蒸馏知识来实现。两种主要范式是"2D-to-3D 蒸馏"(通过几何对应将像素级特征投影到点云上对齐)和"点-文本对齐"(通过对比学习对齐 3D 特征与文本嵌入)。
  2. 现有痛点:两种范式本质上都训练 3D 模型复制 2D 模型的特征表示,存在两个核心问题:(a) 限制了内在 3D 几何学习 — 将 3D 特征对齐到 2D 表示空间,抑制了 3D 几何结构的学习;(b) 继承 2D 预测错误 — 2D 模型容易因遮挡和视角变化产生错误的物体掩码(如图1所示),3D 模型继承这些错误后学到了不正确的分割模式。
  3. 核心矛盾:如何在 2D-to-3D 知识蒸馏过程中有效保留内在的 3D 几何信息?直接融入预训练 3D 特征并不一定能改善性能,因为不同模态之间的异构监督信号会在训练中引入不稳定性。
  4. 本文目标 三个层面的几何-语义一致性问题:(a) 超点内一致性 — 同一超点内的点应共享语义标签,但 2D 投影常导致不一致;(b) 实例内一致性 — 单视角 2D 预测仅覆盖部分实例,导致 3D 空间中实例语义碎片化;(c) 跨实例关系一致性 — 多视角特征聚合引入同类实例间的特征分布漂移。
  5. 切入角度:预训练 3D 模型(如 Sonata)已经在大规模点云数据上学到了强几何先验——同类物体具有相似的几何表示。关键是如何在蒸馏过程中利用这些先验来纠正 2D 层面的错误。
  6. 核心 idea:利用预训练 3D 模型的几何先验,通过三层级(超点、实例、跨实例)的几何-语义一致性建模来引导 2D-to-3D 蒸馏过程,确保蒸馏出的 3D 特征既保留几何结构又具备开放词表语义能力。

方法详解

整体框架

给定场景点云 \(\mathbf{P} \in \mathbb{R}^{N \times 3}\) 和多视角 RGB 图像 \(\mathcal{I}\),GeoGuide 并行提取两类特征:(1) 冻结的预训练 3D 骨干提取几何特征 \(\mathbf{F}_{3d}^G \in \mathbb{R}^{N \times C_1}\);(2) 冻结的 2D 开放词表分割模型提取像素级语义特征 \(\mathbf{F}_{2d}^M\)。通过相机参数建立 2D-3D 对应关系,将 2D 特征投影到点云得到 \(\mathbf{F}_{2d} \in \mathbb{R}^{N \times C}\)。3D 几何特征通过轻量级 MLP 适配器映射到相同语义空间得到 \(\mathbf{F}_{3d}^{\text{sem}}\)。三个层次化模块(USD、IMR、IIRC)从局部到全局引导蒸馏过程。推理时仅需 3D 点云输入,丢弃所有辅助模块。

关键设计

  1. Uncertainty-based Superpoint Distillation (USD):

    • 功能:利用超点内的几何一致性促进语义连贯性,自适应抑制噪声 2D 特征
    • 核心思路:首先通过法线分割得到超点 \(\{Q_i\}_{i=1}^{N_Q}\)。对 3D 几何特征和 2D 语义特征分别在超点内做均值池化得到超点级特征。然后计算超点级与点级特征的差异,拼接后通过 MLP 预测每个点的 2D 特征可靠性权重:\(\mathcal{W} = \text{MLP}(\text{concat}[(\mathbf{S}_{3d}^G - \mathbf{F}_{3d}^G); (\mathbf{S}_{2d} - \mathbf{F}_{2d})])\)。用这些权重对超点内 2D 特征做加权池化得到精炼的超点语义特征 \(\overline{\mathbf{S}}_{2d}\)。最终在超点级和点级分别计算余弦相似度蒸馏损失 \(\mathcal{L}_{sp}\)
    • 设计动机:传统均值池化对预测噪声过于敏感,会放大偏差。通过引入 3D 几何信息估计不确定性权重,可以抑制错误的 2D 预测(如被遮挡或边界模糊区域的错误投影),保留有判别力的正确特征。
  2. Instance-level Mask Reconstruction (IMR):

    • 功能:在实例级别强制语义一致性,恢复每个实例的完整掩码
    • 核心思路:使用类别无关的 3D 实例分割方法获取实例掩码 \(\{M_i\}_{i=1}^{N_M}\)。对每个掩码随机遮挡一部分得到不完整掩码 \(\overline{M}_i\),从 \(\mathbf{F}_{3d}^{\text{sem}}\) 中索引对应点特征并池化,通过线性层得到掩码特征 \(\overline{\mathbf{F}}_i^{\text{mask}}\)。用该特征与全局 \(\mathbf{F}_{3d}^{\text{sem}}\) 计算余弦相似度来预测重建掩码:\(\hat{M}_i = \text{sigmoid}(\cos(\overline{\mathbf{F}}_i^{\text{mask}}, \mathbf{F}_{3d}^{\text{sem}}))\)。用 BCE 损失 \(\mathcal{L}_{\text{mask}}\) 约束重建结果与原始掩码一致。
    • 设计动机:超点通常只覆盖实例的局部区域。通过从部分掩码重建完整掩码,鼓励同一实例内的所有点学习相似的语义特征,实现实例级拓扑完整性。
  3. Inter-Instance Relation Consistency (IIRC):

    • 功能:对齐跨实例的语义关系与几何亲和度,缓解视角引起的语义漂移
    • 核心思路:将实例掩码内的 3D 几何特征和语义特征分别聚合为掩码级嵌入 \(\mathbf{F}_{\text{mask}}^G\)\(\mathbf{F}_{\text{mask}}^{\text{sem}}\)。计算成对相似度矩阵:\(\mathbf{P}_{\text{sim-m}}^G = \mathbf{F}_{\text{mask}}^G {\mathbf{F}_{\text{mask}}^G}^T\)\(\mathbf{P}_{\text{sim-m}}^{\text{sem}} = \mathbf{F}_{\text{mask}}^{\text{sem}} {\mathbf{F}_{\text{mask}}^{\text{sem}}}^T\)。类似地计算超点级的几何和语义相似度矩阵。用 MSE 损失对齐语义相似度与几何相似度:\(\mathcal{L}_{\text{sim}} = \text{MSE}(\mathbf{P}_{\text{sim-m}}^G, \mathbf{P}_{\text{sim-m}}^{\text{sem}}) + \text{MSE}(\mathbf{P}_{\text{sim-sp}}^G, \mathbf{P}_{\text{sim-sp}}^{\text{sem}})\)
    • 设计动机:预训练 3D 模型确保同类物体具有相似的几何表示,但这一先验在 2D-to-3D 蒸馏过程中不会自动保持。通过强制语义相似度矩阵与几何相似度矩阵对齐,防止了蒸馏过程中跨实例几何一致性信息的退化。

损失函数 / 训练策略

总损失:\(\mathcal{L}_{\text{final}} = \lambda_1 \mathcal{L}_{\text{sp}} + \lambda_2 \mathcal{L}_{\text{mask}} + \lambda_3 \mathcal{L}_{\text{sim}}\)

推理时仅需 3D 点云输入,使用适配器输出 \(\mathbf{F}_{3d}^{\text{sem}}\) 与 CLIP 文本嵌入计算相似度进行分类。三个辅助模块全部丢弃,不增加推理开销。

实验关键数据

主实验

开放词表 3D 语义分割(mIoU / mAcc):

方法 ScanNet v2 mIoU nuScenes mIoU Matterport3D mIoU
OpenScene (LS) 54.2 36.7 43.4
SAS (stage1) 59.2 45.4 46.3
SAS (stage2) 61.9 47.5 48.6
GeoGuide (SAS*) 64.8 50.3 51.9

GeoGuide 在 ScanNet v2 上超出 SAS(stage1) +5.6 mIoU,在 nuScenes 上超出 +4.9 mIoU。

长尾场景评估(Matterport3D):

方法 K=21 mIoU K=40 mIoU K=80 mIoU K=160 mIoU
OpenScene (OS) 41.1 33.4 18.1 8.9
DMA (OS) 45.1 37.9 19.7 9.4
GeoGuide (OS) 47.7 38.5 22.0 11.6

消融实验

模块 ScanNet v2 mIoU mAcc 说明
Full model 53.4 74.8 使用 OpenSeg 特征
w/o USD 下降 下降 超点内一致性丧失
w/o IMR 下降 下降 实例级一致性丧失
w/o IIRC 下降 下降 跨实例关系约束缺失

初步实验(验证框架设计动机):

方法 mIoU (OpenSeg) mAcc (OpenSeg)
OpenScene(训练整个3D网络) 47.5 70.7
冻结3D骨干+MLP适配器 50.4 75.2

冻结预训练 3D 骨干+轻量级适配器即可提升 +2.9 mIoU,但对不同 2D 模型效果不一致(LSeg 下略降),说明朴素蒸馏会破坏 3D 预训练学到的几何先验。

关键发现

  • 三个模块的互补性:USD 处理局部超点一致性、IMR 处理实例内完整性、IIRC 处理跨实例关系,分别从微观到宏观层面解决蒸馏偏差
  • 跨 2D 模型的鲁棒泛化:无论使用 LSeg、OpenSeg 还是 SAS 特征,GeoGuide 都能一致提升性能,说明方法解决的是本质的几何不一致问题而非依赖特定 2D 特征
  • mIoU 和 mAcc 同步提升:表明方法不仅改善了分割覆盖度,还提升了每类预测精度,几何一致性建模帮助网络学到了更判别性和类别特定的特征
  • 长尾场景下 GeoGuide 性能退化幅度最小,归功于 IIRC 模块维持了同类实例间的语义一致性

亮点与洞察

  • "保护几何先验"的核心洞察:不是让 3D 模型完全学 2D 特征,而是冻结预训练 3D 骨干、仅训练轻量适配器。同时通过三个模块在蒸馏过程中主动保护和利用几何先验。这种"保守蒸馏"策略值得在其他跨模态迁移场景中借鉴
  • 层次化一致性建模:从超点(局部)→ 实例(区域)→ 跨实例(全局)的三层一致性设计形成了完整的局部到全局语义对齐机制。每一层都有明确的动机和互补作用
  • 推理时零额外开销:所有三个辅助模块仅在训练时使用,推理时完全丢弃。这使得 GeoGuide 在推理效率上与 OpenScene 等基础方法完全相同,实用性很强
  • 不确定性引导的加权聚合思路可推广到任何需要融合多源可能有噪声特征的场景

局限与展望

  • 依赖类别无关实例分割质量:IMR 和 IIRC 模块的效果受3D实例分割(Mask3D等)的准确性限制
  • 未使用自蒸馏策略:SAS 等方法通过费时的自蒸馏可进一步提升性能,GeoGuide 未采用这一策略但仍超越了 SAS
  • 超点和实例的过分割/欠分割问题:超点的质量直接影响 USD 模块,在几何模糊区域可能出现不当分组
  • 跨域泛化:虽做了 ScanNet→Matterport3D 的跨域实验,但室内到室外的大跨度迁移仍有挑战
  • 可以探索更强的 3D 预训练模型(如 PointMAE v2)以获得更好的几何先验

相关工作与启发

  • vs OpenScene:OpenScene 直接训练整个 3D 网络对齐 2D 多视角特征,会破坏几何先验。GeoGuide 冻结 3D 骨干并利用几何先验引导蒸馏,在 ScanNet v2 上以 OpenSeg 特征超出 +5.9 mIoU
  • vs SAS:SAS 集成多个 2D 开放词表模型减少单一模型偏差,但仍忽略 3D 几何结构。GeoGuide 在使用相同 2D 特征时进一步提升 +5.6 mIoU(ScanNet v2),且不需要费时的自蒸馏
  • vs GGSD:GGSD 采用 mean-teacher 框架增强蒸馏,但仍缺乏显式几何约束。GeoGuide 通过层次化几何-语义一致性取得更全面的改善
  • 该工作启示:在跨模态蒸馏中,目标模态的预训练先验不应被丢弃,而应作为蒸馏过程的"教师"来使用

评分

  • 新颖性: ⭐⭐⭐⭐ 三层级几何一致性建模设计合理,但单个模块的技术新颖性有限
  • 实验充分度: ⭐⭐⭐⭐⭐ 三大基准+长尾+跨域评估,消融完整,初步实验验证动机充分
  • 写作质量: ⭐⭐⭐⭐ 动机推导清晰,模块设计逻辑连贯,图表信息量大
  • 价值: ⭐⭐⭐⭐ 在开放词表3D分割中引入层次化几何约束的方向有重要意义,不使用自蒸馏即超SOTA实用性强

相关论文