Boosting Text-Driven Video Segmentation via Geometry-Aware Distillation¶
会议: ECCV2026
官方论文: 4230
PDF: https://media.eventhosts.cc/Conferences/ECCV2026/pdfs/5017.pdf
作者: Tianyu Zhu, Yingping Liang, Hesong Li, Ying Fu
代码: https://github.com/Tony1882880/GeoLaV
领域: 模型压缩(知识蒸馏;指代表达视频目标分割)
关键词: 指代表达视频分割、单目几何预训练、几何感知蒸馏、新视角合成、双教师监督
一句话总结¶
GeoLaV 在 SAMWISE 分割框架上先用单张图像合成连续新视角进行预训练,再在真实视频上蒸馏语义与三维几何教师,使 Ref-YouTube-VOS 的 J&F 达到 70.5,相比同为 202M 参数的 SAMWISE 提高 1.3 分,且不在推理时保留教师或投影头。
研究背景与动机¶
指代表达视频目标分割(RVOS)要求模型根据一句自然语言,在整段视频中持续输出目标掩码。难点不只是第一帧认出物体,还在于镜头变化、遮挡和目标运动后仍然知道“描述指的是谁”。SAMWISE 等方法已经把 SAM2 的分割先验、文本融合与时序记忆结合起来,但其图像预训练通常来自 RefCOCO/+/g:这些数据能教会模型把描述对应到单帧区域,却不能直接提供同一场景在不同视角下的空间对应关系。
把静态图像平移、缩放成多帧可以增加训练输入,却没有根据深度改变物体之间的视差和遮挡。另一方面,真实视频微调通常依靠掩码损失学习时序关系,正确分割当前帧并不等于理解跨帧空间结构。本文因而同时检查两个环节:预训练时是否看到了具有合理几何变化的连续视图,以及微调时是否获得了超出二维掩码的结构监督。
论文不重新设计一个大型语言模型,而是在已有轻量视频分割架构中改变训练数据与辅助监督。核心 idea:先把单图监督扩展成几何一致的多视图学习,再把语义教师和三维几何教师的知识共同压入记忆特征,让部署模型沿用原有分割路径,却更能适应视角变化。
方法详解¶
整体框架¶
输入是一段 RGB 视频及描述目标的句子,输出是每帧的二值目标掩码。GeoLaV 沿用 SAMWISE 的主体结构,以 SAM2 图像编码器、文本编码器、跨模态适配器、记忆机制和掩码解码器完成语言引导分割;新增内容主要位于训练侧,而不是推理时额外重建三维场景。
训练分为单目几何预训练(MGP)和几何感知蒸馏(GAD)两个阶段。MGP 先通过“几何一致视图合成”把单图变成伪视频,再用“语义记忆对齐”学习这些视图的表征;GAD 从该初始化出发,在真实视频上执行“几何语义双教师蒸馏”。图中的三项设计依次对应数据构造、第一阶段监督和第二阶段监督。
%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
A["单图、文本与掩码"] --> B["几何一致视图合成"]
B --> C["语义记忆对齐<br/>MGP 预训练"]
C --> D["几何语义双教师蒸馏<br/>GAD 微调"]
V["真实视频、文本与掩码"] --> D
D --> E["部署分割模型<br/>视频与文本到逐帧掩码"]
关键设计¶
1. 几何一致视图合成:让单图扩展具有真实投影关系,而不只是二维扰动
第一步用前馈几何模型预测单图的深度或点图,论文列举了 π3 与 VGGT 作为可选模型。由于单目深度存在尺度歧义,生成过程对预测深度加入随机缩放和偏置,再依据采样的相机内参将像素提升到三维空间。这里需要的是与原图结构相容的相对深度,而不是精确到物理单位的测距:近处与远处的点在相机移动时应当产生不同的投影变化。
随后,模型沿连续的虚拟相机轨迹移动这些三维点并重新投影,轨迹可以是直线、分段直线或曲线,相邻位姿来自小幅刚体变换,而不是逐帧独立采样。新视角暴露出的空洞由遮挡掩码标出,并交给图像修复网络补齐。实现中,每张 COCO 源图生成五张新视图,再与原图组成伪视频。这样得到的帧序列包含视差和深度顺序变化,比把整张图片当成平面进行单应变换更接近相机运动;但它并没有模拟物体自身的真实非刚性运动,修复区域也不能等同于真实观测。
2. 语义记忆对齐:用稠密教师特征补足合成帧的监督
有了多帧输入,还需要让分割网络从中学习,而不能只把它当作重复图像。伪视频经过冻结的 SAM2 图像编码器,文本经过冻结的文本编码器;各层插入可训练的跨模态适配器,使两种模态在共享表征中交互。融合后的视觉特征进入记忆注意力,并结合提示编码器的输出完成掩码解码。掩码与图像信息经记忆编码器写入记忆库,供后续帧再次读取。上述路径主要继承 SAMWISE/SAM2,本文利用的是已经带有历史上下文的记忆特征,而不是另建独立的追踪分支。
并行的视觉基础模型教师为伪视频提供语义特征,实际采用 DINOv3-ViT-L。学生的记忆注意力特征经过带 GELU 的两层 MLP 投影,以匹配教师特征维度,再使用余弦距离对齐。按原文式(8),核心约束为:
其中,投影后的学生特征是 \(F_{\mathrm{proj}}\),教师输出是 \(F_{\mathrm{VFM}}\)。对齐位置在提示条件化的掩码解码之前,因此不是拿教师直接指定“这句话对应哪个目标”,而是约束具有全局帧信息的中间表征;目标身份仍由文本分割监督决定。这一点也解释了为什么使用不依赖特定指代表达的视觉教师:它补充的是帧级结构与语义,而不是替代语言定位。
MGP 将该辅助损失与分割损失共同优化。论文明确指出,特征监督能弥补合成帧监督不足,但当前全文未清楚交代新视图掩码如何生成、哪些像素参与分割损失,因而不能进一步断言所有合成视图都有完整的人工级掩码标注。该细节是复现时需要核实的部分,而不是笔记自行补出的处理步骤。
3. 几何语义双教师蒸馏:在真实视频上保留语义,同时补入三维结构先验
第二阶段加载 MGP 得到的模型,改用带文本与掩码的真实视频训练。除了视觉基础模型教师,新增冻结的三维感知教师,实验采用 π3;前者提供物体外观与语义边界,后者提供跨帧结构和深度关系。学生仍使用同一条记忆与分割路径,不把几何教师预测的深度或相机参数直接当成最终分割结果。
两位教师的特征空间不同,因此分别配置一个两层 MLP 投影头,从同一学生记忆表征出发,各自对齐对应教师。两个分支都采用余弦相似度损失,蒸馏目标是两项之和,并与文本条件分割监督一起训练。分开投影避免强迫不同维度、不同语义的教师共享一个输出空间,也允许同一记忆表征同时保留语义边界和三维结构信息。
这种监督并不保证学生显式恢复正确三维几何,更准确的说法是用教师表征作为结构正则约束。推理时教师与投影头均移除,真实视频与文本直接经过训练后的分割模型,保持原有的逐帧预测与记忆更新流程。因此“无额外推理开销”针对这些训练专用分支,并不意味着合成数据或训练教师没有成本。
损失函数 / 训练策略¶
第一阶段的总目标是分割损失与上述相似度损失相加;分割损失包含 Dice 与 mask loss,沿用既有方法。第二阶段保留分割监督,并加入语义和几何两个分支的蒸馏损失。缓存中的式(9)与式(10)排版损坏,但相邻正文明确支持两个独立投影头和两个余弦损失;这里不猜测未明确给出的额外权重或空间归约细节。
预训练在 RefCOCO/+/g 上进行 6 个 epoch,学习率从 \(1\times10^{-4}\) 降到 \(2\times10^{-5}\)。随后在 Ref-YouTube-VOS 上以 \(2\times10^{-6}\) 微调 4 个 epoch,用所得模型评估 Ref-YouTube-VOS 和 Ref-DAVIS17;MeViS 按文中协议训练 2 个 epoch。优化器为 Adam,实验使用 4 张 80GB NVIDIA A100。论文将图像编码器、文本编码器与记忆注意力描述为冻结组件,跨模态适配器、投影头及部分提示参数可训练;其余细粒度参数开关仍需结合实现核实。
实验关键数据¶
主实验¶
区域相似度 \(\mathcal{J}\) 衡量掩码区域重叠,轮廓准确率 \(\mathcal{F}\) 衡量边界匹配,J&F 是二者的平均值,均越高越好。下表保留原文表 1 的报告值;Ref-YouTube-VOS 与 MeViS 通过官方服务器评测,Ref-DAVIS17 使用官方代码。各基线主要引用各自论文,并非统一重训,因此应在各自训练协议的背景下比较。
| 方法 | 总参数量 | Ref-YouTube-VOS J&F ↑ | Ref-DAVIS17 J&F ↑ | MeViS J&F ↑ |
|---|---|---|---|---|
| SAMWISE | 202M | 69.2 | 70.6 | 49.5 |
| ReferDINO | 230M | 69.3 | 68.9 | 49.3 |
| GeoLaV | 202M | 70.5 | 72.5 | 50.0 |
| GLUS-A(大型 VLM,单独参照) | 7B | 67.3 | 未报告 | 51.3 |
相对同参数量 SAMWISE,三个数据集分别提高 1.3、1.9、0.5 分;相对表中此前最好的非大型 VLM 方法,增益分别为 1.2、1.9、0.5 分。这里都是分数差,不是相对百分比。GeoLaV 在该非大型 VLM 组领先,但 MeViS 上低于大型 VLM 方法 GLUS-A 的 51.3,不能把结论扩大成全部方法无条件最优。
原文表 2 还单独测量“只在图像上训练、直接评测视频”的泛化能力,不能与上表的视频微调结果混为一谈。
| 图像预训练后直接评测 | Ref-YouTube-VOS J&F ↑ | MeViS J&F ↑ |
|---|---|---|
| SAMWISE | 31.9 | 26.4 |
| GeoLaV | 47.0 | 31.6 |
| 绝对增益 | 15.1 | 5.2 |
消融实验¶
以下为原文表 3 的 J&F 列,所有配置都先进行图像预训练,再进行视频微调。表题没有单独明确数据集名称,因此保留其“组件消融”口径,不擅自把它写成多个基准的平均分。
| 配置 | 第一阶段 | 第二阶段 | J&F ↑ |
|---|---|---|---|
| Vanilla Model | 基础预训练 | 基础微调 | 65.1 |
| 平面增强 | 二维单应变换 | 基础微调 | 66.5 |
| 仅 MGP | MGP | 基础微调 | 67.2 |
| 仅 GAD | 基础预训练 | GAD | 67.5 |
| 完整 GeoLaV | MGP | GAD | 70.5 |
相对该表的 Vanilla Model,MGP 单独带来 2.1 分,GAD 单独带来 2.4 分,联合带来 5.4 分。MGP 比平面增强高 0.7 分,支持收益不只是来自“多看几帧”;但 MGP 同时包含数据合成和语义对齐,本表不能独立量化每一项的贡献。此外,表 3 的 Vanilla Model 为 65.1,而主表 SAMWISE 为 69.2,不能将两者当作完全相同的基线。原表部分 J、F 分项与所列均值也不完全吻合,本笔记保留其报告的 J&F,不自行改表。
关键发现¶
- 最明显的提升出现在图像训练到视频测试的迁移:Ref-YouTube-VOS 从 31.9 升到 47.0,表明预训练输入的跨视角结构值得关注;这不是“完全没有任何预训练”的零样本设置。
- 组件实验中联合配置比单独 MGP 高 3.3 分,比单独 GAD 高 3.0 分,符合两阶段互补的解释,但论文没有多次运行方差来检验差异稳定性。
- 运动描述更重要的 MeViS 上,完整训练后相对同规模 SAMWISE 的提升为 0.5 分,弱于另外两个基准;几何先验不等于已经解决复杂运动语言理解。
亮点与洞察¶
- 把单图数据的价值从“更多静态监督”推进到“可控视角变化”。关键不在帧数本身,而在深度、相机轨迹和遮挡关系共同决定新帧。
- 对齐的是记忆注意力后的共享特征,而非最终目标掩码。教师可以提供通用视觉结构,文本分割损失继续负责指代对象,二者职责清晰。
- 双教师用于训练而不随模型部署。这使结构监督可以增强已有视频分割器,而不必把完整三维模型纳入在线处理路径。
局限与展望¶
- 单图合成依赖估计深度和修复结果,不能恢复原图中不可见区域的真实内容,也不能替代真实动态视频。原文未来方向是开放词汇与更大规模的三维感知多模态分割。
- 当前全文缺少合成掩码监督细节、教师成本拆分与充分的分支消融;尚不能把 MGP 的收益全部归于几何重投影,或认定几何教师一定比其他稠密教师更有效。
- 证据以标准分割指标和定性特征可视化为主,没有独立的定量几何一致性指标。加上部分表格数值与基线口径不够清晰,结论宜限定为所报告协议下的分割改善。
相关工作与启发¶
- 与 SAMWISE 对比:SAMWISE 提供 SAM2、文本适配与记忆融合的基础架构,GeoLaV 主要增加两阶段几何训练信号。应把贡献理解为训练范式增强,而不是全新分割骨干。
- 与 L2M 对比:L2M 为单图提升到三维后学习稠密匹配提供先例,本文将类似视图构造用于 RVOS 预训练,并进一步连接真实视频上的教师蒸馏。
- 与 GLUS 对比:GLUS 使用大型语言模型的全局与局部推理;GeoLaV 选择较小的分割模型配合训练期教师,在多数所列指标上有竞争力,但 MeViS 并未超过 GLUS-A。
评分¶
- 新颖性: 3/5 — 单图几何增强与双教师蒸馏的组合贴合 RVOS,单个技术部件已有先例。
- 实验充分度: 3/5 — 有三个基准、图像到视频泛化与阶段消融,但监督细节、统计稳定性和部分数字口径仍需澄清。
- 写作质量: 3/5 — 两阶段动机容易理解,当前全文的公式提取和若干表格存在可读性或一致性问题。
- 价值: 4/5 — 展示了增强训练监督而不保留额外推理分支的实用途径,尤其适合关注图像到视频迁移的读者。