提出LWF框架,让生成式语言模型学会主动遗忘无关知识
Graceful Forgetting in Generative Language Models
- 用费舍尔信息矩阵加权参数更新,判断哪些知识该丢
- 在微调中定期清除高置信度的无关知识,提升任务表现
- 适合需要精准控制知识保留的生成模型微调场景
预训练-微调范式已成为深度学习的核心方法。然而,预训练阶段获取的部分知识可能对下游任务产生负面影响,即负迁移问题。为解决此问题,优雅遗忘(graceful forgetting)被提出,其核心思想是通过选择性遗忘无关知识来增强目标任务的学习灵活性。然而,该方法在生成式语言模型中仍缺乏探索,且现有遗忘算法常因架构不兼容难以迁移。为此,本文提出学习与遗忘框架(Learning With Forgetting, LWF),利用费舍尔信息矩阵加权参数更新,计算自生成知识的遗忘置信度,并在微调过程中周期性地消除高置信度的知识。实验表明,尽管预训练语言模型中知识交互机制尚不明确,但应用优雅遗忘可有效提升微调性能。
原文摘要 · Abstract (English)
Recently, the pretrain-finetune paradigm has become a cornerstone in various deep learning areas. While in general the pre-trained model would promote both effectiveness and efficiency of downstream tasks fine-tuning, studies have shown that not all knowledge acquired during pre-training is beneficial. Some of the knowledge may actually bring detrimental effects to the fine-tuning tasks, which is also known as negative transfer. To address this problem, graceful forgetting has emerged as a promising approach. The core principle of graceful forgetting is to enhance the learning plasticity of the target task by selectively discarding irrelevant knowledge. However, this approach remains underexplored in the context of generative language models, and it is often challenging to migrate existing forgetting algorithms to these models due to architecture incompatibility. To bridge this gap, in this paper we propose a novel framework, Learning With Forgetting (LWF), to achieve graceful forgetting in generative language models. With Fisher Information Matrix weighting the intended parameter updates, LWF computes forgetting confidence to evaluate self-generated knowledge regarding the forgetting task, and consequently, knowledge with high confidence is periodically unlearned during fine-tuning. Our experiments demonstrate that, although thoroughly uncovering the mechanisms of knowledge interaction remains challenging in pre-trained language models, applying graceful forgetting can contribute to enhanced fine-tuning performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。