通过神经参数搜索,让微调模型更小更强,迁移能力提升
Neural Parameter Search for Slimmer Fine-Tuned Models and Better Transfer
- 基于任务向量差异,低秩空间中搜索关键参数进行剪枝
- 剪枝后模型性能接近原版,存储减少70%以上
- 适合需要高效迁移与部署的视觉/NLP/多模态应用
基础模型及其检查点显著推动了深度学习发展,但在特定领域外微调模型常表现不佳且存在冗余。研究表明,将剪枝后的微调模型与原始预训练模型结合,可缓解遗忘、降低跨任务参数融合干扰,并提升压缩效率。为此,本文提出一种新方法NPS-Pruning,利用任务向量机制,通过计算微调模型与原始模型的差异,在低秩子空间中搜索关键神经参数以实现高效剪枝。该方法在三方面具有应用价值:支持双模型插值增强知识迁移,促进模型合并实现有效知识融合,以及部署压缩后仍保持近似原始性能的轻量模型。在视觉、自然语言处理及多模态基准上的大量实验验证了方法的有效性与鲁棒性,性能显著提升。代码已公开于https://github.com/duguodong7/NPS-Pruning。
原文摘要 · Abstract (English)
Foundation models and their checkpoints have significantly advanced deep learning, boosting performance across various applications. However, fine-tuned models often struggle outside their specific domains and exhibit considerable redundancy. Recent studies suggest that combining a pruned fine-tuned model with the original pre-trained model can mitigate forgetting, reduce interference when merging model parameters across tasks, and improve compression efficiency. In this context, developing an effective pruning strategy for fine-tuned models is crucial. Leveraging the advantages of the task vector mechanism, we preprocess fine-tuned models by calculating the differences between them and the original model. Recognizing that different task vector subspaces contribute variably to model performance, we introduce a novel method called Neural Parameter Search (NPS-Pruning) for slimming down fine-tuned models. This method enhances pruning efficiency by searching through neural parameters of task vectors within low-rank subspaces. Our method has three key applications: enhancing knowledge transfer through pairwise model interpolation, facilitating effective knowledge fusion via model merging, and enabling the deployment of compressed models that retain near-original performance while significantly reducing storage costs. Extensive experiments across vision, NLP, and multi-modal benchmarks demonstrate the effectiveness and robustness of our approach, resulting in substantial performance gains. The code is publicly available at: https://github.com/duguodong7/NPS-Pruning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。