通过参数隔离与组合,缓解语言模型持续学习中的遗忘问题。
Neural Networks Remember More: The Power of Parameter Isolation and Combination
- 用参数隔离避免任务间干扰,保持知识独立性
- 采用任务算术合并参数,提升整体性能
- 在多个基准上超越现有最佳方法
持续学习中,预训练语言模型面临灾难性遗忘问题,即在顺序学习新任务时丢失旧知识。模型保持旧知识的能力称为稳定性,适应新任务的能力称为可塑性。因此,关键在于平衡两者的权衡。本文提出一种新方法,通过参数隔离与后续组合策略,在缓解灾难性遗忘的同时提升模型性能。具体而言,在训练阶段,使用参数隔离方法使模型适应每个下游任务,防止不同任务间的潜在干扰;随后,利用任务算术方法合并所有已训练参数,将整合后的知识注入主干模型。在多个持续语言学习基准上的实证评估表明,该方法显著优于现有最先进方法。
原文摘要 · Abstract (English)
Catastrophic forgetting is a pervasive issue for pre-trained language models (PLMs) during continual learning, where models lose previously acquired knowledge when sequentially trained on a series of tasks. The model's ability to retain old tasks is referred to as stability, while its adaptability to new tasks is called plasticity. Therefore, the key to solving this problem is to find a trade-off between the plasticity and stability of the model. To address this issue, in this paper, we propose a novel method to achieve a balance between model stability and plasticity, thereby mitigating catastrophic forgetting. More specifically, our proposed approach leverages parameter isolation and a subsequent combination strategy. Initially, in the training stage, the model adapts to each downstream task via a parameter isolation method to prevent potential interference among different tasks. We then combine all trained parameters, which contain acquired knowledge, using the task arithmetic method and finally apply them to the backbone model. Empirical evaluations on continual language learning benchmarks substantiate the effectiveness of our approach, revealing a marked enhancement over existing state-of-the-art approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。