通过选择性注意力头实现高效知识保留,显著减少大模型持续学习中的重放数据量。
SEEKR: Selective Attention-Guided Knowledge Retention for Continual Learning of Large Language Models
- 基于注意力权重重要性筛选关键注意力头进行知识蒸馏。
- 仅需1/10甚至1%的重放数据即达现有方法性能水平。
- 适合资源受限场景下的大模型持续学习任务。
持续学习对语言模型动态适应现实需求至关重要。为缓解灾难性遗忘问题,数据重放已被证明是简单有效的策略,后续基于数据重放的蒸馏可进一步提升性能。然而,现有方法未能充分挖掘先前任务中模型所蕴含的知识,导致需要大量重放样本才能取得良好效果。本文首次探讨并强调注意力权重在知识保留中的重要性,提出一种选择性注意力引导的知识保留方法(SEEKR),用于大语言模型的高效数据重放式持续学习。具体而言,SEEKR在选定的注意力头上执行注意力蒸馏,以实现更细粒度的知识保留,其中提出的遗忘度与任务敏感度度量用于识别最具价值的注意力头。在两个大语言模型持续学习基准上的实验结果表明,相较于现有方法,SEEKR在性能与效率方面均具优势:仅需其他方法1/10甚至1%的重放数据即可达到相当或更优的性能。
原文摘要 · Abstract (English)
Continual learning (CL) is crucial for language models to dynamically adapt to the evolving real-world demands. To mitigate the catastrophic forgetting problem in CL, data replay has been proven a simple and effective strategy, and the subsequent data-replay-based distillation can further enhance the performance. However, existing methods fail to fully exploit the knowledge embedded in models from previous tasks, resulting in the need for a relatively large number of replay samples to achieve good results. In this work, we first explore and emphasize the importance of attention weights in knowledge retention, and then propose a SElective attEntion-guided Knowledge Retention method (SEEKR) for data-efficient replay-based continual learning of large language models (LLMs). Specifically, SEEKR performs attention distillation on the selected attention heads for finer-grained knowledge retention, where the proposed forgettability-based and task-sensitivity-based measures are used to identify the most valuable attention heads. Experimental results on two continual learning benchmarks for LLMs demonstrate the superiority of SEEKR over the existing methods on both performance and efficiency. Explicitly, SEEKR achieves comparable or even better performance with only 1/10 of the replayed data used by other methods, and reduces the proportion of replayed data to 1%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。