无需重训练,通用剪枝让视觉语言模型更轻量
Prune Once: Retraining-Free Task-Agnostic Pruning for Vision-Language Models

- 基于激活变化设计通用重要性评估,不依赖具体任务
- 在高压缩率下保持良好性能,无须重新训练
- 适合资源受限场景的视觉语言模型部署
视觉语言模型(VLMs)通过大规模预训练在多模态任务中表现出卓越泛化能力,但其日益增长的计算与内存需求给受限环境部署带来挑战。现有剪枝方法常依赖特定任务或面向大语言模型的重要性度量,难以适用于无任务信息的通用剪枝场景。本文提出无需重训练的PORTA框架,基于通用校准数据估计激活变化,构建跨模态、任务无关的重要程度度量,有效捕捉特征级表征价值。PORTA还引入自适应稀疏分配机制,依据输出特征变异动态设定各层剪枝比例,避免均匀稀疏带来的性能下降。在CLIP、BLIP和Qwen2-VL等多种VLM架构上验证,PORTA在高稀疏度下仍保持优异下游性能,支持高效模型压缩。代码已开源。
原文摘要 · Abstract (English)
Vision-language models (VLMs) have achieved remarkable generalization across diverse multimodal tasks through large-scale pre-training, yet their rapidly increasing computational and memory requirements pose significant challenges for deployment in constrained environments. Existing pruning strategies often depend on task-specific criteria or LLM-oriented importance measures, making them unsuitable for task-agnostic pruning, where no task-specific samples are available at pruning time and the pruned model remains broadly applicable. We introduce a retraining-free VLM pruning framework called PORTA that derives a task- and modality-agnostic importance formulation based on activation variation, estimated from generic calibration data, which reliably captures feature-level representation utility across modalities. PORTA further incorporates an adaptive sparsity allocation mechanism that assigns layer-wise pruning ratios based on output feature variability, avoiding the limitations of uniform sparsity and reducing performance degradation at high compression levels. Extensive experiments across VLM architectures, such as CLIP, BLIP, and Qwen2-VL, demonstrate that PORTA achieves competitive downstream performance under high sparsity without requiring any retraining, supporting efficient VLM compression. Code is available at https://github.com/cau-hai-lab/PORTA.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。