arXiv:2505.19235cs.LGcs.CV2025-05ICML被引 14

通过联合优化令牌与神经元稀疏性,实现视觉语言模型高效推理。

CoreMatching: A Co-adaptive Sparse Inference Framework with Token and Neuron Pruning for Comprehensive Acceleration of Vision-Language Models

  • 设计协同稀疏框架,让关键令牌与神经元相互匹配提升效率。
  • 在十项图像理解任务中实现5倍计算量减少和10倍加速。
  • 适合需要高效部署视觉语言模型的研究者与工程师。

视觉语言模型在多种任务中表现优异,但推理时存在时间和内存开销高的问题。令牌稀疏性缓解了令牌使用低效,神经元稀疏性降低了高维计算负担,两者均是提升效率的有力方案。然而,当前两种稀疏范式多独立发展,普遍认为它们互不干扰。本文首次系统探究二者是否真正独立。通过分析核心神经元与核心令牌间的匹配机制,发现二者在推理中相互影响并共同增强。基于此,提出协同稀疏推理框架CoreMatching,利用令牌与神经元稀疏性的协同效应提升效率。理论分析与实测表明,该方法在十项图像理解任务及三类硬件设备上均优于现有基准。特别地,在NVIDIA Titan Xp上实现5倍浮点运算量减少与10倍整体加速。代码已开源。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) excel across diverse tasks but suffer from high inference costs in time and memory. Token sparsity mitigates inefficiencies in token usage, while neuron sparsity reduces high-dimensional computations, both offering promising solutions to enhance efficiency. Recently, these two sparsity paradigms have evolved largely in parallel, fostering the prevailing assumption that they function independently. However, a fundamental yet underexplored question remains: Do they truly operate in isolation, or is there a deeper underlying interplay that has yet to be uncovered? In this paper, we conduct the first comprehensive investigation into this question. By introducing and analyzing the matching mechanism between Core Neurons and Core Tokens, we found that key neurons and tokens for inference mutually influence and reinforce each other. Building on this insight, we propose CoreMatching, a co-adaptive sparse inference framework, which leverages the synergy between token and neuron sparsity to enhance inference efficiency. Through theoretical analysis and efficiency evaluations, we demonstrate that the proposed method surpasses state-of-the-art baselines on ten image understanding tasks and three hardware devices. Notably, on the NVIDIA Titan Xp, it achieved 5x FLOPs reduction and a 10x overall speedup. Code is released at https://github.com/wangqinsi1/2025-ICML-CoreMatching/tree/main.

视觉语言模型稀疏推理高效部署

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