用[CLS] token自动筛选重要视觉信息,无需训练即可高效压缩多模态模型。
[CLS] Token Tells Everything Needed for Training-free Efficient MLLMs
- 利用[CLS] token对视觉标记的注意力得分作为压缩依据。
- 在多个任务上达到当前最佳效果,计算开销显著降低。
- 适合追求高效部署的多模态模型研究者和工程师。
多模态大语言模型(MLLMs)在众多视觉-语言任务中表现优异,但其高效部署面临计算成本高、内存需求大的挑战。现有方法虽能无训练地压缩视觉标记,但依赖视觉标记与提示文本间的注意力作为重要性指标,忽略与生成文本的相关性,引入感知偏差。本文发现:在MLLMs中,视觉编码器的[CLS] token已蕴含判断哪些视觉标记关键的信息。基于此,提出无需训练的视觉标记压缩方法VTC-CLS:首先以[CLS] token对视觉标记的注意力得分作为剪枝依据;进一步融合不同层[CLS] token的重要性得分,更全面捕捉关键视觉信息。大量实验表明,相较于基线方法,VTC-CLS在各类任务上均达领先性能,且在无训练情况下显著降低计算开销,展现出优越性。代码与模型已公开于https://github.com/THU-MIG/VTC-CLS。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) have recently demonstrated strong performance across a wide range of vision-language tasks, garnering significant attention in the computer vision. However, their efficient deployment remains a substantial challenge due to high computational costs and memory requirements. Recognizing the redundancy of information within the vision modality, recent studies have explored methods for compressing visual tokens in MLLMs to enhance efficiency in a training-free manner. Despite their effectiveness, existing methods like Fast rely on the attention between visual tokens and prompt text tokens as the importance indicator, overlooking the relevance to response text and thus introducing perception bias. In this paper, we demonstrate that in MLLMs, the [CLS] token in the visual encoder inherently knows which visual tokens are important for MLLMs. Building on this prior, we introduce a simple yet effective method for train-free visual token compression, called VTC-CLS. Firstly, it leverages the attention score of the [CLS] token on visual tokens as an importance indicator for pruning visual tokens. Besides, we also explore ensembling the importance scores derived by the [CLS] token from different layers to capture the key visual information more comprehensively. Extensive experiments demonstrate that our VTC-CLS achieves the state-of-the-art performance across various tasks compared with baseline methods. It also brings notably less computational costs in a training-free manner, highlighting its effectiveness and superiority. Code and models are available at \url{https://github.com/THU-MIG/VTC-CLS}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。