arXiv:2605.18013cs.CVcs.AI2026-05被引 2

TinySAM 2 用压缩记忆和轻量编码器,让视频分割模型更省资源。

TinySAM 2: Extreme Memory Compression for Efficient Track Anything Model

论文配图:TinySAM 2: Extreme Memory Compression for Efficient Track Anything Model
图 1 · 摘自论文原文
  • 选关键帧存记忆,空间时间双域压缩token
  • 性能达SAM 2.1的90%,内存仅用7%、数据仅3%
  • 适合移动端部署,兼顾效率与精度

Segment Anything Model 2 (SAM 2) 是视频分割领域的核心基础模型,引入记忆库机制,在半监督视频对象分割和跟踪任务中表现优异。但其多阶段图像编码器与记忆模块的复杂计算特性,限制了实际部署。为此,我们提出TinySAM 2,一种轻量级视频分割模型,在性能与效率间取得平衡。首先,设计记忆质量管理机制,仅保留高信息量的历史帧作为记忆;其次,提出联合时空令牌压缩方法:空间上使用平均池化消除冗余令牌,时间上基于令牌级相似度在记忆库中选择关键帧。此外,采用RepViT作为轻量图像编码器,进一步减少参数量。在DAVIS和SA-V等挑战性数据集上的大量实验表明,TinySAM 2达到SAM 2.1 90%的性能,仅需7%的内存令牌和3%的训练数据。本研究有效缓解了SAM 2在参数量、计算负载和部署成本方面的瓶颈,为视频分割模型在设备端的广泛应用提供了高效解决方案。

原文摘要 · Abstract (English)

Segment Anything Model 2 (SAM 2) serves as a core foundation model in the field of video segmentation. Building upon the original SAM model, it introduces a memory bank mechanism and demonstrates outstanding performance in tasks such as semi-supervised video object segmentation and tracking anything. However, the complex computational characteristics of SAM 2's multi-stage image encoder and memory module have raised the barrier to the model's deployment in practical applications. To address this issue, we propose TinySAM 2, a lightweight video segmentation model that balances performance and efficiency. First, a memory quality management mechanism is introduced to select and retain high-informative historical frames as the memory. In addition, a joint-spatial-temporal token compression is proposed that reduces the memory storage and computational cost. Specifically, average pooling is employed to first compress redundancy tokens in the spatial domain. In the temporal domain, informative tokens are selected across frames in the memory bank based on token-level similarity measurement. Besides, we take RepViT as the lightweight image encoder, which further reduces the model parameters. Extensive experiments on challenging datasets such as DAVIS and SA-V demonstrate that TinySAM 2 achieves 90% of the performance of SAM 2.1, with only 7% memory tokens and 3% training data. This study effectively alleviates the bottlenecks in parameter count, computational load, and deployment costs associated with SAM 2, providing a resource-efficient solution for the widespread application of video segmentation models on devices.

视频分割轻量模型记忆压缩

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