arXiv:2503.15369cs.CV2025-03CVPR被引 11

仅用64个样本实现大模型高效剪枝,兼顾准确率与推理速度。

EfficientLLaVA:Generalizable Auto-Pruning for Large Vision-language Models

  • 基于结构风险最小化设计可泛化的剪枝策略搜索方法。
  • 在ScienceQA上达83.05%准确率,推理速度提升1.8倍。
  • 适合资源受限设备部署多模态大模型的开发者。

虽然多模态大语言模型在复杂推理任务中表现优异,但在部署时面临模型复杂度高、资源消耗大的挑战,尤其在资源受限设备上。本文提出一种面向大视觉-语言模型的自动剪枝方法,以提升多模态推理效率。传统方法依赖原模型训练数据选择剪枝比例,但因网络规模大、语料覆盖广,搜索成本过高而难以应用。相比之下,本方法仅需少量样本(64个)即可通过最大化剪枝策略在未知数据上的泛化能力,实现精度与效率的最优平衡。我们基于结构风险最小化原理建模剪枝策略的泛化差距,结合任务性能与泛化能力,在给定搜索空间内迭代优化剪枝策略,并通过优化视觉投影器动态扩展更高性能上限的搜索空间。在ScienceQA、Vizwiz、MM-vet和LLaVA-Bench四个视觉问答数据集上进行了大量实验。结果表明,仅使用64个样本进行剪枝策略搜索,EfficientLLaVA在ScienceQA上达到83.05%准确率,相比dense LLaVA-v1.5-7B模型实现×1.8的加速。

原文摘要 · Abstract (English)

While multimodal large language models demonstrate strong performance in complex reasoning tasks, they pose significant challenges related to model complexity during deployment, especially for resource-limited devices. In this paper, we propose an automatic pruning method for large vision-language models to enhance the efficiency of multimodal reasoning. Conventional methods rely on the training data of the original model to select the proper pruning ratio for different network components. However, these methods are impractical for large vision-language models due to the unaffordable search costs caused by web-scale training corpus. In contrast, our approach only leverages a small number of samples to search for the desired pruning policy by maximizing its generalization ability on unknown training data while maintaining the model accuracy, which enables the achievement of an optimal trade-off between accuracy and efficiency for large visual language models. Specifically, we formulate the generalization gap of the pruning strategy using the structural risk minimization principle. Based on both task performance and generalization capability, we iteratively search for the optimal pruning policy within a given search space and optimize the vision projector to evolve the search space with higher upper bound of performance. We conduct extensive experiments on the ScienceQA, Vizwiz, MM-vet, and LLaVA-Bench datasets for the task of visual question answering. Using only 64 samples for pruning policy search, EfficientLLaVA achieves an accuracy of 83.05% on ScienceQA, along with a $\times$ 1.8 speedup compared to the dense LLaVA-v1.5-7B model.

模型剪枝多模态效率优化

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