通过感知知识相关性动态激活奇异值,提升大模型微调效率与性能。
KaSA: Knowledge-Aware Singular-Value Adaptation of Large Language Models
- 基于奇异值分解,按任务相关性动态激活知识
- 在16个基准上超越14种主流微调方法,平均提升显著
- 适合需高效适配的场景,尤其对知识噪声敏感的任务
大型语言模型(LLMs)规模不断增大,导致在特定任务或领域上微调时计算开销和内存占用过高。参数高效微调(PEFT)方法通过仅训练少量参数来缓解此问题,其中LoRA因其简洁高效而广受关注,并催生了众多变体。然而,这类方法忽略与目标任务无关或噪声化的知识,影响模型表现并导致次优结果。为此,我们提出知识感知奇异值自适应(KaSA),利用奇异值分解(SVD)结合知识感知的奇异值,根据任务相关性动态激活知识。我们在多种LLM上对自然语言理解(NLU)、生成(NLG)、指令遵循和常识推理等任务进行了广泛实验。结果显示,KaSA在16个基准和4个合成数据集上持续优于FFT及14种主流PEFT基线,验证了其有效性与适应性。代码已开源:https://github.com/juyongjiang/KaSA。
原文摘要 · Abstract (English)
The increasing sizes of large language models (LLMs) result in significant computational overhead and memory usage when adapting these models to specific tasks or domains. Various parameter-efficient fine-tuning (PEFT) methods have been devised to mitigate these challenges by training a small set of parameters for the task-specific updates of the model weights. Among PEFT methods, LoRA stands out for its simplicity and efficiency, inspiring the development of a series of variants. However, LoRA and its successors disregard the knowledge that is noisy or irrelevant to the targeted task, detrimentally impacting model performance and leading to suboptimality. To address this limitation, we introduce Knowledge-aware Singular-value Adaptation (KaSA), a PEFT method that leverages singular value decomposition (SVD) with knowledge-aware singular values to dynamically activate knowledge based on its relevance to the task at hand. We conduct extensive experiments across a range of LLMs on tasks spanning natural language understanding (NLU), generation (NLG), instruction following, and commonsense reasoning. The experimental results demonstrate that KaSA consistently outperforms FFT and 14 popular PEFT baselines across 16 benchmarks and 4 synthetic datasets, underscoring our method's efficacy and adaptability. The source code of our method is available at https://github.com/juyongjiang/KaSA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。