arXiv:2506.01304cs.CV2025-06CVPR被引 13

用不到0.2%训练成本,让SAM实现精准视频分割

SAM-I2V: Upgrading SAM to Support Promptable Video Segmentation with Less than 0.2% Training Cost

  • 基于预训练SAM升级,引入时空特征提取与记忆过滤机制
  • 性能达SAM 2的90%以上,训练成本仅其0.2%
  • 适合想低成本部署视频分割的开发者和研究者

以分割一切模型(SAM)为代表的基础模型显著推动了可提示图像分割的发展。然而将能力扩展至视频面临巨大挑战,尤其在动态场景中实现精确且时序一致的掩码传播。SAM 2通过从头训练大规模图文数据来学习复杂时空关联,导致极高的训练成本,阻碍研究与应用。本文提出SAM-I2V,一种高效的图像到视频升级方法,用于构建可提示视频分割(PVS)模型。该方法通过三个关键创新:(i) 基于SAM静态图像编码器的图像到视频特征提取升级模块,实现时空感知;(ii) 选择最相关历史帧的记忆过滤策略,提升历史信息利用效率;(iii) 利用对象记忆的“记忆即提示”机制,确保动态场景中时序一致性掩码传播。全面实验表明,本方法在仅使用SAM 2 0.2%训练成本下,性能超过其90%。工作为PVS提供了资源高效路径,降低了模型设计门槛,推动更广泛的应用与进展。代码与模型已开源。

原文摘要 · Abstract (English)

Foundation models like the Segment Anything Model (SAM) have significantly advanced promptable image segmentation in computer vision. However, extending these capabilities to videos presents substantial challenges, particularly in ensuring precise and temporally consistent mask propagation in dynamic scenes. SAM 2 attempts to address this by training a model on massive image and video data from scratch to learn complex spatiotemporal associations, resulting in huge training costs that hinder research and practical deployment. In this paper, we introduce SAM-I2V, an effective image-to-video upgradation method for cultivating a promptable video segmentation (PVS) model. Our approach strategically upgrades the pre-trained SAM to support PVS, significantly reducing training complexity and resource requirements. To achieve this, we introduce three key innovations: (i) an image-to-video feature extraction upgrader built upon SAM's static image encoder to enable spatiotemporal video perception, (ii) a memory filtering strategy that selects the most relevant past frames for more effective utilization of historical information, and (iii) a memory-as-prompt mechanism leveraging object memory to ensure temporally consistent mask propagation in dynamic scenes. Comprehensive experiments demonstrate that our method achieves over 90% of SAM 2's performance while using only 0.2% of its training cost. Our work presents a resource-efficient pathway to PVS, lowering barriers for further research in PVS model design and enabling broader applications and advancements in the field. Code and model are available at: https://github.com/showlab/SAM-I2V.

视频分割SAM高效训练提示分割

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。