Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

PruneVid: Visual Token Pruning for Efficient Video Large Language Models

About

In this paper, we introduce PruneVid, a visual token pruning method designed to enhance the efficiency of multi-modal video understanding. Large Language Models (LLMs) have shown promising performance in video tasks due to their extended capabilities in comprehending visual modalities. However, the substantial redundancy in video data presents significant computational challenges for LLMs. To address this issue, we introduce a training-free method that 1) minimizes video redundancy by merging spatial-temporal tokens, and 2) leverages LLMs' reasoning capabilities to selectively prune visual features relevant to question tokens, enhancing model efficiency. We validate our method across multiple video benchmarks, which demonstrate that PruneVid can prune over 80% of tokens while maintaining competitive performance combined with different model networks. This highlights its superior effectiveness and efficiency compared to existing pruning methods. Code: https://github.com/Visual-AI/PruneVid.

Xiaohu Huang, Hao Zhou, Kai Han• 2024

Related benchmarks

TaskDatasetResultRank
Video UnderstandingMVBench
Accuracy57.6
425
Video UnderstandingVideoMME
Score (Long)54.3
248
Long Video UnderstandingLongVideoBench
Score59.2
248
Video UnderstandingVideoMME
Overall Score58
222
Video UnderstandingMLVU
Score64.54
221
Video UnderstandingEgoSchema
EgoSchema Score62.8
158
Long Video UnderstandingMLVU
Score65.5
154
Video UnderstandingVideo-MME
Overall Score58.19
92
Video UnderstandingLVB
Accuracy67.23
89
Long Video UnderstandingLongVideo-Bench
Score57.7
89
Showing 10 of 28 rows

Other info

Follow for update