arXiv:2503.10501cs.CV2025-03被引 21

不重训练就能大幅压缩视觉令牌,保持模型精度。

TokenCarve: Information-Preserving Visual Token Compression in Multimodal Large Language Models

  • 基于信息保留策略,分两阶段剪枝与合并视觉令牌。
  • 将视觉令牌减至原数量的22.2%,推理速度提升1.23倍。
  • 无需重新训练,适合快速部署于各类多模态大模型。

多模态大语言模型(MLLMs)日益流行,但视觉令牌带来的高计算开销仍是主要挑战。现有训练型压缩方法需昂贵重训练,而无训练方法在激进压缩下性能下降明显。本研究发现,MLLM性能退化与注意力输出矩阵的信息损失加速密切相关。基于此洞察,提出无需训练、即插即用的两阶段压缩框架TokenCarve:第一阶段采用信息保留引导选择(IPGS)剔除低信息令牌;第二阶段进一步用IPGS指导令牌合并,最小化信息损失。在11个数据集和2种模型变体上的实验表明,TokenCarve可将视觉令牌减少至原数的22.2%,实现1.23倍推理加速、KV缓存存储降低64%,仅带来1.54%准确率下降。代码已开源。

原文摘要 · Abstract (English)

Multimodal Large Language Models (MLLMs) are becoming increasingly popular, while the high computational cost associated with multimodal data input, particularly from visual tokens, poses a significant challenge. Existing training-based token compression methods improve inference efficiency but require costly retraining, while training-free methods struggle to maintain performance when aggressively reducing token counts. In this study, we reveal that the performance degradation of MLLM closely correlates with the accelerated loss of information in the attention output matrix. This insight introduces a novel information-preserving perspective, making it possible to maintain performance even under extreme token compression. Based on this finding, we propose TokenCarve, a training-free, plug-and-play, two-stage token compression framework. The first stage employs an Information-Preservation-Guided Selection (IPGS) strategy to prune low-information tokens, while the second stage further leverages IPGS to guide token merging, minimizing information loss. Extensive experiments on 11 datasets and 2 model variants demonstrate the effectiveness of TokenCarve. It can even reduce the number of visual tokens to 22.2% of the original count, achieving a 1.23x speedup in inference, a 64% reduction in KV cache storage, and only a 1.54% drop in accuracy. Our code is available at https://github.com/ShawnTan86/TokenCarve.

多模态令牌压缩推理加速无训练

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