arXiv:2511.19518cs.CVcs.AI2025-11被引 1

基于信息论的剪枝方法,让视觉语言模型更轻更快

Towards Efficient VLMs: Information-Theoretic Driven Compression via Adaptive Structural Pruning

  • 用信息瓶颈原理衡量注意力头重要性,动态剪枝
  • 压缩后减少3.2倍计算量,速度提升1.8倍,性能几乎不变
  • 适合追求高效部署的多模态大模型研究者

视觉语言模型(VLMs)在多模态任务中表现卓越,但模型规模持续增长导致部署效率低下。现有压缩方法依赖启发式指标或经验规则,缺乏信息保留的理论保障。本文提出InfoPrune,一种基于信息论的自适应结构压缩框架。基于信息瓶颈原理,将剪枝建模为保留任务相关语义与剔除冗余依赖之间的权衡。通过引入基于熵的有效秩(eRank)和柯尔莫哥洛夫-斯米尔诺夫(KS)距离,量化每个注意力头的贡献,形成兼顾结构稀疏性与信息效率的统一标准。在此基础上,设计两种互补方案:(1) 基于训练的注意力头剪枝,以信息损失为目标优化;(2) 无需训练的前馈网络压缩,采用自适应低秩近似。在VQAv2、TextVQA和GQA上的实验表明,InfoPrune可实现最高3.2倍的FLOP减少和1.8倍加速,性能下降可忽略,为高效多模态大模型提供了理论坚实且实用有效的路径。

原文摘要 · Abstract (English)

Recent advances in vision-language models (VLMs) have shown remarkable performance across multimodal tasks, yet their ever-growing scale poses severe challenges for deployment and efficiency. Existing compression methods often rely on heuristic importance metrics or empirical pruning rules, lacking theoretical guarantees about information preservation. In this work, we propose InfoPrune, an information-theoretic framework for adaptive structural compression of VLMs. Grounded in the Information Bottleneck principle, we formulate pruning as a trade-off between retaining task-relevant semantics and discarding redundant dependencies. To quantify the contribution of each attention head, we introduce an entropy-based effective rank (eRank) and employ the Kolmogorov--Smirnov (KS) distance to measure the divergence between original and compressed structures. This yields a unified criterion that jointly considers structural sparsity and informational efficiency. Building on this foundation, we further design two complementary schemes: (1) a training-based head pruning guided by the proposed information loss objective, and (2) a training-free FFN compression via adaptive low-rank approximation. Extensive experiments on VQAv2, TextVQA, and GQA demonstrate that InfoPrune achieves up to 3.2x FLOP reduction and 1.8x acceleration with negligible performance degradation, establishing a theoretically grounded and practically effective step toward efficient multimodal large models.

模型压缩信息论视觉语言模型

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