arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.08607cs.CLcs.AI
收藏

反转遗忘-保留目标:一种基于 Logit 差值的高效 LLM 遗忘框架

Reversing the Forget-Retain Objectives: An Efficient LLM Unlearning Framework from Logit Difference

UC Santa Barbara(加州大学圣巴巴拉分校);MIT-IBM Watson AI Lab(麻省理工学院-IBM沃森人工智能实验室);Cisco Research(思科研究部);Michigan State University(密歇根州立大学)

Jiabao Ji, Yujian Liu, Yang Zhang, Gaowen Liu, Ramana Rao Kompella, Sijia Liu, Shiyu Chang

AI总结:

提出 ULD 框架,通过辅助 LLM 反转遗忘与保留目标并利用 logit 差值实现高效 LLM 遗忘,在保持效用的同时将训练时间缩短逾三倍。

摘要:

随着大语言模型(LLM)展现出从文档中学习的广泛能力,LLM 遗忘成为一个日益重要的研究方向,用于解决 LLM 在隐私、版权等方面的担忧。传统的 LLM 遗忘任务通常包含两个目标:(1)目标 LLM 应当遗忘指定遗忘文档中的知识;(2)它应当保留 LLM 拥有的其他知识,为此我们假设可以获取少量保留文档。为实现这两个目标,一类主流 LLM 遗忘方法引入了结合两个目标的优化框架——最大化遗忘文档上的预测损失,同时最小化保留文档上的预测损失,但这面临两个挑战:输出退化和灾难性遗忘。本文提出一种名为 Unlearning from Logit Difference(ULD,基于 Logit 差值的遗忘)的新型遗忘框架,它引入一个辅助 LLM,其目标与遗忘目标相反:记住遗忘文档并遗忘保留知识。随后,ULD 通过计算目标 LLM 与辅助 LLM 之间的 logit 差值来推导出被遗忘的 LLM。我们表明,这种反转目标会自然解决上述两个挑战,同时显著提升训练效率。大量实验证明,我们的方法高效实现了预期遗忘,同时保持 LLM 的整体能力,并将训练时间缩短三倍以上。值得注意的是,我们的方法在 ToFU 基准上损失 0% 的模型效用,而基线方法为达到相当的遗忘质量,平均可能牺牲 17% 的效用。我们的代码将在 https://github.com/UCSB-NLP-Chang/ULD 公开发布。

英文摘要:

As Large Language Models (LLMs) demonstrate extensive capability in learning from documents, LLM unlearning becomes an increasingly important research area to address concerns of LLMs in terms of privacy, copyright, etc. A conventional LLM unlearning task typically involves two goals: (1) The target LLM should forget the knowledge in the specified forget documents, and (2) it should retain the other knowledge that the LLM possesses, for which we assume access to a small number of retain documents. To achieve both goals, a mainstream class of LLM unlearning methods introduces an optimization framework with a combination of two objectives - maximizing the prediction loss on the forget documents while minimizing that on the retain documents, which suffers from two challenges, degenerated output and catastrophic forgetting. In this paper, we propose a novel unlearning framework called Unlearning from Logit Difference (ULD), which introduces an assistant LLM that aims to achieve the opposite of the unlearning goals: remembering the forget documents and forgetting the retain knowledge. ULD then derives the unlearned LLM by computing the logit difference between the target and the assistant LLMs. We show that such reversed objectives would naturally resolve both aforementioned challenges while significantly improving the training efficiency. Extensive experiments demonstrate that our method efficiently achieves the intended forgetting while preserving the LLM's overall capabilities, reducing training time by more than threefold. Notably, our method loses 0% of model utility on the ToFU benchmark, whereas baseline methods may sacrifice 17% of utility on average to achieve comparable forget quality. Our code will be publicly available at https://github.com/UCSB-NLP-Chang/ULD.

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

↑