基于协同引导和协同融合的稳定半监督遥感图像语义分割 Toward Stable Semi-Supervised Remote Sensing Segmentation via Co-Guidance and Co-Fusion
通过融合CLIP和DINOv3的异构先验解决半监督遥感分割中的伪标签漂移问题
前置知识
伪标签漂移
半监督学习中,模型通过自生成的伪标签训练无标注数据时,由于确认偏差导致错误预测被不断强化和累积的现象。当初始预测不可靠时,模型会持续强化错误模式,导致训练过程逐渐偏离正确方向,最终陷入局部最优或完全崩溃。
本文的核心问题就是解决伪标签漂移,理解这个概念才能明白为什么要引入异构基础模型和协同引导机制
确认偏差
机器学习中模型倾向于相信并强化自身已有预测的倾向,即使这些预测是错误的。在半监督学习中,如果模型一开始做出了错误预测,它会用这些错误预测作为监督信号训练自己,导致错误不断放大。
确认偏差是导致伪标签漂移的根本原因,理解它才能理解为什么需要异构双学生架构来打破自我强化的恶性循环
弱到强一致性
半监督学习的核心思想,通过对同一输入施加弱增强(如简单几何变换)和强增强(如色彩抖动、高斯模糊、CutMix等),强制模型对强增强视图的预测与对弱增强视图的高置信度预测保持一致。
这是本文采用的基础训练范式,理解它才能理解一致性损失 $L_{ct}$ 的设计原理和作用
视觉基础模型
在大规模数据上预训练的通用视觉模型,包括视觉语言模型(如CLIP通过图文对齐学习可迁移的视觉概念)和自监督模型(如DINOv3通过自蒸馏学习高质量密集特征)。这些模型蕴含丰富的先验知识,可以直接迁移到下游任务。
本文的创新核心就是利用CLIP和DINOv3这两个异构基础模型的互补先验来稳定半监督训练,理解它们的特点是理解本文方法的关键
异构协同训练
使用两个或多个不同架构或不同初始化的模型同时训练,让它们相互监督。与传统的同构协同训练(相同架构不同初始化)不同,异构模型具有不同的表征偏好和错误模式,可以更有效地纠正彼此的错误。
本文的异构双学生架构是异构协同训练的具体实现,理解它才能理解为什么CLIP和DINOv3的组合能比同构配置更有效
研究动机
现有的半监督遥感图像语义分割方法存在严重的伪标签漂移问题。在标注极端稀缺的场景下(如1/24或1/40的标注比例),传统方法如FixMatch和UniMatch经常在训练早期就陷入确认偏差,导致错误预测被不断强化。例如,在WHDLD数据集的1/24划分实验中,监督基线的伪标签准确率在前10个epoch内剧烈波动,在50%到90%之间震荡,表现出极端的不稳定性。标准方法如FixMatch和UniMatch则表现出早期饱和现象,迅速收敛到次优水平(约80%-88%),表明它们在训练早期就被确认偏差困住,无法纠正初始错误。由于缺乏强有力的外部指导来纠正错误,这些方法在区分语义相似类别或在严重标注稀缺下精确定位物体边界时经常遇到困难。在遥感场景中,由于类别高度相似(如不同类型的路面或植被)、物体尺度变化极大、场景布局复杂,这个问题尤为突出。
本文的目标是本文旨在构建一个稳定的半监督遥感图像语义分割框架,能够在标注严重稀缺的条件下保持训练稳定性和分割精度。具体目标包括:1)有效缓解伪标签漂移问题,防止错误预测的累积和强化;2)利用异构视觉基础模型的互补先验知识,建立可靠的互学习过程;3)在多个遥感数据集上实现领先的性能,特别是在标注比例极低的极端场景下;4)同时保证语义一致性和边界精度,生成高质量的分割结果。
与已有工作不同的是,现有工作主要采用同构协同训练架构(如Dual-Student)或标准教师-学生范式,这些方法虽然试图通过多模型相互监督来缓解确认偏差,但由于模型架构和初始化的相似性,往往收敛到相似的错误模式,无法有效打破自我强化的恶性循环。本文的独特切入角度是利用异构视觉基础模型(CLIP和DINOv3)的互补先验特性——CLIP提供全局语义理解,DINOv3提供局部结构细节——构建异构双学生架构。更重要的是,本文首次在半监督学习框架中系统性地融合了视觉语言模型和自监督模型的先验,提出了显式-隐式语义协同引导机制和全局-局部特征协同融合策略,为解决伪标签漂移问题提供了全新的思路。与那些针对全监督或弱监督场景的融合方法(如Talk2DINO、CLIP-DINOiser)不同,本文专注于半监督学习中的互学习稳定性问题,这是现有工作未曾探索的方向。
核心方法
Co2S框架采用异构双学生半监督分割流水线,利用CLIP和DINOv3两个不同的ViT-B/16基础模型的互补先验来缓解伪标签漂移。两个学生并行处理相同的增强视图并输出分割logits。对于标注数据,每个学生都由真实标注监督。对于无标注数据,每个学生通过自身增强预测进行弱到强一致性正则化,同时引入全局-局部协同融合机制,基于像素级置信度在两个学生之间执行相互监督,形成稳定性约束来抑制伪标签漂移。框架的核心创新包括两个组件:显式-隐式语义协同引导机制,通过文本嵌入和可学习查询分别为CLIP学生和DINOv3学生提供语义指导;以及全局-局部特征协同融合策略,基于置信度仲裁来融合CLIP的全局上下文信息和DINOv3的局部细节。这些组件共同建立了一个稳定的互学习过程,确保即使在标注严重稀缺的情况下也能保持卓越的分割性能。
核心创新点在于构建异构双学生架构并引入显式-隐式语义协同引导和全局-局部特征协同融合两个机制。与现有同构协同训练方法不同,Co2S利用CLIP和DINOv3的固有差异特性——CLIP擅长全局语义理解,其注意力图呈现扩散式激活模式;而DINOv3擅长局部结构捕获,其注意力图呈现锐利响应——来打破确认偏差。显式语义引导通过冻结的CLIP文本编码器提供的文本嵌入,为CLIP学生提供明确的类别级语义指导;隐式语义引导通过可学习查询为DINOv3学生提供灵活的类别级语义表示。全局-局部特征协同融合策略基于像素级置信度动态仲裁两个学生的预测,在两者都高置信度时进行对称互学习,只有一个高置信度时低置信度学生向高置信度同伴学习,两者都低置信度时忽略该像素。这种设计确保了训练过程的稳定性和监督信号的可靠性。
方法步骤详情
方法流程分为三个主要步骤。首先是显式-隐式语义协同引导:对于CLIP学生,采用概念基础的提示集成策略,为每个语义类别 $k$ 精心构建一组细粒度概念描述 $C_k = \{c_{k,i}\}_{i=1}^{S_k}$(例如对于道路类别:{highway, main road, street, ...}),通过提示模板函数 $T(\cdot)$ 格式化后由冻结的CLIP文本编码器 $E_{txt}$ 编码,通过平均聚合得到类别原型 $t_k = \frac{1}{S_k}\sum_{i=1}^{S_k} E_{txt}(T(c_{k,i}))$,最终堆叠所有类别的原型得到显式文本查询矩阵 $F_t^{clip} = [t_1, \ldots, t_N]$。对于DINOv3学生,利用可学习查询 $Q = [q_1, \ldots, q_N]$ 提供隐式类别级指导,这些查询是作为可训练的嵌入层实现的,与DINOv3主干和分割解码器一起优化。两个学生都采用同构但参数独立的查询基础解码器,通过点积相似度计算分割logits:$Z_{global} = Sim(F_t^{clip}, F_{global}^{clip})$ 和 $Z_{local} = Sim(Q, F_{local}^{dino})$。其次是全局-局部特征协同融合:给定无标注图像 $x_u$ 的弱增强视图,通过两个学生得到全局logits $Z_{global}$ 和局部logits $Z_{local}$,计算置信度图 $C_{global}^{pred}$ 和 $C_{local}^{pred}$,基于三种置信度场景构建交互掩码:1)两者都高于阈值且相等时对称互学习,2)只有一方高于阈值时低置信度学生向高置信度同伴学习,3)两者都低于阈值时忽略该像素。稳定性损失定义为 $L_{sta}(t) = \lambda(t) \cdot MSE(p_\ell, p_h) \cdot 1(C \geq \tau)$,其中 $p_\ell$ 和 $p_h$ 是低置信度和高置信度学生的类别概率向量,$\tau$ 是置信度阈值,$\lambda(t)$ 是随训练步骤的线性增长权重。最后是总训练目标:对于标注样本 $(x_l, y_l)$,生成弱增强视图 $x_w^l$ 输入两个学生,得到 $Z_{global}$ 和 $Z_{local}$,监督目标为 $L_{global}^{sup} = CE(Z_{global}, y_l)$ 和 $L_{local}^{sup} = CE(Z_{local}, y_l)$。对于无标注样本 $x_u$,严格遵循UniMatch的双流扰动范式,生成弱增强视图 $x_w^u$、两个强增强视图 $x_{s1}^u$ 和 $x_{s2}^u$、以及一个特征扰动视图 $x_{fp}^u$。对于每个像素位置 $(i,j)$,分配伪标签 $\hat{y}_{u,i,j} = \arg\max_n(p_{w,i,j}^n)$ 和置信度 $c_{i,j} = \max_n(p_{w,i,j}^n)$。定义通用掩码交叉熵函数 $H(Z, \hat{y}_u, c) = \sum_{i,j} 1(c_{i,j} \geq \tau) \cdot CE(Z_{:,i,j}, \hat{y}_{u,i,j})$,一致性损失为 $L_{ct} = \frac{1}{2}H(Z_{fp}, \hat{y}_u, c) + \frac{1}{4}H(Z_{s1}, \hat{y}_u, c) + H(Z_{s2}, \hat{y}_u, c)$。每个学生的总训练目标为 $L = \lambda_{sup}L_{sup} + \lambda_{ct}L_{ct} + \lambda_{sta}L_{sta}$,其中 $\lambda_{sup} = \lambda_{ct} = 0.5$,$\lambda_{sta}$ 采用余弦坡度调度。
技术新颖性
本文的技术新颖性体现在多个方面。首先,首次在半监督遥感分割框架中系统性地融合了视觉语言模型(CLIP)和自监督模型(DINOv3)的先验知识,开创了异构基础模型协同用于半监督学习的先例。其次,提出了显式-隐式语义协同引导机制,通过文本嵌入提供明确的类别级语义指导,同时通过可学习查询捕获数据特定的视觉特征,这两者的协同作用确保了语义正确性和数据适应性的平衡。第三,设计了全局-局部特征协同融合策略,基于像素级置信度动态仲裁两个学生的预测,这种设计不仅融合了CLIP的全局语义和DINOv3的局部细节,还通过置信度门控机制过滤了低质量监督信号,有效抑制了伪标签漂移。最后,在架构设计上,两个学生采用同构但参数独立的查询基础解码器,这种设计保持了架构统一性的同时允许不同语义源的注入,是一个巧妙的技术选择。与现有工作相比,本文方法的创新性不仅在于组件设计,更在于将这些组件有机整合到一个完整的训练框架中,形成了从初始化、优化到特征融合的全流程创新。
实验结果
在六个广泛使用的遥感分割基准上的实验结果一致表明Co2S实现了领先性能,特别是在标注极其稀缺的条件下表现优异。在WHDLD数据集的极端稀疏1/24划分下,Co2S以61.1%的mIoU超过强竞争对手UniMatch 3.7%,展示了准确解析多样化土地覆盖特征的能力。在LoveDA数据集的1/40比例下,该方法相比监督基线实现了12.3%的显著提升,在面对显著的城市-农村分布偏移时表现出卓越性能。对于超高分辨率的Potsdam数据集,Co2S在1/32划分下以74.3%的mIoU获得最佳性能,超越了DWL等遥感专用设计方法。扩展到大规模GID-15数据集,该方法在1/8划分下超过UniMatch 1.5%,表明在广阔空间区域处理多样化土地覆盖特征的熟练程度。此外,在MER和MSL数据集的地外环境中,Co2S在这些以极端类别不平衡为特征的基准上实现了最高精度。值得注意的是,在所有基准上观察到一个一致趋势:随着标注比例的降低,Co2S相对于基线方法的性能优势会放大。这一现象验证了Co2S的稳定性,确认协同基础模型先验有效地防止了模型崩溃,即使在最低数据标注制度下也能维持高精度。伪标签质量分析显示,Co2S在第一个epoch内就达到超过95%的伪标签准确率,并以最小的方差维持这一优越水平,而标准方法如FixMatch和UniMatch表现出早期饱和,快速收敛到约80%-88%的次优水平。这表明Co2S从开始就成功建立了高质量的监督信号,从而有效地从一开始就扼杀了伪标签漂移,确保了可靠的优化过程。消融实验结果表明,仅使用显式语义引导(文本嵌入)就能在基线58.97 mIoU的基础上提升1.8%达到60.77 mIoU,这是由于CLIP在预训练期间获得的文本-图像对齐能力使文本嵌入能够立即作为强大可靠的语义锚点。仅使用隐式语义引导(可学习查询)反而导致轻微性能下降到58.86 mIoU,这是因为可学习查询是随机初始化的参数缺乏预训练先验,而半监督设置中的稀缺标注样本无法提供足够的监督使其有效收敛。只有当两种机制都参与时才能达到61.09 mIoU的最佳性能,这表明文本嵌入确保语义正确性,而可学习查询捕获数据特定的视觉特征,两者协同增强语义一致性。异构双学生设计的消融实验表明,两个学生都使用DINOv3的同构配置表现很差,仅获得45.20 mIoU,这表明没有强大的全局语义先验,仅初始化为局部结构特征的学生难以维持类别一致性,导致严重的语义漂移。两个学生都使用CLIP的同构配置由于丰富的全局语义表示达到竞争性的60.78 mIoU,但相同的初始化导致高度相关的预测,这种缺乏互补观点的设计阻止了学生纠正彼此的错误,使它们容易受到耦合错误积累和确认偏差的影响。异构架构结合CLIP和DINOv3达到最高的61.09 mIoU性能,这一优势确认框架成功地协同了CLIP的全局语义表示和DINOv3的局部结构细节。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| WHDLD土地覆盖分类 | mIoU(%) | 61.1 | UniMatch 57.4 | +3.7 |
| LoveDA城乡分割 | mIoU(%) | 58.2 | 监督基线 45.9 | +12.3 |
| Potsdam城市解析 | mIoU(%) | 74.3 | DWL 74.2 | +0.1 |
| GID-15大规模分类 | mIoU(%) | 75.4 | UniMatch 73.9 | +1.5 |
| MER火星地表分析 | mIoU(%) | 56.8 | UniMatch 56.0 | +0.8 |
| MSL火星地表分析 | mIoU(%) | 60.9 | DWL 59.0 | +1.9 |
局限与改进
作者承认了几点局限性。首先,CLIP和DINOv3都采用ViT-B/16架构,计算开销相对较大,在资源受限的环境下部署可能存在挑战。其次,当前方法专注于2D图像分割,对于3D点云或视频等遥感数据格式的适用性尚未探索。第三,文本提示工程(概念描述的构建)需要一定的领域知识,对于新数据集的适配需要人工设计合适的概念集合。此外,从观察来看,虽然Co2S在低标注比例下表现优异,但在高标注比例(如1/2或全监督)下的优势相对较小,这表明随着标注数据的增加,基础模型先验的重要性逐渐降低。另一个潜在的局限性是当前的置信度阈值 $\tau$ 是固定设置的(0.95),虽然实验证明其有效性,但对于不同数据集可能需要自适应调整。最后,当前的异构组合仅探索了CLIP和DINOv3这一对搭配,与其他基础模型(如MAE、BEiT、iBOT等)的组合效果可能需要进一步探索。
独立分析的弱点
尽管Co2S取得了优异的性能,但仍存在一些可改进的弱点。首先是计算效率问题:采用两个ViT-B/16骨干网络导致推理和训练开销约为单一模型的两倍,这对于实时应用或大规模部署可能是一个瓶颈。改进方向包括引入知识蒸馏将异构双学生的知识压缩到单一模型,或者探索更高效的轻量级基础模型组合。其次是文本提示工程的人工依赖:当前方法需要为每个类别精心设计细粒度概念描述,这需要领域专家知识且对于新数据集可能需要重新设计。改进方向可以引入自动提示学习,通过可学习的提示模板或大语言模型自动生成概念描述。第三是置信度阈值的固定设置:当前方法使用统一的置信度阈值0.95,但不同数据集、不同类别可能需要不同的阈值设置。改进方向可以是类别自适应阈值或动态阈值调度策略。第四是当前方法主要关注2D图像,对于多光谱、高光谱、SAR等遥感数据类型的适用性尚未验证,未来的改进可以扩展到多模态融合。最后是当前实验主要在光学遥感图像上验证,对于极端天气条件(云、雾、阴影)的鲁棒性还需要进一步验证。
未来方向
作者提出的未来研究方向包括探索更多样化的基础模型组合,如将CLIP与其他自监督模型(如MAE、BEiT、iBOT、SimMIM)配对,以验证协同效应的普遍性。另一个方向是扩展到其他遥感任务,如目标检测、变化检测、实例分割等,验证异构基础模型先验的有效性。基于研究成果的可延伸方向包括:1)将Co2S框架扩展到多模态遥感数据,融合光学图像、SAR、激光雷达等异构数据源;2)研究开放词汇半监督分割,利用CLIP的零样本能力处理训练时未出现的类别;3)探索动态架构搜索,自动寻找最优的基础模型组合;4)引入主动学习,通过不确定性采样选择最有价值的无标注样本进行人工标注,进一步提高半监督效率;5)研究跨域半监督学习,利用源域的基础模型先验辅助目标域的半监督训练;6)探索时空一致性约束,将时间序列信息引入半监督遥感分割,提高时间序列遥感图像的分割稳定性。
复现评估
论文提供了项目页面(https://xavierjiezou.github.io/Co2S/),这表明作者计划开源代码和模型,有利于复现。实现细节方面,论文明确描述了优化器设置(AdamW,权重衰减0.01,多项式学习率衰减调度,power=0.9)、数据增强协议(严格遵循UniMatch)、训练时长(80个epoch,MER数据集150个epoch)、置信度阈值(0.95)、损失权重($\lambda_{sup} = \lambda_{ct} = 0.5$,$\lambda_{sta}$ 采用余弦坡度调度)等关键超参数。实验环境为配备NVIDIA RTX 3090 GPU的工作站,这个硬件配置相对常见,研究者可以访问到。评估指标为标准mIoU,这避免了指标选择的主观性。使用的六个数据集(WHDLD、LoveDA、Potsdam、GID-15、MER、MSL)都是公开可用的,且论文提供了详细的数据集划分说明(不同的标注比例如1/24、1/16、1/8、1/4等)。总体来说,复现难度中等,主要挑战在于两个ViT-B/16模型同时训练的计算需求较高,需要较强的GPU资源。如果作者能够提供预训练权重和完整代码,复现应该相对直接。需要注意的是,CLIP和DINOv3的预训练模型需要从官方源获取,这可能导致版本差异影响复现结果。
论文图表
该表格展示了不同方法在WHDLD数据集上在不同标注比例(1/24、1/16、1/8、1/4)下的分割结果(mIoU)。Co2S在所有标注比例下都取得了最佳性能,特别是在极端稀疏的1/24划分下达到61.1% mIoU,超过强竞争对手UniMatch 3.7%。
这个表格对理解论文重要,因为它提供了在不同标注比例下的详细性能比较,证明了Co2S在标注稀缺场景下的优势,并且展示了随着标注比例降低,优势放大的趋势。
该表格展示了不同方法在LoveDA数据集上在不同标注比例(1/40、1/16、1/8、1/4)下的分割结果。Co2S在所有标注比例下都取得了最佳性能,特别是在1/40比例下达到58.2% mIoU,相比监督基线45.9%提升了12.3%。
这个表格对理解论文重要,因为它展示了Co2S在城乡场景转换的复杂数据集上的优势,特别是在极低标注比例(1/40)下的显著提升,证明了方法的鲁棒性。
该表格展示了不同方法在Potsdam超高分辨率数据集上在不同标注比例(1/32、1/16、1/8、1/4)下的分割结果。Co2S在1/32划分下以74.3% mIoU获得最佳性能,超越了DWL等遥感专用设计方法。
这个表格对理解论文重要,因为它验证了Co2S在超高分辨率航空影像上的有效性,这是遥感分割的一个重要应用场景。
该表格展示了不同方法在MER和MSL两个火星地表数据集上在不同标注比例(1/8、1/4)下的分割结果。Co2S在这两个数据集的所有标注比例下都取得了最佳性能,MER的1/8划分下达到56.8% mIoU,MSL的1/8划分下达到60.9% mIoU。
这个表格对理解论文重要,因为它展示了Co2S在地外环境(火星地表)这种极端类别不平衡场景下的优势,证明了方法的泛化能力不限于地球遥感数据。
该表格展示了不同方法在GID-15大规模数据集上在不同标注比例(1/8、1/4)下的分割结果。Co2S在所有标注比例下都取得了最佳性能,在1/8划分下达到75.4% mIoU,超过UniMatch 1.5%。
这个表格对理解论文重要,因为它验证了Co2S在大规模数据集(15个细粒度类别)上的有效性,展示了方法处理类别复杂性的能力。
该表格展示了显式和隐式语义指导机制的消融研究。基线模型(无指导)达到58.97 mIoU。仅使用隐式类别级指导(可学习查询)导致轻微性能下降到58.86 mIoU。仅使用显式类别级指导(文本嵌入)获得1.8%的显著提升,达到60.77 mIoU。当两种机制都参与时达到61.09 mIoU的最佳性能。
这个表格对理解论文重要,因为它验证了显式-隐式语义协同引导机制的有效性,证明了文本嵌入和可学习查询的协同作用。
该表格展示了双学生架构异构性的消融研究。两个学生都使用DINOv3的同构配置(Local + Local)仅获得45.20 mIoU。两个学生都使用CLIP的同构配置(Global + Global)达到60.78 mIoU。异构架构结合CLIP和DINOv3(Global + Local)达到最高的61.09 mIoU性能。
这个表格对理解论文重要,因为它验证了异构双学生设计的必要性,证明了CLIP的全局语义和DINOv3的局部细节的互补性是取得最佳性能的关键。
该表格展示了不同优化目标贡献的消融研究。仅使用监督损失 $L_{sup}$ 的基线达到59.98 mIoU。加入一致性损失 $L_{ct}$ 提升到60.73 mIoU。进一步加入稳定性损失 $L_{sta}$ 达到最佳性能61.09 mIoU。
这个表格对理解论文重要,因为它验证了每个损失项的作用,特别是证明了稳定性损失 $L_{sta}$ 对缓解伪标签漂移的重要性。
该表格展示了CLIP与不同自监督模型先验协同的消融研究。CLIP与自身的同构基线达到60.78 mIoU。将CLIP与任何考察的自监督模型(MAE、BEiT、iBOT、SimMIM、DINOv3)配对都能获得相对于该基线的性能提升。DINOv3带来了最大的0.31提升。
这个表格对理解论文重要,它验证了不同自监督先验与CLIP的协同效应,证明了互补视觉先验能够增强相互语义交互学习,超越CLIP单独能达到的性能。