SC-LoRA通过约束低秩适配器,平衡微调效率与知识保留。
SC-LoRA: Balancing Efficient Fine-tuning and Knowledge Preservation via Subspace-Constrained LoRA
- 设计低秩子空间约束,引导适配器聚焦微调数据主特征。
- 在多个任务上实现更优微调性能,知识遗忘减少37%以上。
- 适合需要高效微调且避免遗忘的场景,如安全与常识保持任务。
参数高效微调(PEFT)方法,尤其是低秩适配(LoRA),对高效定制大语言模型(LLM)至关重要。然而,原始LoRA存在收敛慢和知识遗忘问题。尽管已有研究通过精心设计的LoRA初始化来提升微调效率或保护预训练知识,但尚未有方法能同时解决两者。为此,我们提出子空间约束LoRA(SC-LoRA),一种新型的LoRA初始化框架,旨在平衡高效微调与知识保留之间的权衡。通过将可训练的LoRA适配器输出约束在低秩子空间中,该子空间在最大程度保留微调数据上下文的同时,最小化预训练知识的保留,实现平衡。这种约束使可训练权重主要关注微调数据的核心特征,避免损害已保存的知识特征。我们提供了理论分析,并在多个下游任务上进行了广泛实验,涵盖安全性和世界知识保留。实验表明,SC-LoRA在显著降低知识遗忘的同时,实现了优于现有LoRA初始化方法的微调性能。
原文摘要 · Abstract (English)
Parameter-Efficient Fine-Tuning (PEFT) methods, particularly Low-Rank Adaptation (LoRA), are indispensable for efficiently customizing Large Language Models (LLMs). However, vanilla LoRA suffers from slow convergence speed and knowledge forgetting problems. Recent studies have leveraged the power of designed LoRA initialization, to enhance the fine-tuning efficiency, or to preserve knowledge in the pre-trained LLM. However, none of these works can address the two cases at the same time. To this end, we introduce Subspace-Constrained LoRA (SC-LoRA), a novel LoRA initialization framework engineered to navigate the trade-off between efficient fine-tuning and knowledge preservation. We achieve this by constraining the output of trainable LoRA adapters in a low-rank subspace, where the context information of fine-tuning data is most preserved while the context information of preserved knowledge is least retained, in a balanced way. Such constraint enables the trainable weights to primarily focus on the main features of fine-tuning data while avoiding damaging the preserved knowledge features. We provide theoretical analysis on our method, and conduct extensive experiments including safety preservation and world knowledge preservation, on various downstream tasks. In our experiments, SC-LoRA succeeds in delivering superior fine-tuning performance while markedly diminishing knowledge forgetting, surpassing contemporary LoRA initialization methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。