基于梯度与原始值选择参数,实现更稳定高效的低秩微调
GaLLoP: Gradient-based Sparse Learning on Low-Magnitude Parameters
- 根据下游任务梯度大、预训练值小的参数优先微调
- 在LLaMA3 8B和Gemma 2B上性能优于或匹配主流方法
- 有效防止灾难性遗忘,适合追求鲁棒性的微调场景
稀疏微调通过仅调整模型参数的子集来适配下游任务。然而,其效果取决于是否能最优选择待微调参数。本文提出新型稀疏微调方法GaLLoP:仅对下游任务中梯度幅值最大、预训练值最小的参数进行微调,直观上优先选择任务相关性强但对预训练知识干扰小的参数。在LLaMA3 8B和Gemma 2B作为基础模型的实验中,GaLLoP始终在分布内与分布外性能上优于或匹配现有主流参数高效微调技术(包括LoRA、DoRA和SAFT)。分析表明,GaLLoP缓解了灾难性遗忘与任务数据记忆问题,因关键预训练参数保持不变,并在多数随机种子下表现出更稳定的性能,泛化能力更强。
原文摘要 · Abstract (English)
Sparse fine-tuning techniques adapt LLMs to downstream tasks by only tuning a sparse subset of model parameters. However, the effectiveness of sparse adaptation depends on optimally selecting the model parameters to be fine-tuned. In this work, we introduce a novel sparse fine-tuning technique named GaLLoP: Gradient-based Sparse Learning on Low-Magnitude Parameters, which fine-tunes only those model parameters which have the largest gradient magnitudes on downstream tasks and the smallest pre-trained magnitudes, intuitively prioritizing parameters that are highly task-relevant, but minimally disruptive to pre-trained knowledge. Our experimentation with LLaMA3 8B and Gemma 2B as base models shows that GaLLoP consistently improves or matches the in-distribution as well as out-of-distribution performance obtained via the usage of other leading parameter-efficient fine-tuning techniques, including LoRA, DoRA, and SAFT. Our analysis demonstrates that GaLLoP mitigates catastrophic forgetting and memorization of task data, as important pre-trained parameters remain unchanged, and stabilizes performance relative to other fine-tuning techniques, robustly generalizing across most random seeds.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。