arXiv:2412.17383cs.CL2024-12中稿 · AAAI被引 1

通过双模型记忆融合,缓解大模型微调中的遗忘问题。

Interweaving Memories of a Siamese Large Language Model

  • 构建孪生模型,分别提取原始与微调后的记忆
  • 采用交织机制动态融合双记忆,提升生成质量
  • 适配主流开源模型和微调方法,通用性强

参数高效微调(PEFT)通过少量参数调整优化大语言模型(LLM),以适应下游任务。但易引发灾难性遗忘,导致新知识覆盖原有世界知识。为此,我们提出一种模型无关的PEFT框架IMSM,即「孪生大模型记忆交织」。该框架在现有PEFT基础上,引入孪生结构:给定输入查询时,模型分别基于预训练和微调参数生成两种记忆;通过交织机制调控两者贡献,生成下一词元。该方法理论适用于所有开源LLM及现有PEFT方法。我们在多个基准数据集上评估主流开源模型表现,结果表明,IMSM在时间与空间效率上与基础PEFT方法相当,显著提升性能并有效缓解灾难性遗忘。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) methods optimize large language models (LLMs) by modifying or introducing a small number of parameters to enhance alignment with downstream tasks. However, they can result in catastrophic forgetting, where LLMs prioritize new knowledge at the expense of comprehensive world knowledge. A promising approach to mitigate this issue is to recall prior memories based on the original knowledge. To this end, we propose a model-agnostic PEFT framework, IMSM, which Interweaves Memories of a Siamese Large Language Model. Specifically, our siamese LLM is equipped with an existing PEFT method. Given an incoming query, it generates two distinct memories based on the pre-trained and fine-tuned parameters. IMSM then incorporates an interweaving mechanism that regulates the contributions of both original and enhanced memories when generating the next token. This framework is theoretically applicable to all open-source LLMs and existing PEFT methods. We conduct extensive experiments across various benchmark datasets, evaluating the performance of popular open-source LLMs using the proposed IMSM, in comparison to both classical and leading PEFT methods. Our findings indicate that IMSM maintains comparable time and space efficiency to backbone PEFT methods while significantly improving performance and effectively mitigating catastrophic forgetting.

大模型微调记忆融合防遗忘

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。