arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.13347cs.CV

vid-TLDR:面向轻量级视频Transformer的无训练Token合并

vid-TLDR: Training Free Token merging for Light-weight Video Transformer

Korea University(高丽大学)

Joonmyung Choi, Sanghyeok Lee, Jaewon Chu, Minhyuk Choi, Hyunwoo J. Kim

AI总结:

针对视频Transformer计算成本高和背景块影响泛化的问题,提出无需训练的vid-TLDR方法,利用注意力图识别显著区域并合并背景Token,在降低计算复杂度的同时保持有竞争力的性能。

摘要:

视频Transformer凭借其卓越的表达能力和灵活性,已成为各种视频下游任务的主流解决方案。然而,这些视频Transformer因整个视频帧中大量Token而产生沉重的计算成本,这已成为训练模型的主要障碍。此外,与主要内容无关的图像块(例如背景)会降低模型的泛化性能。为解决这些问题,我们提出面向轻量级视频Transformer的无训练Token合并方法(vid-TLDR),旨在通过合并背景Token而无需额外训练来提升视频Transformer的效率。对于vid-TLDR,我们引入一种仅利用注意力图捕获视频中显著区域的新方法。进一步地,我们引入显著性感知的Token合并策略,通过丢弃背景Token并锐化目标分数来实现。我们的实验表明,vid-TLDR显著降低了视频Transformer的计算复杂度,同时与未使用vid-TLDR的基础模型相比取得了有竞争力的性能。代码可在https://github.com/mlvlab/vid-TLDR获取。

英文摘要:

Video Transformers have become the prevalent solution for various video downstream tasks with superior expressive power and flexibility. However, these video transformers suffer from heavy computational costs induced by the massive number of tokens across the entire video frames, which has been the major barrier to training the model. Further, the patches irrelevant to the main contents, e.g., backgrounds, degrade the generalization performance of models. To tackle these issues, we propose training free token merging for lightweight video Transformer (vid-TLDR) that aims to enhance the efficiency of video Transformers by merging the background tokens without additional training. For vid-TLDR, we introduce a novel approach to capture the salient regions in videos only with the attention map. Further, we introduce the saliency-aware token merging strategy by dropping the background tokens and sharpening the object scores. Our experiments show that vid-TLDR significantly mitigates the computational complexity of video Transformers while achieving competitive performance compared to the base model without vid-TLDR. Code is available at https://github.com/mlvlab/vid-TLDR.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑