arXiv:2504.10854cs.CV2025-04被引 7

提出无训练视觉令牌剪枝方法,加速大模型分割任务。

LVLM_CSP: Accelerating Large Vision Language Models via Clustering, Scattering, and Pruning for Reasoning Segmentation

  • 分三阶段:聚类、打散、剪枝,逐步优化令牌处理
  • 70%计算量减少,仅损失1%分割精度
  • 专为精准语义空间推理设计,适合高精度分割场景

大型视觉语言模型(LVLMs)被广泛用于指导视觉基础模型完成推理分割任务,表现优异。然而,其巨大的计算开销带来新挑战,主要源于处理数百个图像令牌。有效缓解该问题的方法是减少图像令牌数量,即图像令牌剪枝。现有剪枝方法多针对视觉问答等高层理解任务,而基于文本查询生成精确视觉掩码需更强的语义与空间推理能力,因此剪枝必须在推理过程中精细控制每个令牌。我们实证分析发现,现有方法难以平衡计算开销降低与高分割精度维持。为此,本文提出无需训练的视觉令牌剪枝方法 LVLM_CSP,包含三个阶段:首先用部分选中令牌进行粗粒度视觉推理;其次执行细粒度推理;最后在最后一阶段剪除大部分视觉令牌。大量实验表明,该方法在7B LVLM上实现65%图像令牌推理FLOPs降低且精度几乎不变,70%降低时仅损失1%精度。

原文摘要 · Abstract (English)

Large Vision Language Models (LVLMs) have been widely adopted to guide vision foundation models in performing reasoning segmentation tasks, achieving impressive performance. However, the substantial computational overhead associated with LVLMs presents a new challenge. The primary source of this computational cost arises from processing hundreds of image tokens. Therefore, an effective strategy to mitigate such overhead is to reduce the number of image tokens, a process known as image token pruning. Previous studies on image token pruning for LVLMs have primarily focused on high level visual understanding tasks, such as visual question answering and image captioning. In contrast, guiding vision foundation models to generate accurate visual masks based on textual queries demands precise semantic and spatial reasoning capabilities. Consequently, pruning methods must carefully control individual image tokens throughout the LVLM reasoning process. Our empirical analysis reveals that existing methods struggle to adequately balance reductions in computational overhead with the necessity to maintain high segmentation accuracy. In this work, we propose LVLM_CSP, a novel training free visual token pruning method specifically designed for LVLM based reasoning segmentation tasks. LVLM_CSP consists of three stages: clustering, scattering, and pruning. Initially, the LVLM performs coarse-grained visual reasoning using a subset of selected image tokens. Next, fine grained reasoning is conducted, and finally, most visual tokens are pruned in the last stage. Extensive experiments demonstrate that LVLM_CSP achieves a 65% reduction in image token inference FLOPs with virtually no accuracy degradation, and a 70% reduction with only a minor 1% drop in accuracy on the 7B LVLM.

视觉语言模型图像剪枝分割任务高效推理

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