arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2402.00751cs.LGcs.AIcs.CR
收藏

面向大语言模型上下文学习数据的快速精确遗忘

Fast Exact Unlearning for In-Context Learning Data for LLMs

Department of Computer Science, University of Waterloo, Waterloo, Canada(滑铁卢大学计算机科学系);Department of Computer Science, University of Toronto, Toronto, Canada(多伦多大学计算机科学系);Vector Institute, Toronto, Canada(向量研究所)

Andrei I. Muresanu, Anvith Thudi, Michael R. Zhang, Nicolas Papernot

AI总结:

针对大语言模型微调数据的精确删除需求,本文用上下文学习替代 SGD 微调,并以量化 k-means 实现常数时间遗忘,在保持相近性能的同时显著降低遗忘成本。

摘要:

现代机器学习模型的训练成本高昂,而追溯性删除特定训练数据所带来的挑战正受到越来越多关注。在深度学习流程中实现精确遗忘——即生成的模型表现得如同某些数据从未被纳入训练一样——仍是一个尚未解决的问题。本文重新审视深度学习中的精确遗忘,并表明对于大语言模型(LLM),我们能够高效、精确地遗忘“微调数据”(用于适配预训练模型的数据)。这源于两点观察。首先,我们可以使用上下文学习使 LLM 适配微调数据集,而无需采用基于 SGD 的算法。其次,我们表明可以通过量化 k-means 完成准确的上下文学习,从而支持实际上为常数时间的遗忘操作。评估显示,该遗忘方案与基于微调的替代方法具有相近性能,但大幅降低了遗忘成本。我们的研究还强调,当调整学习算法以获得更快的遗忘操作时,需要新的遗忘成本度量。

英文摘要:

Modern machine learning models are expensive to train, and there is a growing concern about the challenge of retroactively removing specific training data. Achieving exact unlearning in deep learning pipelines--producing models as if certain data had never been included in training--remains an open problem. In this paper, we revisit exact unlearning in deep learning and show that for large language models (LLMs) we can efficiently exactly unlearn "fine-tuning data" (the data used to adapt a pre-trained model). This follows from two observations. First, we can use in-context learning to adapt the LLM to the fine-tuning dataset instead of SGD based algorithms. Second, we show that accurate in-context learning can be done with quantized k-means, which allows for effectively constant time unlearning operations. Our evaluation shows that this unlearning recipe has similar performance to fine-tuning alternatives, but vastly reduces the unlearning costs. Our study also highlights the need for new measures of unlearning cost when adapting the learning algorithm to have faster unlearn operations.

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

↑