VidEoMT:你的ViT其实也是一个视频分割模型 VidEoMT: Your ViT is Secretly Also a Video Segmentation Model
通过查询传播和融合,将视频分割统一到纯ViT编码器中,实现10倍加速且精度相当
前置知识
Vision Transformer (ViT)
Vision Transformer是一种将图像分割为固定大小的patch(通常16x16),通过线性映射将其转换为token序列,然后用标准Transformer编码器处理的视觉模型。ViT不依赖卷积操作,而是通过自注意力机制捕获全局依赖关系。DINOv2是基于ViT架构的视觉基础模型,通过大规模自监督预训练获得了强大的视觉特征表示能力,其特征具有跨视角一致性——同一物体在不同视角下产生相似的特征表示。这种特性对于视频分割中的跨帧物体追踪至关重要。
本文的核心主张是DINOv2预训练的ViT编码器可以直接承担视频分割任务,无需额外的解码器或追踪模块。理解ViT的工作原理和DINOv2的预训练特性是理解本文假设和实验验证的关键前提。
Mask Transformer / Mask2Former
Mask Transformer是一种基于查询(query)的图像分割范式。它使用一组可学习的查询向量,通过Transformer解码器与图像特征进行交叉注意力交互,每个查询预测一个分割掩码和对应的类别标签。Mask2Former是这一范式的代表性实现,由ViT-Adapter提供多尺度特征,像素解码器处理特征,以及Transformer解码器更新查询。该架构在图像和视频分割领域被广泛采用,是CAVIS、DVIS++等当前最先进方法的基础组件。
本文的关键创新之一是将Mask2Former中复杂的解码器和适配器组件替换为轻量级的编码器内处理,理解原始Mask Transformer架构有助于理解这一简化的意义。
Video Instance Segmentation (VIS)
视频实例分割是一项综合任务,要求模型在视频的每一帧中同时完成三个子任务:(1)检测并分割出每个物体实例的像素级掩码;(2)对每个物体进行语义类别分类;(3)在不同帧之间追踪同一物体,保持其身份标识的一致性。这要求模型具备定位、分类和时序关联三种能力。在线(online)设置要求在时间步t时,预测只能依赖当前帧和之前的帧,不能看到未来帧。本文提出的VidEoMT正是针对这一在线设置的通用方法,同时适用于VIS、VPS和VSS三种视频分割任务。
本文的目标是在VIS/VPS/VSS三个视频分割子任务上用统一的编码器架构实现竞争性精度,理解VIS的评估标准(AP、AR等)和在线约束是评估实验结果的基础。
EoMT (Encoder-only Mask Transformer)
EoMT是由Kerssies等人提出的一种极简图像分割方法。它直接将可学习的查询向量注入到大型预训练ViT编码器的最后几层(通常是最后L2=4层),与patch token一起联合处理。经过编码器处理后,更新后的查询可以直接用于预测分割掩码和类别标签,无需任何额外的Transformer解码器、像素解码器或适配器。尽管结构极简,EoMT仍能达到与复杂Mask Transformer架构相当的分割精度。EoMT证明了大容量预训练ViT编码器自身就具备了定位和分类物体的能力。
EoMT是本文VidEoMT方法的直接灵感来源和基础架构。本文在此基础上进一步证明ViT编码器不仅能做图像分割,还能做跨帧物体追踪。
DINOv2 / 视觉基础模型 (VFM)
DINOv2是Meta发布的视觉基础模型,采用ViT架构,在约1.42亿张图像上进行自监督预训练。其训练目标促使模型学习跨视角一致的特征表示——即同一物体在不同图像或视角下产生相似的特征向量。这种跨视角一致性对于视频追踪至关重要,因为它允许模型通过特征匹配来识别不同帧中的同一物体。视觉基础模型(VFM)通常指经过大规模预训练的通用视觉模型,可以通过微调适应各种下游任务。
本文的核心假设就是DINOv2预训练赋予了ViT编码器足够的能力来承担视频分割中的追踪功能,从而可以移除复杂的专用追踪模块。预训练质量和规模直接决定了本文方法的有效性。
查询传播 (Query Propagation)
查询传播是一种轻量级的时序建模机制。在第一帧中,模型使用可学习查询进行分割;在后续帧中,模型将前一帧输出的物体查询直接作为输入注入当前帧的ViT编码器,替代可学习查询。由于每帧的输出查询顺序是固定的(同一物体总是由同一个查询索引表示),这种传播机制使得信息可以跨帧流动,无需额外的计算开销。然而,纯查询传播会导致可学习查询的影响力随时间减弱,使模型难以识别新出现的物体。
查询传播是VidEoMT实现编码器内时序建模的核心机制之一,它使得模型能够在不增加计算成本的情况下实现跨帧物体关联。
研究动机
当前最先进的在线视频分割模型(如CAVIS、DVIS++、DVIS-DAQ)都遵循一个共同的范式:将视频分割分解为两个独立的组件——一个负责逐帧生成分割掩码和类别标签的分割器(segmenter),以及一个负责跨帧关联物体身份的追踪器(tracker)。这两个组件各自都包含大量专用模块:分割器由ViT编码器、ViT-Adapter、像素解码器和Transformer解码器组成;追踪器由上下文感知特征提取、重标识层和多个Transformer块组成。以CAVIS为例,其参数量达358M,处理一帧需要838 GFLOPs,在H100 GPU上仅能达到15 FPS。这种日益增长的架构复杂性带来了显著的计算开销和工程复杂度,严重限制了模型在对推理速度要求严格的应用场景中的部署。作者观察到这种趋势后提出质疑:这些复杂的专用组件是否真的必要?是否可以用更简单的方式达到相当的精度?
本文的目标是本文的目标是设计一种极简的编码器架构来完成视频分割任务,在保持与当前最先进方法竞争性精度的同时,大幅提升推理效率。具体而言,作者希望验证以下假设:(1)经过大规模预训练的ViT编码器(如DINOv2)已经具备了足够的能力来承担视频分割中分割和追踪的全部功能;(2)通过在编码器内部引入轻量级的时序建模机制,可以消除对复杂专用追踪模块的需求;(3)这种简化的编码器唯一架构可以实现数量级的速度提升。作者追求的不仅是微小的效率改进,而是将视频分割模型的速度从15 FPS提升到100+ FPS的范式级变革。
与已有工作不同的是,本文的独特切入角度在于对现有视频分割模型进行系统性的'做减法'实验。与大多数研究工作致力于添加新模块来提升性能不同,作者采取了一条反直觉的路线:从当前最先进的CAVIS模型出发,逐步移除其各个专用组件,观察精度的变化。他们首先用EoMT替换复杂的分割器,发现速度提升3倍而精度仅下降0.8 AP;接着移除上下文感知特征和重标识层,进一步加速1.8倍且精度几乎不变;然后完全移除追踪器,速度提升10倍以上但精度下降7.6 AP。在这一分析基础上,作者提出用查询传播和查询融合这两个极简机制来恢复时序建模能力。这种'先拆后建'的研究范式使得本文的创新建立在严谨的消融实验之上,每一步简化都有明确的实验验证作为支撑。
核心方法
VidEoMT的方法可以用一个简洁的直觉来概括:如果一个足够大的ViT编码器已经通过大规模预训练学会了定位和分类物体的能力,那么它也可能学会了跨帧追踪物体的能力,因为DINOv2等视觉基础模型的训练目标本身就鼓励学习跨视角一致的特征表示。基于这一直觉,作者将视频分割任务统一到一个单一的ViT编码器中。整体技术路线是:首先在第一帧中使用标准EoMT的设置,将可学习查询注入ViT编码器的最后几层来生成分割预测;然后在后续帧中,通过查询传播机制将前一帧的物体查询传递到当前帧,并通过查询融合将传播的查询与可学习查询相结合,以平衡时序连续性和对新物体的检测能力。整个过程不引入任何Transformer解码器、像素解码器或专用追踪模块,完全在编码器内部完成。
VidEoMT的核心创新在于两个机制:查询传播和查询融合。查询传播的本质思想是:在EoMT中,每个查询在训练过程中被监督来表示特定的物体,因此同一查询索引在不同帧中应该表示同一物体。基于此,可以将前一帧的输出查询直接作为下一帧的输入,替代可学习查询,从而在不增加任何计算开销的情况下实现跨帧信息流动。查询融合的公式为 $Q_t^F = \text{Linear}(Q_{t-1}^S) + Q^{lrn}$,通过线性层变换前一帧的输出查询,然后与原始可学习查询进行逐元素相加。这一设计的精妙之处在于:传播的查询携带了时序上下文(过去的信息),而可学习查询保持了对新出现物体的适应性(不过度依赖历史信息)。与CAVIS等方法相比,VidEoMT不需要上下文感知特征提取(需要卷积滤波)、不需要重标识层(需要对比学习损失)、不需要独立的追踪器Transformer块,而是用一个简单的线性层加逐元素加法来完成全部的时序建模。
方法步骤详情
VidEoMT的推理流程如下。对于第一帧(t=0):输入图像经过线性Patch嵌入和位置编码后,得到patch token序列;可学习查询 $Q^{lrn} = \{q_i^{lrn} \in \mathbb{R}^D\}_{i=1}^N$ 被拼接到patch token之后,两者共同经过前L1层编码器块处理;然后在最后L2层(通常L2=4)中继续联合处理,最终输出更新后的查询 $Q_0^S$,用于预测第一帧的分割掩码和类别标签。对于后续帧(t>0):不再使用可学习查询,而是通过查询传播使用前一帧的输出 $Q_{t-1}^S$;前一帧的查询首先经过一个线性层变换,然后与可学习查询通过逐元素加法融合,得到融合查询 $Q_t^F$;融合查询被注入到当前帧ViT编码器的最后L2层,与patch token联合处理后输出分割预测和新的查询 $Q_t^S$。训练方面,采用与Mask2Former相同的损失函数(分类用交叉熵,分割用BCE+Dice),总损失为 $\mathcal{L}_{tot} = \lambda_{bce}\mathcal{L}_{bce} + \lambda_{dice}\mathcal{L}_{dice} + \lambda_{ce}\mathcal{L}_{ce}$,权重分别为5.0、5.0和2.0。训练分两阶段:第一阶段仅训练图像分割(COCO实例分割+目标视频数据集),第二阶段引入时序监督进行视频分割微调。模型使用N=200个可学习查询,特征维度D=1024。
技术新颖性
VidEoMT的技术新颖性体现在多个层面。首先,在架构层面,它是第一个实现纯编码器视频分割的模型。此前的方法无论是CAVIS、DVIS++还是DVIS-DAQ,都遵循'分割器+追踪器'的范式,至少包含一个Transformer解码器和若干追踪专用模块。VidEoMT将这一切统一到单一的ViT编码器中,这是一个范式级的创新。其次,在时序建模机制上,查询传播和查询融合的设计极其简洁——仅需一个线性层和一次逐元素加法——却能有效替代整个追踪器模块(包含多个Transformer块、上下文感知特征、重标识层等)。这种以极少参数量和计算量实现相当效果的能力是前所未有的。第三,在实验方法论上,作者采用的'逐步移除'消融策略(表1)为视频分割领域提供了一个清晰的'什么组件真正重要'的分析框架,这种研究范式本身具有方法论价值。最后,本文发现模型规模和预训练规模是决定编码器内时序建模能力的关键因素,这一发现为视觉基础模型的视频理解能力提供了新的见解。
实验结果
本文在六个主流视频分割基准上进行了全面评估,核心发现如下。在YouTube-VIS 2019验证集上,VidEoMT(ViT-L/DINOv2)达到68.6 AP,与CAVIS的68.9 AP仅差0.3,但推理速度从15 FPS提升到160 FPS,实现10倍以上加速。在YouTube-VIS 2021上,VidEoMT达到63.1 AP(CAVIS为64.6,差距1.5),速度同为160 FPS。在YouTube-VIS 2022上,VidEoMT反而以42.6 APL超过CAVIS的39.5 APL(提升3.1),速度161 FPS对15 FPS。在OVIS(遮挡视频实例分割)上,VidEoMT达52.5 AP(CAVIS为53.2,差距0.7),115 FPS对15 FPS。在VIPSeg视频全景分割上,VidEoMT达55.2 VPQ和48.9 STQ,与CAVIS的56.9 VPQ和51.0 STQ差距不大,但速度从10 FPS提升到75 FPS。最令人瞩目的是VSPW视频语义分割上的结果:VidEoMT不仅速度从13 FPS提升到73 FPS,而且mIoU从62.8提升到64.9(+2.1),mVC16从94.2提升到95.0(+0.8),在精度上也超越了所有基线。逐步消融实验(表1)显示:用EoMT替换分割器可将速度提升近3倍(42 FPS),AP仅降0.8;进一步移除上下文感知特征和重标识层,速度达74 FPS,AP几乎不变;完全移除追踪器后速度达162 FPS但AP下降7.6;加入查询传播恢复2.6 AP;加入查询融合最终恢复到68.6 AP,速度160 FPS。模型规模实验(表9)表明,随着ViT规模从S(57M参数)增长到L(318M参数),VidEoMT与CAVIS的性能差距从2.7 AP缩小到0.3 AP。预训练实验(表8)表明,使用DINOv2/DINOv3/EVA-02等大规模预训练时,VidEoMT性能接近CAVIS;使用IN21K时差距扩大到1.4 AP;使用IN1K时差距达2.7 AP,验证了大规模预训练的关键作用。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| YouTube-VIS 2019 视频实例分割 | AP (Average Precision) | 68.6 | CAVIS: 68.9 | 精度持平,速度160 FPS vs 15 FPS(10.7倍加速),GFLOPs从838降至566 |
| YouTube-VIS 2021 视频实例分割 | AP | 63.1 | CAVIS: 64.6 | 精度差距1.5 AP,速度160 FPS vs 15 FPS(10.7倍加速) |
| YouTube-VIS 2022 视频实例分割 | APL (Large objects) | 42.6 | CAVIS: 39.5 | 精度超越3.1 APL,速度161 FPS vs 15 FPS(10.7倍加速) |
| OVIS 遮挡视频实例分割 | AP | 52.5 | CAVIS: 53.2 | 精度差距0.7 AP,速度115 FPS vs 15 FPS(7.7倍加速) |
| VIPSeg 视频全景分割 | VPQ | 55.2 | CAVIS: 56.9 | 精度差距1.7 VPQ,速度75 FPS vs 10 FPS(7.5倍加速) |
| VSPW 视频语义分割 | mIoU | 64.9 | DVIS++: 62.8 | 精度提升2.1 mIoU,速度73 FPS vs 13 FPS(5.6倍加速) |
局限与改进
尽管VidEoMT展现了令人印象深刻的效率-精度权衡,但仍存在以下局限性。首先,VidEoMT严重依赖大规模预训练(如DINOv2),在使用较弱的预训练(如ImageNet-1K)时,与CAVIS的性能差距会显著扩大到2.7 AP,这限制了其在预训练资源受限场景下的适用性。其次,在OVIS(遮挡视频实例分割)基准上,VidEoMT与最优方法DVIS-DAQ存在约1.8 AP的差距,在遮挡严重的场景中追踪能力仍不如复杂的专用追踪器。第三,VidEoMT的性能对模型规模敏感:在ViT-S(25M参数)时与CAVIS的差距为2.7 AP,仅当模型增大到ViT-L(318M参数)时差距才缩小到0.3 AP,这意味着在资源受限场景下(需要小模型),性能优势会减弱。第四,查询传播机制使模型仅能获取前一帧的信息,无法像完整的追踪器那样维护长期的物体状态记忆,这在物体长时间消失后重新出现的场景中可能成为瓶颈。最后,作者观察到DINOv3预训练相比DINOv2仅带来微小改进(68.9 vs 68.6 AP),这可能与DINOv3设计为冻结使用(而非微调)有关,暗示VidEoMT可能未能充分利用更新的预训练模型的潜力。
独立分析的弱点
VidEoMT有几个值得深入分析的弱点和对应的改进方向。第一,查询传播仅使用前一帧的信息,这意味着物体的时序信息传播范围有限。在物体短暂被遮挡后重新出现的场景中(如OVIS数据集),模型可能难以正确恢复追踪。改进方向:可以设计多帧查询聚合机制,例如使用指数移动平均或注意力池化来维护长期物体状态,而不是简单地用前一帧的查询替换当前帧的输入。第二,查询融合采用简单的线性层加逐元素加法,这种融合方式可能过于简单,无法有效处理物体数量变化(如新物体出现或物体消失)的复杂情况。改进方向:可以用自适应门控机制替代简单加法,让模型动态控制时序信息和新检测信息的融合比例。第三,N=200个可学习查询是固定的,这意味着模型最多只能追踪200个物体。在密集场景(如VIPSeg全景分割中有大量物体)中,这一限制可能导致部分物体无法被检测。改进方向:可以设计动态查询分配策略,根据场景复杂度自适应调整查询数量。第四,VidEoMT在训练时需要两阶段训练(先图像分割再视频分割),这比端到端训练更复杂。改进方向:探索单阶段联合训练策略,可能通过课程学习逐步引入时序约束。
未来方向
基于VidEoMT的成果,未来研究可以从多个方向展开。首先,将编码器唯一视频分割范式扩展到更多任务,如视频目标检测、视频语义分割的实例级追踪,甚至是视频问答等需要时序理解的任务。作者的工作为'ViT编码器可以承担时序建模'这一假设提供了强有力的实验证据,这可能启发更多任务的架构简化。其次,探索更高效的查询传播策略,例如结合记忆库(memory bank)维护多帧的物体信息,或者设计分层传播机制在不同时间尺度上建模时序关系。第三,研究预训练策略对视频分割能力的影响:DINOv2的跨视角一致性预训练目标已经暗示了视频追踪的潜力,是否可以设计专门针对视频任务的预训练目标(如跨帧一致性约束)来进一步增强编码器的时序建模能力?第四,将VidEoMT与新兴的视觉基础模型(如DINOv3、SigLIP等)结合,探索更大规模预训练是否能进一步缩小与复杂模型的精度差距。最后,将编码器唯一设计应用于视频生成、视频编辑等生成式任务中,验证其在更广泛视频理解场景中的通用性。
复现评估
本文的复现条件较为良好。作者已公开代码(https://www.tue-mps.org/videomt/),这对于复现至关重要。模型架构相对简单,核心创新(查询传播和融合)仅需一个线性层和一次逐元素加法,实现复杂度低。训练方面,使用标准的DINOv2预训练ViT-L(可通过Hugging Face获取),损失函数采用与Mask2Former相同的设置,超参数(batch size 8、学习率10^-4、LLRD因子0.6等)均有明确说明。然而,复现存在一定门槛:(1)训练使用8张NVIDIA H100 GPU,对算力要求较高;(2)FPS测量使用FlashAttention v2和torch.compile,在不同硬件和PyTorch版本上结果可能有差异;(3)两阶段训练(先COCO图像分割再视频数据集微调)需要额外的训练时间和数据准备。总体而言,具备中等以上GPU资源的研究团队应该能够在1-2周内完成复现。
论文图表
展示了从CAVIS到VidEoMT的4个中间架构步骤的可视化。Step (1)展示了用EoMT替换分割器后的架构;Step (2)展示了移除上下文感知特征后的架构;Step (3)展示了进一步移除重标识层后的架构;Step (4)展示了完全移除追踪器(即纯EoMT)后的架构。每一步都清晰标注了剩余的模块。
该图是表1的架构可视化补充,帮助读者直观理解每个消融步骤具体移除了哪些组件。
不同查询传播策略的对比。无传播:61.3 AP/162 FPS;仅传播:63.9 AP/162 FPS;非物体重置:67.8 AP/157 FPS;TrackFormer方式:67.7 AP/117 FPS;融合(VidEoMT):68.6 AP/160 FPS。
该表格验证了查询融合策略相比其他传播方式的优越性,同时展示了非物体重置策略作为替代方案的效果。
补充的模型规模实验,额外报告了EoMT+CAVIS组合在不同规模下的结果。ViT-L:EoMT+CAVIS 68.1 AP/42 FPS vs VidEoMT 68.6 AP/160 FPS;ViT-B:EoMT+CAVIS 57.4 AP/67 FPS vs VidEoMT 58.2 AP/251 FPS;ViT-S:EoMT+CAVIS 50.3 AP/93 FPS vs VidEoMT 52.8 AP/294 FPS。VidEoMT在所有规模上都优于EoMT+CAVIS。
该表格补充了主文中表9的结果,证明VidEoMT不仅比CAVIS简单,也比EoMT+CAVIS的简单组合更优。
在YouTube-VIS 2019数据集上CAVIS(15 FPS)和VidEoMT(160 FPS)的定性结果对比,展示了多个视频序列中两种方法的分割和追踪结果。
定性结果直观展示了VidEoMT在实际视频序列中能够产生与CAVIS视觉上相当的分割和追踪效果。
在OVIS数据集上CAVIS(15 FPS)和VidEoMT(112 FPS)的定性结果对比,展示了在遮挡场景下两种方法的追踪表现。
OVIS是遮挡基准,这些定性结果有助于理解VidEoMT在遮挡场景中的追踪能力。
在VIPSeg数据集上CAVIS(10 FPS)和VidEoMT(75 FPS)的定性结果对比,展示了全景分割场景中两种方法的表现。
VIPSeg包含更多物体类别和更密集的场景,定性结果展示了VidEoMT在全景分割任务上的适用性。