arXiv:2504.17040cs.CVcs.AI2025-04被引 14

动态压缩视觉令牌,让多模态模型更快更省资源。

DyMU: Dynamic Merging and Virtual Unmerging for Efficient VLMs

  • 根据图像复杂度动态合并相似视觉令牌,减少计算量。
  • 无需微调即可还原完整注意力机制,保持下游任务性能。
  • 训练免费,适配主流多模态模型,灵活控制计算开销。

我们提出DyMU,一种无需训练的高效框架,可动态降低视觉语言模型(VLM)的计算负担,同时保持高任务性能。方法包含两个核心组件:首先,动态令牌合并(DToMe)基于图像复杂度合并相似视觉令牌,解决视觉变换器固定长度输出的固有低效问题;其次,虚拟令牌解合并(VTU)通过高效重构完整序列的注意力动态,模拟大型语言模型所需的令牌序列,从而在不增加微调成本的前提下保持下游性能。与以往方法不同,该方法能根据图像内容动态调整令牌压缩策略,且完全无需训练,适用于多数前沿VLM架构。在图像与视频理解任务上的广泛实验表明,DyMU可将平均视觉令牌数减少32%至85%,并在多种VLM架构(包括近期流行的AnyRes基视觉编码器)上实现与全长度模型相当的性能。定性分析进一步验证了DToMe能有效依据图像复杂度调整压缩程度,且相比现有系统为用户提供更高的计算成本可控性。

原文摘要 · Abstract (English)

We present DyMU, an efficient, training-free framework that dynamically reduces the computational burden of vision-language models (VLMs) while maintaining high task performance. Our approach comprises two key components. First, Dynamic Token Merging (DToMe) reduces the number of visual token embeddings by merging similar tokens based on image complexity, addressing the inherent inefficiency of fixed-length outputs in vision transformers. Second, Virtual Token Unmerging (VTU) simulates the expected token sequence for large language models (LLMs) by efficiently reconstructing the attention dynamics of a full sequence, thus preserving the downstream performance without additional fine-tuning. Unlike previous approaches, our method dynamically adapts token compression to the content of the image and operates completely training-free, making it readily applicable to most state-of-the-art VLM architectures. Extensive experiments on image and video understanding tasks demonstrate that DyMU can reduce the average visual token count by 32%-85% while achieving comparable performance to full-length models across diverse VLM architectures, including the recently popularized AnyRes-based visual encoders. Furthermore, through qualitative analyses, we demonstrate that DToMe effectively adapts token reduction based on image complexity and, unlike existing systems, provides users more control over computational costs. Project page: https://mikewangwzhl.github.io/dymu/.

多模态模型压缩视觉语言动态优化

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