跳转至

General and Task-Oriented Video Segmentation

会议: ECCV 2024
arXiv: 2407.06540
代码: https://github.com/kagawa588/GvSeg (有)
领域: 图像分割
关键词: 视频分割, 通用框架, 任务导向, 查询匹配, 时序对比学习

一句话总结

GvSeg 提出了一个通用视频分割框架,通过将分割目标解耦为外观、形状和位置三个因素,并根据任务需求(VIS/VSS/VPS/EVS)动态调整这三个因素在查询初始化、匹配和采样中的参与度,在统一架构下实现了四种视频分割任务的SOTA性能。

研究背景与动机

视频分割是计算机视觉的基础任务,包含四个主要子任务:视频实例分割(VIS)、视频语义分割(VSS)、视频全景分割(VPS)和样例引导视频分割(EVS)。目前业界有两种趋势:一是为每个子任务设计专用模型,效果好但冗余且无法跨任务迁移;二是追求通用解决方案,但现有通用框架(如 Video K-Net、TubeFormer、Tube-Link)高度同质化——对所有任务采用相同的查询初始化、匹配和时空学习策略,忽视了不同任务之间的内在差异,导致次优性能。

核心矛盾:通用性与任务特异性之间的权衡。既要保持统一架构,又要适应不同任务的特性(如 VIS 强调实例区分,VSS 强调语义理解)。

切入角度:重新思考分割目标的构成要素——外观(appearance)、形状(shape)、位置(position),并根据不同任务需求灵活调整这三者的参与程度。

核心idea:在不修改网络结构的前提下,通过形状-位置描述子(Shape-Position Descriptor)和任务导向策略实现对不同视频分割任务的自适应。

方法详解

整体框架

GvSeg 采用半在线的 query-based 视频分割范式,基于 Mask2Former 构建。输入为包含三帧的视频片段,通过 Transformer 编码器-解码器提取帧级查询,再通过跨帧查询匹配实现目标关联。关键创新在于三个架构无关的设计:Shape-Position Descriptor、任务导向的查询初始化与关联策略、任务导向的时序对比学习。

关键设计

  1. 形状-位置描述子(Shape-Position Descriptor)

功能:为每个目标物体构建一个同时编码形状和相对空间位置信息的紧凑描述子。

为什么:现有方法仅依赖外观特征(appearance)进行跨帧匹配,忽略了形状和位置线索。在低光照、遮挡等场景下,外观特征不可靠,形状和位置信息能提供关键补充。

怎么做:受 Shape Context 启发,从目标轮廓上均匀采样 M=200 个锚点,转换到以目标中心为原点的极坐标系,将极坐标空间划分为 u×v = 36×12 个 bin,统计每个 bin 中锚点数量得到直方图 H。此外,对中心点落在 mask 外的 bin 设为负值 −1/√d_model,使描述子不仅编码形状还编码位置。

关键公式:H_ij 统计极坐标 bin 中锚点的数量,形状相似但位置不同的物体(如 B 和 C)通过负值区分布不同来区分。

与之前方法的区别:不同于仅用外观特征做匹配的方法,GvSeg 的 SPA 查询匹配将描述子 H 注入 Transformer 解码器的 SelfAttn 前(类似 APE 编码),并在跨帧匹配时将 H 叠加到查询嵌入上:S_ij = cosine(q̂_i^t + H_i^t, q̂_j^{t+1} + H_j^{t+1})。

  1. 任务导向的查询初始化与目标关联

功能:针对四种任务的特性,分别设计不同的查询初始化和关联策略。

为什么:VIS 需要强实例区分(强调形状/位置),VSS 需要强语义理解(不应过度依赖形状/位置),VPS 需要兼顾两者,EVS 需要利用给定的标注提示。统一策略无法同时满足。

怎么做: - EVS:从标注提示(点/框/掩码)对应的骨干特征采样初始化查询,应用 SPA 匹配增强实例区分。 - VIS:将骨干特征划分为 S×S 网格,随机选取 N 个元素初始化查询,编码位置和外观信息,应用 SPA 匹配。 - VSS:利用训练过程中动量更新的语义类查询 q̄ 初始化,不使用 SPA 匹配(形状/位置会损害语义泛化)。 - VPS:组合 VIS(thing)和 VSS(stuff)策略。

与之前方法的区别:之前的通用方案(如 Video K-Net)要么用统一策略处理所有任务,要么通过修改架构(如双路径设计)来适应不同任务。GvSeg 在不改变架构的前提下,仅通过调整三要素的参与度实现任务适应。

  1. 任务导向的时序对比学习(Task-Oriented TCL)

功能:改进跨帧的正负样本采样策略。

为什么:现有 TCL 仅从时间邻域选参考帧,忽略远距帧,导致正/负样本数量有限。同时,远距帧中同一实例可能形状/位置变化较大,不适合作正样本。

怎么做: - 实例类任务(VIS/EVS/VPS-thing):通过计算形状-位置描述子变化量 ΔH = ‖H^{t+n} − H^t‖_2 / ‖H^t‖_2,设置阈值 τ=0.2 决定远距帧样本是正是负。ΔH < τ 为正样本,否则为负样本。 - 语义类任务(VSS/VPS-stuff):维护 FIFO 队列(每类 N_Q=100 个查询),从整个训练集采样,丰富语义描述。

损失函数 / 训练策略

  • 遵循 Mask2Former 的标准训练损失(包含匈牙利匹配、mask 损失、分类损失)
  • 时序对比学习损失用于跨帧查询嵌入的对比学习
  • 训练迭代数:OVIS/VSPW/VIPSeg/KITTI 为 10K,YouTube-VOS18/YouTube-VIS21 为 15K
  • 优化器:AdamW,初始 lr=0.001,step 衰减
  • 数据增强:翻转、随机缩放和裁剪
  • 对 YouTube 系列数据集使用 MS COCO 伪视频预训练

实验关键数据

主实验

数据集 指标 本文 (R-50) 之前SOTA 提升
VIPSeg (VPS) VPQ 44.0 39.2 (Tube-Link) +4.8
VIPSeg (VPS) STQ 44.9 39.5 (Tube-Link) +5.4
VSPW (VSS) mIoU 44.5 43.4 (Tube-Link) +1.1
Occluded-VIS AP 35.9 31.1 (TarVIS) +4.8
YouTube-VIS21 AP 49.6 48.3 (TarVIS) +1.3
YouTube-VOS18 (EVS) G 81.5 79.2 (TarVIS) +2.3
BURST (EVS) H_all 35.9 32.3 (XMem) +3.6

使用更强的 Swin-L 骨干时,GvSeg 进一步刷新了所有数据集的 SOTA。

消融实验

配置 VPQ STQ 说明
Baseline 36.0 37.3 无任何改进
+ SPA query matching 37.8 38.5 形状-位置感知匹配有效
+ Task-oriented init & asso. 40.1 40.7 任务导向初始化与关联
+ Task-oriented TCL 41.2 42.0 任务导向对比学习
GvSeg (全部) 44.0 44.9 各组件互补

关键发现

  • 形状-位置描述子为跨帧匹配带来了 1.8% VPQ 的提升,验证了形状/位置线索的重要性
  • 任务导向的查询初始化最关键,贡献了最大的性能增益
  • 时序对比学习中,ΔH 阈值 τ=0.2 和队列长度 N_Q=100 已接近最优,且几乎不增加训练开销
  • 在遮挡严重的 OVIS 数据集上提升尤为明显(+4.8% AP),说明形状/位置描述子有效应对遮挡

亮点与洞察

  • 解耦思想精妙:将分割目标解耦为外观/形状/位置三个相互正交的因素,既能独立建模又能灵活组合,是一个优美的设计理念
  • 架构无关:所有创新设计均不修改网络结构,保持统一架构处理四种任务,实际部署简单
  • 对任务特性的深刻理解:VSS 不使用 SPA 匹配、VPS 组合 thing/stuff 策略等设计体现了对各任务本质差异的准确把握
  • 形状-位置描述子的双重用途:不仅用于增强查询匹配,还用于指导 TCL 中的采样决策

局限与展望

  • 形状-位置描述子依赖于准确的 mask 预测,在首帧预测质量差时可能引入噪声
  • 极坐标直方图的 bin 数量(36×12)是手工设计的超参,可能不是所有场景的最优选择
  • 半在线模式需要逐帧推理,长视频处理效率有限
  • 仅验证了 Mask2Former 作为基础分割器,未探索与其他架构(如 SAM)的结合
  • TCL 中正负样本阈值 τ=0.2 对不同数据集可能需要调整

相关工作与启发

  • Mask2Former:GvSeg 的基础分割器,证明了 query-based 范式在多任务中的潜力
  • Shape Context:形状-位置描述子的灵感来源,将传统形状描述方法巧妙迁移到深度学习query中
  • Tube-Link:之前最强的通用视频分割方案,被 GvSeg 大幅超越
  • 启发:形状+位置的解耦建模可推广到其他视觉跟踪/匹配任务;任务自适应策略可启发多任务学习中的任务特化设计

评分

  • 新颖性: ⭐⭐⭐⭐ 将分割目标解耦为三因素并实现任务自适应是有新意的,但查询匹配和对比学习本身不算全新
  • 实验充分度: ⭐⭐⭐⭐⭐ 四种任务、七个数据集、全面的消融和可视化,极为充分
  • 写作质量: ⭐⭐⭐⭐ 逻辑清晰,但公式较多,部分细节需要仔细阅读
  • 价值: ⭐⭐⭐⭐⭐ 统一框架在所有任务上全面超越专用和通用方案,实用价值极高

相关论文