arXiv:2501.17391cs.CVcs.AI2025-01被引 1

无需训练即可大幅减少视觉令牌,加速多模态大模型推理。

LFTR: Learning-Free Token Reduction for Multimodal Large Language Models

  • 利用视觉表示冗余,无须训练直接压缩令牌
  • 最多减少16倍视觉令牌,性能不降反升
  • 适配主流开源多模态模型,可与加速技术叠加

多模态大语言模型(MLLMs)在多种任务中表现优异,但部署受限于高计算开销和长推理时间。由于视觉模态信息远超文本模态,编码后产生大量令牌,导致注意力机制的二次复杂度带来显著计算负担。现有令牌压缩方法通常依赖特定架构,需大量微调,难以通用。本文提出无需学习的令牌压缩(LFTR)方法,可无缝集成至多数开源MLLM架构,无需额外微调。通过利用视觉表示中的冗余,该方法有效减少令牌数量,同时保持甚至提升模型推理性能。我们在LLaVA、MiniGPT、QwenVL等多架构上验证,结果表明,LFTR实现最高16倍的视觉令牌压缩,在主流视觉问答基准上性能持平或更优,且与视觉编码器压缩、后训练量化等技术兼容,进一步推动MLLM高效部署。项目代码已公开。

原文摘要 · Abstract (English)

Multimodal Large Language Models (MLLMs) have demonstrated exceptional success in various multimodal tasks, yet their deployment is frequently limited by substantial computational demands and prolonged inference times. Given that the vision modality typically contains more comprehensive information than the text modality, resulting in encoded representations comprising an extensive number of tokens, leading to significant computational overhead due to the quadratic complexity of the attention mechanism. Current token reduction methods are typically restricted to specific model architectures and often necessitate extensive retraining or fine-tuning, restricting their applicability to many state-of-the-art models. In this paper, we introduce a learning-free token reduction (LFTR) method designed for MLLMs. LFTR can be seamlessly integrated into most open-source MLLM architectures without requiring additional fine-tuning. By capitalizing on the redundancy in visual representations, our approach effectively reduces tokens while preserving the general inference performance of MLLMs. We conduct experiments on multiple MLLM architectures (LLaVA, MiniGPT, QwenVL), and our results show that LFTR achieves up to a $16\times$ reduction of visual tokens while maintaining or even enhancing performance on mainstream vision question-answering benchmarks, all in a learning-free setting. Additionally, LFTR is complementary to other acceleration techniques, such as vision encoder compression and post-training quantization, further promoting the efficient deployment of MLLMs. Our project is available at https://anonymous.4open.science/r/LFTR-AAAI-0528.

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

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