arXiv:2410.02744cs.CLcs.AI2024-10ICML被引 2

提出中性残差机制,让大模型扩展新领域时不遗忘旧知识。

Neutral Residues: Revisiting Adapters for Model Extension

  • 通过修改适配器结构,使新增残差块在原领域输出接近零
  • 在英译新语言任务中显著优于微调和LoRA,兼顾新旧任务性能
  • 适合需要持续学习、避免灾难性遗忘的模型扩展场景

我们解决将预训练大语言模型扩展到训练中未见新领域的问题。标准方法如微调或低秩适配(LoRA)虽能实现领域适应,但未正式增加模型容量,常导致新领域表现与原领域性能之间的权衡。本文从数据、架构和训练流程三方面重新审视并改进适配器,提出中性残差方法:通过设计使每个新增残差块在原领域输出趋近于零。该方法在将最先进的英文模型扩展至新语言时表现优异,显著优于微调、LoRA及传统适配器,在学习新语言的同时有效保留英语性能。

原文摘要 · Abstract (English)

We address the problem of extending a pretrained large language model to a new domain that was not seen during training. Standard techniques, such as finetuning or low-rank adaptation (LoRA) are successful at domain adaptation, but do not formally add capacity to the model. This often leads to a trade-off, between performing well on the new domain vs. degrading performance on the original domain. Here, we revisit and improve adapters to extend LLMs from three angles: data, architecture and training procedure, which are advantageously considered jointly. The resulting method, called neutral residues, modifies adapters in a way that leads each new residual block to output near-zeros on the original domain. This solution leads to strong results when adapting a state-of-the-art model originally trained on English to a new language. Neutral residues significantly outperform competing approaches such as finetuning, LoRA or vanilla adapters in terms of the trade-off between learning the new language and not forgetting English.

模型扩展适配器持续学习语言迁移

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