通过结构化剪枝与轻量恢复训练,高效压缩视觉语言模型。
Structural Pruning of Large Vision Language Models: A Comprehensive Study on Pruning Dynamics, Recovery, and Data Efficiency

- 对语言模型主干采用层间与宽度剪枝,提升压缩灵活性。
- 仅用5%数据即可恢复95%以上性能,适合资源受限场景。
- 结合微调与隐状态蒸馏,实现多剪枝率下的最优恢复效果。
大型视觉语言模型(LVLMs)虽具备强大能力,但其高计算与内存开销制约了在边缘设备上的部署。现有参数缩减方法多从小型语言模型训练开始,灵活性差且计算成本高。本文提出一种互补路径:对已有LVLM进行结构化剪枝(层间与宽度剪枝),再通过轻量恢复训练压缩。我们评估了监督微调与基于输出概率和隐藏状态的知识蒸馏策略,并考察仅用少量数据进行恢复训练的可行性。结果表明,在低资源场景下,宽度剪枝表现更优;在低压缩比时,仅微调多模态投影器即可;而结合微调与隐藏状态蒸馏可跨剪枝率实现最佳恢复。值得注意的是,仅使用5%原始数据即可保持超过95%的原始性能。本研究在三个3B至7B参数量级的代表性LVLM家族上展开实证分析,为无大量计算资源或数据条件下的模型压缩提供实用指导。代码已开源:https://github.com/YiranHuangIrene/VLMCompression.git。
原文摘要 · Abstract (English)
While Large Vision Language Models (LVLMs) demonstrate impressive capabilities, their substantial computational and memory requirements pose deployment challenges on resource-constrained edge devices. Current parameter reduction techniques primarily involve training LVLMs from small language models, but these methods offer limited flexibility and remain computationally intensive. We study a complementary route: compressing existing LVLMs by applying structured pruning to the language model backbone, followed by lightweight recovery training. Specifically, we investigate two structural pruning paradigms: layerwise and widthwise pruning, and pair them with supervised finetuning and knowledge distillation on logits and hidden states. Additionally, we assess the feasibility of conducting recovery training with only a small fraction of the available data. Our results show that widthwise pruning generally maintains better performance in low-resource scenarios, where computational resources are limited or there is insufficient finetuning data. As for the recovery training, finetuning only the multimodal projector is sufficient at small compression levels. Furthermore, a combination of supervised finetuning and hidden-state distillation yields optimal recovery across various pruning levels. Notably, effective recovery can be achieved using just 5% of the original data, while retaining over 95% of the original performance. Through empirical study on three representative LVLM families ranging from 3B to 7B parameters, this study offers actionable insights for practitioners to compress LVLMs without extensive computation resources or sufficient data. The code base is available at https://github.com/YiranHuangIrene/VLMCompression.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。