arXiv:2506.07587cs.LGcs.AI2025-06被引 2

通过剪枝优化参数高效微调配置,降低大模型微调成本。

PrunePEFT: Iterative Hybrid Pruning for Parameter-Efficient Fine-tuning of LLMs

  • 将PEFT搜索转为剪枝问题,迭代剔除冗余模块。
  • 在多个数据集上实现与原方案相当性能,参数量减少30%以上。
  • 适合需要高效微调大模型的研究者和工程团队。

参数高效微调(PEFT)方法已成为微调预训练语言模型的有效途径。相比全参数微调(FFT),PEFT在显著减少可训练参数的同时保持了相近的任务性能,大幅降低了训练与存储成本。然而,使用PEFT需面对庞大的设计空间,如模块类型及插入层的选择。不当配置可能导致次优结果。传统方法如结构搜索虽有效,但常引入巨大额外开销。本文提出一种新方法PrunePEFT,将PEFT策略搜索建模为剪枝问题,引入一种混合剪枝策略,利用不同PEFT模块对剪枝的敏感性差异。该方法通过迭代移除冗余或冲突的PEFT模块,优化微调配置。通过高效识别关键模块,显著降低传统架构搜索带来的计算负担,使大模型微调更具可扩展性和效率。

原文摘要 · Abstract (English)

Parameter Efficient Fine-Tuning (PEFT) methods have emerged as effective and promising approaches for fine-tuning pre-trained language models. Compared with Full parameter Fine-Tuning (FFT), PEFT achieved comparable task performance with a substantial reduction of trainable parameters, which largely saved the training and storage costs. However, using the PEFT method requires considering a vast design space, such as the type of PEFT modules and their insertion layers. Inadequate configurations can lead to sub-optimal results. Conventional solutions such as architectural search techniques, while effective, tend to introduce substantial additional overhead. In this paper, we propose a novel approach, PrunePEFT, which formulates the PEFT strategy search as a pruning problem and introduces a hybrid pruning strategy that capitalizes on the sensitivity of pruning methods to different PEFT modules. This method extends traditional pruning techniques by iteratively removing redundant or conflicting PEFT modules, thereby optimizing the fine-tuned configuration. By efficiently identifying the most relevant modules, our approach significantly reduces the computational burden typically associated with architectural search processes, making it a more scalable and efficient solution for fine-tuning large pre-trained models.

参数高效微调模型剪枝大模型

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