轻量级模型实现高精度视频目标分割,手机端也能流畅运行。
Efficient Track Anything
- 用简单ViT替代复杂编码器,降低计算开销。
- 新内存模块使帧特征提取和分割速度提升2倍以上。
- 适合移动端部署,手机上可达10帧/秒,效果接近大模型。
Segment Anything Model 2(SAM 2)在视频目标分割与跟踪任务中表现优异,其核心在于多阶段图像编码器和记忆机制。然而,这些组件计算复杂度高,限制了其在移动设备等实际场景的应用。为此,本文提出EfficientTAMs——轻量级的“跟踪任何”模型,在保持高质量结果的同时显著降低延迟与模型尺寸。方法上,采用非分层的普通Vision Transformer(ViT)作为图像编码器,并设计高效记忆模块,大幅减少帧特征提取和当前帧分割的计算负担。在SA-1B和SA-V数据集上训练后,EfficientTAMs在多个视频分割基准测试中表现良好:使用普通ViT的模型相比SAM 2(HieraB+SAM 2)实现约2倍加速(A100),参数量减少约2.4倍;在图像分割任务中,相比原始SAM提速约20倍,参数量减少20倍。在iPhone 15 Pro Max等移动端设备上,可实现约10帧/秒的视频目标分割,具备实用化潜力。
原文摘要 · Abstract (English)
Segment Anything Model 2 (SAM 2) has emerged as a powerful tool for video object segmentation and tracking anything. Key components of SAM 2 that drive the impressive video object segmentation performance include a large multistage image encoder for frame feature extraction and a memory mechanism that stores memory contexts from past frames to help current frame segmentation. The high computation complexity of multistage image encoder and memory module has limited its applications in real-world tasks, e.g., video object segmentation on mobile devices. To address this limitation, we propose EfficientTAMs, lightweight track anything models that produce high-quality results with low latency and model size. Our idea is based on revisiting the plain, nonhierarchical Vision Transformer (ViT) as an image encoder for video object segmentation, and introducing an efficient memory module, which reduces the complexity for both frame feature extraction and memory computation for current frame segmentation. We take vanilla lightweight ViTs and efficient memory module to build EfficientTAMs, and train the models on SA-1B and SA-V datasets for video object segmentation and track anything tasks. We evaluate on multiple video segmentation benchmarks including semi-supervised VOS and promptable video segmentation, and find that our proposed EfficientTAM with vanilla ViT perform comparably to SAM 2 model (HieraB+SAM 2) with ~2x speedup on A100 and ~2.4x parameter reduction. On segment anything image tasks, our EfficientTAMs also perform favorably over original SAM with ~20x speedup on A100 and ~20x parameter reduction. On mobile devices such as iPhone 15 Pro Max, our EfficientTAMs can run at ~10 FPS for performing video object segmentation with reasonable quality, highlighting the capability of small models for on-device video object segmentation applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。