通过结构化剪枝与高效恢复训练,低成本压缩多模态大模型。
Investigating Structural Pruning and Recovery Techniques for Compressing Multimodal Large Language Models: An Empirical Study
- 对语言模型主干进行层间和宽度剪枝,结合微调与知识蒸馏恢复性能。
- 仅用5%数据即可保留95%原始性能,小压缩比下只需微调多模态投影器。
- 适合资源受限场景,为实际部署多模态大模型提供实证指导。
多模态大语言模型(MLLMs)虽能力强大,但计算与内存开销大,难以实用部署。现有参数缩减方法多依赖从小语言模型训练,灵活性差且计算成本高。本文提出直接对已有MLLMs进行结构化剪枝并配合高效恢复训练。研究了应用于MLLM语言模型主干的两种剪枝范式——层间剪枝与宽度剪枝,结合监督微调与知识蒸馏。同时评估仅用少量数据进行恢复训练的可行性。结果表明:在低资源环境下,宽度剪枝表现更优;小压缩比(<20%)时,仅微调多模态投影器即可;结合监督微调与隐藏状态蒸馏可在不同剪枝水平下实现最佳恢复效果。尤其值得注意的是,仅需5%原始训练数据,即可保留超过95%的原始性能。在两个代表性模型LLaVA-v1.5-7B和Bunny-v1.0-3B上的实证研究,为无大量算力或数据条件下的MLLM压缩提供了可操作的实践建议。
原文摘要 · Abstract (English)
While Multimodal Large Language Models (MLLMs) demonstrate impressive capabilities, their substantial computational and memory requirements pose significant barriers to practical deployment. Current parameter reduction techniques primarily involve training MLLMs from Small Language Models (SLMs), but these methods offer limited flexibility and remain computationally intensive. To address this gap, we propose to directly compress existing MLLMs through structural pruning combined with efficient recovery training. Specifically, we investigate two structural pruning paradigms--layerwise and widthwise pruning--applied to the language model backbone of MLLMs, alongside supervised finetuning and knowledge distillation. 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 with limited computational resources or insufficient finetuning data. As for the recovery training, finetuning only the multimodal projector is sufficient at small compression levels (< 20%). Furthermore, a combination of supervised finetuning and hidden-state distillation yields optimal recovery across various pruning levels. Notably, effective recovery can be achieved with as little as 5% of the original training data, while retaining over 95% of the original performance. Through empirical study on two representative MLLMs, i.e., LLaVA-v1.5-7B and Bunny-v1.0-3B, this study offers actionable insights for practitioners aiming to compress MLLMs effectively without extensive computation resources or sufficient data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。