← 返回 2026-03-31

INSID3:基于 DINOv3 的训练无关上下文分割 INSID3: Training-Free In-Context Segmentation with DINOv3

Claudia Cuttano, Gabriele Trivigno, Christoph Reich, Daniel Cremers, Carlo Masone, Stefan Roth 📅 2026-03-30 👍 7 2026-07-13 08:36
DINOv3 上下文分割 自监督学习 语义对应 零样本学习

仅用 DINOv3 特征实现训练无关的 one-shot 语义/部分/个性化分割,超越现有方法 7.5% mIoU

前置知识

上下文分割

上下文分割是一种 one-shot 学习任务,给定一个标注的参考示例(如图像中标记的目标区域),模型需要在新的目标图像中分割出相同概念的所有实例。概念可以是语义类别(如所有狗)、物体部分(如狗的耳朵)或个性化实例(如我的狗)。这类似于大语言模型通过上下文示例适应新任务,将这种能力迁移到计算机视觉领域。

本文的核心任务是上下文分割,理解这个概念是理解问题定义、实验设置和方法动机的基础。

视觉基础模型

视觉基础模型是在大规模数据上预训练的通用视觉模型,如 DINOv2、DINOv3、SAM、CLIP 等。这些模型通过自监督或弱监督学习获得丰富的视觉表示,可以迁移到下游任务而无需大量标注数据。DINOv3 是一个纯自监督的模型,通过局部一致性约束和 Gram anchoring 目标学习密集的局部化特征,具有强大的语义对应和空间结构保持能力。

DINOv3 是本文方法的核心支柱,理解其特性和与传统 VFMs 的差异(特别是位置偏差问题)是理解本文方法的关键。

层次聚类

层次聚类是一种自底向上的聚类算法,从每个数据点开始作为独立簇,然后逐步合并最相似的簇,直到满足停止条件。与 K-means 需要预先指定簇数不同,层次聚类通过相似度阈值 τ 控制粒度,更适应开放世界的可变粒度需求。在 INSID3 中,用于将目标图像的密集特征分解为语义一致的区域候选。

层次聚类是本文方法的第一步,用于构建图像的结构化表示,理解其原理和参数 τ 的作用是理解后续步骤的基础。

语义对应

语义对应旨在建立不同图像中语义相似区域之间的密集像素级映射关系。例如,给定两张包含不同品种狗的图像,语义对应任务要求找到一张图中狗的鼻子在另一张图中的对应位置。这通常通过计算特征相似度、匹配关键点或学习对应场来实现。可靠的语义对应是上下文分割的核心,因为它需要将参考概念定位到目标图像中。

语义对应是上下文分割的基础组件,本文揭示的位置偏差问题直接影响语义对应的质量,因此理解这个概念对于理解本文的动机和贡献至关重要。

位置编码

位置编码是 Transformer 架构中注入空间位置信息的机制,使模型能够区分不同位置的 token。常见的位置编码包括正弦编码(如原始 Transformer)、可学习编码等。在视觉 Transformer 中,位置编码可以是 2D 的(分离编码 h 和 w)或 1D 的(将图像展平为序列)。DINOv3 的训练目标(特别是 Gram anchoring)可能无意中增强了绝对空间相关性,导致特征在语义信号较弱时被位置信息主导。

本文揭示的位置偏差问题本质上与位置编码和训练目标的交互有关,理解这个概念有助于理解为什么 DINOv3 会产生系统性位置匹配错误以及如何去除它。

研究动机

现有上下文分割方法面临两类根本问题。第一类是任务特定的微调方法(如 SegIC 在 DINOv2 上训练分割解码器,DiffewS 微调 Stable Diffusion),它们在训练分布内表现出色(如 SegIC 在 COCO-20i 上达到 76.1% mIoU),但当测试分布偏离训练分布时性能急剧下降(SegIC 在 LVIS-92i 上降至 44.6%,在 iSAID 上降至 39.9%)。这是因为微调将模型耦合到特定的训练数据分布,限制了其对未见域和粒度的灵活性。第二类是训练无关方法(如 Matcher、GF-SAM),它们通过组合多个预训练组件(DINOv2 用于语义对应 + SAM 用于生成掩码)来避免微调的陷阱。虽然这些方法泛化能力更强(GF-SAM 平均 47.6% mIoU),但它们需要协调多个 VFMs,增加显著计算开销(GF-SAM 需要 1,030ms 推理时间,945M 参数),且无法充分利用对应与分割之间的内在协同。

本文的目标是本文的核心目标是证明:一个单一的自监督视觉基础模型是否足以同时支持语义匹配和分割,而无需任何监督或辅助模型?具体而言,作者希望:(1) 展示分割能力可以直接从 DINOv3 的密集自监督表示中涌现,而不需要任何解码器、微调或模型组合;(2) 揭示并解决 DINOv3 特征在跨图像匹配中的系统性位置偏差问题;(3) 实现跨语义、部分和个性化分割的最强泛化性能,同时保持架构的简洁性和效率。

与已有工作不同的是,本文的独特切入角度是从极简主义视角重新审视上下文分割。与需要额外监督或模型组合的现有方法不同,本文提出一个关键问题:能否利用 DINOv3 的两个核心优势——强大的语义对应能力和密集的局部化特征结构——在单一 backbone 内直接实现上下文分割?这个视角与现有工作形成鲜明对比:微调方法将 DINOv2 视为需要增强特征提取器,SAM-based 方法将 DINOv2 和 SAM 视为互补组件,而本文将 DINOv3 视为一个已经蕴含分割能力的完整系统。此外,本文还发现了前人未注意到的 DINOv3 位置偏差问题,这是一个全新的研究视角,其解决方案不仅改进了本文方法,还泛化到语义对应等其他任务。

核心方法

INSID3 的整体思路是充分利用 DINOv3 特征的双重性质:跨图像的语义对应能力和图像内的空间自相似性。方法分为四个核心步骤。首先,通过分析发现 DINOv3 特征存在系统性位置偏差(相似位置的特征即使语义无关也会错误匹配),作者提出用噪声图像估计位置子空间并投影到正交补空间来去除偏差。第二步,使用层次聚类将目标图像分解为语义一致的候选区域,这些区域自然形成从物体到部分的不同粒度。第三步,通过反向对应和跨图像相似度从候选区域中选择与参考最匹配的种子区域,这里去偏后的特征用于跨图像匹配以避免位置偏差干扰,而原始特征用于图像内聚类以保持空间结构。第四步,将种子区域与其他候选区域聚合,聚合得分结合了跨图像语义对应和图像内结构一致性两个信号,最终生成空间连贯的完整分割掩码。整个方法完全无需训练,仅需三个超参数(聚类阈值 τ=0.6、聚合阈值 α=0.2、去偏秩 s=500),在所有数据集和任务上保持一致。

INSID3 的核心创新点在于三个层面:(1) 首次证明自监督 VFM 本身足以实现训练无关的上下文分割,打破了之前认为必须依赖监督预训练(SAM)或任务特定微调的假设;(2) 发现并系统分析了 DINOv3 特征的位置偏差问题,提出了简单但有效的训练无关修正方案(从噪声图像估计位置子空间并投影去除);(3) 设计了一个 principled 的聚合机制,同时利用跨图像语义对应和图像内自相似性两个互补信号,解决了单靠语义对应在遮挡或视角变化下不可靠的问题。与现有方法的本质区别在于:INSID3 将对应和分割统一在单一特征空间内,而 Matcher/GF-SAM 将它们解耦到不同模型(DINOv2 + SAM)并通过额外步骤协调;INSID3 完全无需训练,而 SegIC/DiffewS 需要大量监督信号。

方法步骤详情

INSID3 的完整流程包含四个步骤。步骤1:位置去偏。首先生成一个噪声图像 I_noise ∼ N(0, 1) 并通过 DINOv3 编码器获得特征 F_noise = Φ(I_noise)。对 F_noise 进行奇异值分解 F_noise = UΣV⊤,选取前 s=500 个右奇异向量 B = V[:, 1:s] 作为位置子空间的基。然后将参考和目标特征投影到位置子空间的正交补空间:F̃_r = F_r(1_D - BB⊤),F̃_t = F_t(1_D - BB⊤)。这样就去除了位置偏差,得到去偏后的特征用于跨图像匹配。步骤2:细粒度聚类。对原始目标特征 F_t 使用层次聚类,通过阈值 τ=0.6 控制粒度,将图像分解为 K 个不相交的语义一致区域 {G₁, ..., G_K},满足 ∪_k G_k = Ω 且 G_i ∩ G_j = ∅ ∀i ≠ j。步骤3:种子聚类选择。首先通过反向对应定位候选:对每个目标 patch i,找到其在去偏参考特征中的最近邻 NN(i) = arg max_j ⟨F̃_ti, F̃_rj⟩,保留最近邻落在参考掩码内的目标 patch 作为候选集 C_NN = {i | M_r(NN(i)) = 1}。然后将与 C_NN 有交集的聚类作为候选聚类 C_cand = {G_k | G_k ∩ C_NN ≠ ∅}。对每个候选聚类 G_k 和参考区域 R 计算原型:p̃_tk = (1/|G_k|) Σ_{i∈G_k} F̃_ti,p̃_r = (1/|R|) Σ_{j∈R} F̃_rj。计算跨图像相似度 s^cross_k = ⟨p̃_tk, p̃_r⟩,选择种子聚类 G* = arg max_{G_k∈C_cand} s^cross_k。步骤4:聚类聚合。首先计算每个聚类在原始特征空间的自身相似度:p̄_tk = (1/|G_k|) Σ_{i∈G_k} F_ti,s^intra_k = ⟨p̄_tk, p̄_t*⟩。然后组合跨图像和图像内相似度得到最终得分 S_k = s^cross_k · s^intra_k,选择满足 S_k ≥ α(α=0.2)的聚类与种子聚类合并:M_final = G* ∪ {G_k ∈ C_cand | S_k ≥ α}。最后应用 CRF 进行掩码细化。

技术新颖性

INSID3 的技术新颖性体现在多个方面。首先,在方法论上,它是第一个完全无需训练的上下文分割方法,仅依赖自监督的 DINOv3 特征,打破了现有方法依赖监督预训练(SAM)或任务特定微调的范式。其次,本文发现的位置偏差问题是全新的观察:作者通过可视化相似度图和 PCA 分析揭示,DINOv3 特征在不同图像中存在系统性的位置对齐激活(如图 4 所示,左侧参考的特征在目标图像左侧也会产生高响应),这在 DINOv2 中几乎不存在。作者提出的去偏方案通过从噪声图像估计位置子空间并投影去除,不仅简单高效(仅需一次矩阵乘法),而且泛化到语义对应等其他任务(在 SPair-71k 上提升高达 6.6 PCK)。第三,INSID3 的聚合机制巧妙结合了跨图像语义对应和图像内自相似性两个信号,通过乘性得分 S_k = s^cross_k · s^intra_k 同时要求聚类与参考语义对齐且与种子结构一致,这比单独使用任一信号都更鲁棒。最后,在工程实现上,INSID3 仅需三个超参数且在所有任务上保持一致,展现了方法的稳定性和简洁性。

Region-level grouping from DINOv3.
Figure 2: Region-level grouping from DINOv3.
Overview of INSID3.
Figure 3: Overview of INSID3.
Positional bias in DINOv3 features.
Figure 4: Positional bias in DINOv3 features.

实验结果

INSID3 在三个分割任务上均取得了显著的性能提升。在 one-shot 语义分割任务中,INSID3 在 6 个数据集上平均达到 55.1% mIoU,超过最强的训练无关基线 GF-SAM (+7.5% pts.) 且参数量少 3 倍(304M vs 945M)。具体而言,在 LVIS-92i 上达到 41.8% vs GF-SAM 的 35.2%(+6.6% pts.),在 ISIC 上达到 54.9% vs GF-SAM 的 53.1%(+1.8% pts.),在 SUIM 上达到 52.1% vs GF-SAM 的 47.1%(+5.0% pts.),在 Chest X-Ray 上达到 78.8% vs GF-SAM 的 51.0%(+27.8% pts.)。与微调方法相比,INSID3 在分布内(如 COCO-20i)略逊于 SegIC(57.6% vs 76.1%),但在分布外数据集(如 LVIS-92i、iSAID、X-Ray)上显著超越(41.8% vs 44.6% on LVIS,78.8% vs 34.5% on X-Ray),展示了更强的泛化能力。在 one-shot 部分分割任务中,INSID3 在 PASCAL-Part 上达到 50.5% mIoU,超过 GF-SAM 的 44.5%(+6.0% pts.)和 SegIC 的 39.9%(+10.6% pts.);在 PACO-Part 上达到 38.7% mIoU,超过 GF-SAM 的 36.3%(+2.4% pts.)和 SegIC 的 25.9%(+12.8% pts.)。在 one-shot 个性化分割任务中,INSID3 在 PerMIS 上达到 67.0% mIoU,超过 GF-SAM 的 54.1%(+12.9% pts.)、SegIC 的 51.8%(+15.2% pts.)和 DiffewS 的 35.2%(+31.8% pts.)。去偏策略的效果在语义对应任务 SPair-71k 上也得到了验证:DINOv3-L 原始特征达到 50.6% PCK@0.10,去偏后提升到 52.0%(+1.4% pts.),在 PCK@0.20 上从 66.4% 提升到 68.6%(+2.2% pts.)。消融实验表明去偏、聚类和聚合都是必要的:去偏在 COCO-20i 上提升 +3.1% mIoU,聚类(τ=0.6)提供结构化表示,聚合(α=0.2)结合跨图像和图像内相似度比单独使用任一信号更好(57.6% vs 54.6% using only cross-similarity on COCO)。5-shot 实验显示 INSID3 能有效利用多个参考示例,平均达到 59.9% mIoU,超过 GF-SAM 的 52.8%(+7.1% pts.)。计算效率方面,INSID3 在 RTX 4090 上单次推理仅需 302ms,远快于 GF-SAM 的 1,030ms,主要耗时在层次聚类(166ms)和 CRF(55ms)。空掩码实验显示当目标图像不包含参考概念时,INSID3 正确预测空掩码的比例为 85%,优于 SegIC 的 36% 和 SegGPT 的 79%。

Comparison of INSID3 (mIoU in %, ↑) on one-shot semantic, part, and personalized segmentation.
Table 1: Comparison of INSID3 (mIoU in %, ↑) on one-shot semantic, part, and personalized segmentation.
Semantic correspondence on SPair-71k (PCK@T in %, ↑). Comparison across DINOv3 backbones, w/ and w/o debiasing.
Table 2: Semantic correspondence on SPair-71k (PCK@T in %, ↑). Comparison across DINOv3 backbones, w/ and w/o debiasing.
Effect of clustering and aggregation. We compare our approach with tuned baselines on COCO-20i and PASCAL-Part (mIoU %, ↑).
Table 3: Effect of clustering and aggregation. We compare our approach with tuned baselines on COCO-20i and PASCAL-Part (mIoU %, ↑).
Comparison of INSID3 with GF-SAM [69] and SegIC [41] on one-shot semantic (left), part (top right), and personalized (bottom right) segmentation.
Figure 5: Comparison of INSID3 with GF-SAM [69] and SegIC [41] on one-shot semantic (left), part (top right), and personalized (bottom right) segmentation.
查看结构化数据
任务指标本文基线提升
语义分割 mIoU (%) 55.1 (avg across LVIS-92i, COCO-20i, ISIC, SUIM, iSAID, X-Ray) GF-SAM: 47.6% +7.5%
部分分割 mIoU (%) 44.6 (avg across PASCAL-Part, PACO-Part) GF-SAM: 40.4% +4.2%
个性化分割 mIoU (%) 67.0 (PerMIS) GF-SAM: 54.1% +12.9%
语义对应 PCK@0.20 (%) 68.6 (DINOv3-L debiased on SPair-71k) 66.4 (DINOv3-L original) +2.2%
推理速度 Runtime (ms) 302 (INSID3 on RTX 4090) 1,030 (GF-SAM) 3.4× faster

局限与改进

作者在论文附录 F 中讨论了几个局限性:(1) INSID3 目前一次只能处理一个目标概念,当目标图像包含多个参考概念时需要分别提供参考提示;(2) INSID3 依赖掩码来定义目标概念,而像 SAM 这样的模型可以使用更轻量的空间标注(点、边界框),这些标注形式通常更便宜和快速收集;(3) 本文专注于语义级的上下文分割,即将同一概念的实例合并为一个掩码,而不是分别分割不同实例,这在需要实例级推理的场景中不适用;(4) INSID3 的最终分割质量仍然依赖于底层骨干网络(DINOv3)的表示能力,虽然去偏改善了跨图像匹配,但如果 DINOv3 在某些语义类别上本身表示能力弱,INSID3 也难以弥补。此外,从本文的观察中还可以补充一些局限性:(5) 层次聚类的复杂度是 O(P²),其中 P 是 patch 数量,虽然在高分辨率图像上仍然可行(1,760×1,760 分辨率仅需 1,102ms),但可能成为极端高分辨率图像的瓶颈;(6) 方法仅利用单个参考示例的特征原型,对于类别内变化大的场景可能不够鲁棒,虽然 5-shot 实验显示可以扩展但需要多次推理;(7) 去偏策略假设位置子空间是稳定的,这在实验中得到了验证(使用 1、5 或 10 个噪声图像得到几乎相同的结果),但如果位置子空间在不同域间有显著差异,可能需要域自适应的去偏。

独立分析的弱点

INSID3 的第一个弱点是在多个概念同时存在的场景下需要多次推理,增加了计算开销。一个改进方向是设计多概念的联合推理机制,例如为每个概念选择一个种子聚类,然后通过竞争性聚合来同时分割多个概念,避免重复计算聚类和相似度。第二个弱点是依赖掩码提示,而点或边界框提示更轻量。改进方向是探索如何将稀疏提示(点)映射到密集特征表示,例如使用注意力机制从点位置扩展到区域,或者学习一个轻量的提示编码器将点/框转换为特征原型。第三个弱点是语义级而非实例级分割。改进方向是利用同一实例内聚类之间的强自相似性(文中提到的观察),设计迭代机制:先选择一个种子聚类并聚合形成第一个实例掩码,然后从剩余区域重新选择种子,重复直到没有满足条件的聚类,从而分离同一语义的不同实例。第四个弱点是去偏方法可能在不同域间泛化受限。改进方向是开发域自适应的去偏,例如为不同域估计不同的位置子空间,或者使用在线学习的方法动态调整去偏投影矩阵。第五个弱点是层次聚类的计算复杂度。改进方向是探索更高效的聚类算法,如近似层次聚类、基于图的快速社区发现,或者利用 DINOv3 特征的局部性设计层级分解算法(从粗到细)。第六个弱点是单参考示例对类别内大变化的鲁棒性不足。改进方向是设计更鲁棒的原型构造方法,例如使用多个原型的混合、注意力加权或对抗性采样来覆盖类别内变化。最后,方法的性能仍受限于 DINOv3 骨干网络的表示能力。改进方向包括探索其他自监督 VFMs(如未来的 DINOv4)、多模态特征融合(如结合文本-图像特征)或通过无监督的表示精炼来增强 DINOv3 在特定域的表示能力。

未来方向

作者提出的未来工作方向包括:(1) 扩展到多概念同时分割,能够在单次推理中处理多个参考概念;(2) 支持更轻量的提示模态(点、边界框),降低标注成本并支持交互式使用;(3) 实现实例级推理,分别分割同一语义的不同实例。基于本文成果的可延伸方向包括:(4) 将去偏策略应用到其他依赖跨图像特征匹配的任务,如零样本检测、跨域图像配准、视频目标跟踪等;(5) 探索如何将 INSID3 的框架扩展到视频场景,利用时序一致性进一步改善分割质量;(6) 研究如何将 INSID3 与生成模型结合,例如利用分割结果引导图像编辑或生成;(7) 探索在无监督场景发现中的应用,即不提供任何参考,让模型自动发现图像中的语义区域(这可能需要修改聚合机制,自动选择有意义的种子聚类);(8) 研究如何将 INSID3 应用于 3D 场景理解,例如将 2D 分割提升到 3D 物体分割;(9) 探索与主动学习的结合,即让模型主动请求最有利于改进分割的参考示例;(10) 研究如何将 INSID3 的原理应用到其他模态(如音频、点云)的上下文分割任务。作者还提到,未来自监督表示的进展(如更强的局部化特征、更好的跨图像对应能力)可以进一步增强这个范式。

复现评估

INSID3 的复现难度较低。作者在论文中提供了详细的实现细节,包括所有超参数(τ=0.6, α=0.2, s=500)在所有任务上保持一致,无需数据集或任务特定调整。方法的核心组件(去偏投影、层次聚类、相似度计算、聚合)都是标准算法,实现相对直接。计算资源方面,INSID3 主要依赖 DINOv3-L 骨干网络(300M+ 参数),在单张 RTX 4090 上运行,推理时间约 302ms。训练不需要,因为方法完全训练无关。数据方面,评估使用了公开数据集(LVIS-92i、COCO-20i、ISIC、Chest X-Ray、iSAID、SUIM、PASCAL-Part、PACO-Part、PerMIS、SPair-71k),都可以公开获取。作者提到项目主页(https://visinf.github.io/INSID3)会提供代码和模型,这将进一步降低复现难度。潜在的复现挑战包括:(1) DINOv3-L 需要大量显存(作者使用 32-bit 精度,但可以用 half-precision 优化);(2) 层次聚类的具体实现可能影响性能(作者提到使用 GPU-accelerated 版本);(3) CRF 细化步骤需要合适的参数。总体而言,只要按照论文描述的设置实现,应该能够复现报告的结果。