vid-TLDR: Training Free Token merging for Light-weight Video Transformer
This addresses efficiency issues in video Transformers for computer vision applications, but it is incremental as it builds on existing token merging techniques.
The paper tackles the high computational cost and reduced generalization of video Transformers due to excessive background tokens by proposing vid-TLDR, a training-free method that merges background tokens using attention maps, achieving competitive performance while significantly reducing complexity.
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.