arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2504.08966cs.CV

PACT:基于剪枝和聚类的视觉语言模型加速方法

PACT: Pruning and Clustering-Based Token Reduction for Faster Visual Language Models

École Polytechnique(巴黎综合理工学院);IP Paris(巴黎理工学院);Université Sorbonne Paris Nord(巴黎北索邦大学);Crédit Agricole S.A(法国农业信贷集团)

Mohamed Dhouib, Davide Buscaldi, Sonia Vanier, Aymen Shabou

AI总结:

针对视觉语言模型因视觉标记冗余导致计算资源消耗大的问题,提出PACT方法,通过剪枝无关标记和合并冗余标记减少推理时间和内存使用,使用新重要性度量和距离受限密度峰值聚类算法,实验验证其有效性。

摘要:

视觉语言模型(VLMs)在推理时需要大量计算资源,因为表示视觉信息需要额外的输入标记。然而,这些视觉标记通常包含冗余和不重要的信息,导致标记数量不必要地增加。为了解决这个问题,我们提出了PACT方法,通过在语言模型的早期层剪枝无关标记并合并视觉冗余标记来减少推理时间和内存使用。我们的方法使用一种新的重要性度量来识别不重要的标记,无需依赖注意力分数,使其与FlashAttention兼容。我们还提出了一种新的聚类算法,称为距离受限密度峰值聚类(Distance Bounded Density Peak Clustering),该算法在通过预定义阈值约束簇内元素之间距离的同时,高效地对视觉标记进行聚类。我们通过广泛的实验证明了PACT的有效性。

英文摘要:

Visual Language Models require substantial computational resources for inference due to the additional input tokens needed to represent visual information. However, these visual tokens often contain redundant and unimportant information, resulting in an unnecessarily high number of tokens. To address this, we introduce PACT, a method that reduces inference time and memory usage by pruning irrelevant tokens and merging visually redundant ones at an early layer of the language model. Our approach uses a novel importance metric to identify unimportant tokens without relying on attention scores, making it compatible with FlashAttention. We also propose a novel clustering algorithm, called Distance Bounded Density Peak Clustering, which efficiently clusters visual tokens while constraining the distances between elements within a cluster by a predefined threshold. We demonstrate the effectiveness of PACT through extensive experiments.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑