提出DiffoRA,让大模型只对关键模块微调,节省参数且效果更好。
DiffoRA: Enabling Parameter-Efficient Fine-Tuning via Differential Module Selection
- 用可学习的DAM矩阵自动选择该微调的模块
- 在多个基准上达到当前最佳性能,参数量极小
- 适合资源有限但追求高精度的模型微调场景
参数高效微调(PEFT)方法被广泛研究用于大语言模型的下游任务。现有方法中,低秩适应(LoRA)因其简洁设计而流行,通过将低秩矩阵引入预训练模型实现微调。然而,LoRA及其自适应优化版本通常对所有模块分配相同矩阵,或基于重要性评分调整内部秩。本文认为,并非所有模型模块都需微调。受此启发,我们提出一种新方案DiffoRA,实现低秩分解矩阵的自适应选择。核心是差分适应矩阵(DAM),用于判断哪些模块最适合作微调。我们从理论上分析了该矩阵对模型收敛速度和泛化能力的影响,并通过连续松弛与离散化结合、权重共享优化构建DAM。我们完整实现了DiffoRA,并设计了全面实验评估其性能。结果表明,DiffoRA在多个基准上均达到当前最优表现。
原文摘要 · Abstract (English)
The Parameter-Efficient Fine-Tuning (PEFT) methods have been extensively researched for large language models in downstream tasks. Among all the existing approaches, the Low-Rank Adaptation (LoRA) has gained popularity for its streamlined design by incorporating low-rank matrices into existing pre-trained models. Though effective, LoRA, as well as its adaptive optimizations, either allocate the same matrix to all the modules or adjust the interior rank of the components based on importance scoring indicators. In this paper, we argue that not all the modules in LLMs are suitable and necessary to be fine-tuned. Enlightened by this insight, we propose a new PEFT scheme called DiffoRA, which enables adaptive adoption of the low-rank decomposition matrices. At the core of DiffoRA lies a Differential Adaptation Matrix (DAM) to determine which module is the most suitable and essential for fine-tuning. We theoretically explain how the designed matrix impacts the convergence rate and generalization capability of a pre-trained model. We then construct the DAM via continuous relaxation and discretization with weight-sharing optimizations. We fully implement DiffoRA and design comprehensive experiments to evaluate its performance. The experimental results demonstrate that DiffoRA delivers state-of-the-art results across multiple benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。