arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.08096cs.LG
收藏

多阶段微调中缓解灾难性遗忘的高效复现方案

An Efficient Rehearsal Scheme for Catastrophic Forgetting Mitigation during Multi-stage Fine-tuning

University of California, Los Angeles(加利福尼亚大学洛杉矶分校);Google Inc(谷歌公司);Google Cloud AI Research(谷歌云人工智能研究院)

Andrew Bai, Chih-Kuan Yeh, Cho-Jui Hsieh, Ankur Taly

AI总结:

针对多阶段微调中灾难性遗忘问题,提出mix-cd采样方案,在固定计算预算下优先复现“附带损害”样本,无需额外模型推理即可高效估计样本密度,计算高效且优于现有持续学习方法。

摘要:

在自然语言处理中,在新任务或领域上增量微调基础模型现已成为事实上的方法。这种方法的一个已知缺陷是微调期间发生的对先前知识的灾难性遗忘。缓解这种遗忘的常见方法是在微调期间复现来自先前任务的样本。一些现有工作假设使用固定内存缓冲区存储先前任务示例,同时依赖当前模型的推理(前向传播)从缓冲区中选择复现样本。然而,鉴于模型推理的计算成本不断增加,而数据存储成本不断降低,我们关注的场景是使用固定计算预算而非固定内存预算来复现样本。我们提出一种名为mix-cd的采样方案,该方案优先复现“附带损害”样本——即先前模型能正确预测但增量微调后的模型遗忘的样本。我们方案的关键是一种无需额外模型推理即可高效估计附带损害样本密度的程序。我们的方法计算高效、易于实现,并且在计算受限的场景中优于多种领先的持续学习方法。所有代码将在https://github.com/jybai/mix-cd-rehearsal公开。

英文摘要:

Incrementally fine-tuning foundational models on new tasks or domains is now the de facto approach in NLP. A known pitfall of this approach is the \emph{catastrophic forgetting} of prior knowledge that happens during fine-tuning. A common approach to alleviate such forgetting is to rehearse samples from prior tasks during fine-tuning. Several existing works assume a fixed memory buffer to store prior task examples, while relying on inferences (forward passes) with the model at hand for choosing examples for rehearsal from the buffer. However, given the increasing computational cost of model inference, and decreasing cost of data storage, we focus on the setting to rehearse samples with a fixed computational budget instead of a fixed memory budget. We propose a sampling scheme, \texttt{\bf mix-cd}, that prioritizes rehearsal of ``collateral damage'' samples, which are samples predicted correctly by the prior model but forgotten by the incrementally tuned one. The crux of our scheme is a procedure to efficiently estimate the density of collateral damage samples without incurring additional model inferences. Our approach is computationally efficient, easy to implement, and outperforms several leading continual learning methods in compute-constrained settings. All the code will be publicly available at https://github.com/jybai/mix-cd-rehearsal.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑