提出SECURA方法,解决大模型微调时遗忘问题并提升性能。
SECURA: Sigmoid-Enhanced CUR Decomposition with Uninterrupted Retention and Low-Rank Adaptation in Large Language Models
- 用改进的奇异值分解实现参数高效微调,增强权重保留能力。
- 在多个任务上平均提升3.59%(多选题)和2.51%(问答题)表现。
- 特别适合需要持续学习的场景,如长期知识维护与更新。
随着大语言模型快速发展,全量微调因计算成本过高已难实现,且易引发灾难性遗忘。为此,低秩适应(LoRA)被提出,仅微调少量参数即可接近全量微调效果。但其仍继承原有设计缺陷,存在遗忘风险。本文提出新型参数高效微调方法SECURA:Sigmoid增强的CUR分解低秩适应,引入S-MagNorm归一化机制,提升参数保留与微调效率。在数学推理(GSM8K)、复杂问答(CNNDM)、翻译(NewsDE)及多选推理(LogiQA)等任务上验证,对Gemma2 2B、Qwen2 1.5B、Qwen2 7B、Llama3 8B、Llama3.1 8B五种模型,在四类多选题任务中平均提升3.59%,五类问答任务平均提升2.51%。在16次连续学习测试中达最优表现,保持超过70%基础知识准确率,优于经验回放(ER)、顺序学习(SEQ)、EWC、I-LoRA和CUR-LoRA。
原文摘要 · Abstract (English)
With the rapid development of large language models (LLMs), fully fine-tuning (FT) these models is becoming increasingly infeasible due to high computational demands. Moreover, FT also increases the risk of catastrophic forgetting. As an alternative, Low-Rank Adaptation (LoRA) has been proposed. By fine-tuning only a small subset of parameters, LoRA achieves performance similar to FT while significantly reducing resource requirements. However, since LoRA inherits FT's design, the issue of catastrophic forgetting still remains. To address these limitations, we propose SECURA: Sigmoid-Enhanced CUR Decomposition LoRA, a novel PEFT variant designed to mitigate catastrophic forgetting while improving fine-tuning performance. Our method introduces a novel normalization technique, Sigmoid-based Magnitude Norm (S-MagNorm), which enhances parameter retention and fine-tuning efficiency. SECURA has been evaluated on a diverse range of tasks, including mathematical problem-solving (GSM8K), complex question-answering (CNNDM), translation (NewsDE), and complex multiple-choice reasoning (LogiQA). Experimental results demonstrate that it achieves an average fine-tuning improvement of 3.59% across four MCQ tasks and 2.51% across five QA tasks on Gemma2 2B, Qwen2 1.5B, Qwen2 7B, Llama3 8B, and Llama3.1 8B, outperforming DoRA. Additionally, SECURA demonstrates superior knowledge retention capabilities, achieving state-of-the-art performance in 16 continual learning tests and maintaining more than 70% accuracy on LLMs' basic knowledge compared to Experience Replay (ER), sequential learning (SEQ), EWC, I-LoRA, and CUR-LoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。