arXiv:2409.01156cs.CV2024-09ICLR被引 47

通过帧间信息合并,大幅降低视频检索的计算开销。

TempMe: Video Temporal Token Merging for Efficient Text-Video Retrieval

论文配图:TempMe: Video Temporal Token Merging for Efficient Text-Video Retrieval
图 1 · 摘自论文原文
  • 提出渐进式多粒度帧合并机制,压缩冗余时空特征。
  • 仅用0.5万参数实现95%输出令牌减少和51%计算量下降。
  • 适合需要低资源部署的视频检索应用,尤其适配移动端。

现有文本-视频检索方法多以CLIP等图文预训练模型为骨干,对每帧独立编码,导致计算开销高,难以实际部署。针对这一问题,本文提出一种参数高效且推理高效的文本-视频检索架构TempMe,旨在最小化可训练参数与模型复杂度。核心在于设计渐进式多粒度框架,通过逐步合并相邻片段,有效消除视频中连续帧间的时序冗余,增强跨帧建模能力。实验表明,相较于现有参数高效方法,TempMe仅需0.50M可训练参数,在保持性能的同时,输出令牌减少95%,GFLOPs降低51%,推理速度提升1.8倍,R-Sum提升4.4%;全微调下更实现7.9%的R-Sum增益,训练速度加快1.57倍,GPU内存占用降低至75.2%。代码已开源。

原文摘要 · Abstract (English)

Most text-video retrieval methods utilize the text-image pre-trained models like CLIP as a backbone. These methods process each sampled frame independently by the image encoder, resulting in high computational overhead and limiting practical deployment. Addressing this, we focus on efficient text-video retrieval by tackling two key challenges: 1. From the perspective of trainable parameters, current parameter-efficient fine-tuning methods incur high inference costs; 2. From the perspective of model complexity, current token compression methods are mainly designed for images to reduce spatial redundancy but overlook temporal redundancy in consecutive frames of a video. To tackle these challenges, we propose Temporal Token Merging (TempMe), a parameter-efficient and training-inference efficient text-video retrieval architecture that minimizes trainable parameters and model complexity. Specifically, we introduce a progressive multi-granularity framework. By gradually combining neighboring clips, we reduce spatio-temporal redundancy and enhance temporal modeling across different frames, leading to improved efficiency and performance. Extensive experiments validate the superiority of our TempMe. Compared to previous parameter-efficient text-video retrieval methods, TempMe achieves superior performance with just 0.50M trainable parameters. It significantly reduces output tokens by 95% and GFLOPs by 51%, while achieving a 1.8X speedup and a 4.4% R-Sum improvement. With full fine-tuning, TempMe achieves a significant 7.9% R-Sum improvement, trains 1.57X faster, and utilizes 75.2% GPU memory usage. The code is available at https://github.com/LunarShen/TempMe.

视频检索模型压缩高效推理时空建模

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