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

TOFU:面向大语言模型的虚构遗忘任务

TOFU: A Task of Fictitious Unlearning for LLMs

Carnegie Mellon University(卡内基梅隆大学)

Pratyush Maini, Zhili Feng, Avi Schwarzschild, Zachary C. Lipton, J. Zico Kolter

AI总结:

本文提出TOFU基准,通过200个合成作者档案及遗忘集和配套指标,系统评估大语言模型遗忘效果,并发现现有基线方法均无法实现有效遗忘。

摘要:

在海量网络语料上训练的大语言模型能够记忆并复现敏感或私有数据,从而引发法律和伦理方面的担忧。遗忘(unlearning),即通过调优模型使其忘记训练数据中存在的信息,为我们在训练之后保护私有数据提供了一种途径。尽管目前已存在若干用于此类遗忘的方法,但尚不清楚它们在多大程度上能够使模型等同于那些从一开始就从未学习过待遗忘数据的模型。为应对这一挑战,我们提出了 TOFU,即虚构遗忘任务(Task of Fictitious Unlearning),作为一个基准,旨在帮助加深我们对遗忘问题的理解。我们提供了一个包含 200 个多样化合成作者档案的数据集,每个档案由 20 个问答对组成,并将这些档案的一个子集称为遗忘集(forget set),作为遗忘的目标。我们构建了一套协同工作的评估指标,以提供遗忘效果的整体图景。最后,我们给出了一组来自现有遗忘算法的基线结果。重要的是,我们所考虑的所有基线均未表现出有效的遗忘,这促使人们继续努力开发能够有效调优模型的遗忘方法,使模型真正表现得如同从未在遗忘数据上进行过训练一样。

英文摘要:

Large language models trained on massive corpora of data from the web can memorize and reproduce sensitive or private data raising both legal and ethical concerns. Unlearning, or tuning models to forget information present in their training data, provides us with a way to protect private data after training. Although several methods exist for such unlearning, it is unclear to what extent they result in models equivalent to those where the data to be forgotten was never learned in the first place. To address this challenge, we present TOFU, a Task of Fictitious Unlearning, as a benchmark aimed at helping deepen our understanding of unlearning. We offer a dataset of 200 diverse synthetic author profiles, each consisting of 20 question-answer pairs, and a subset of these profiles called the forget set that serves as the target for unlearning. We compile a suite of metrics that work together to provide a holistic picture of unlearning efficacy. Finally, we provide a set of baseline results from existing unlearning algorithms. Importantly, none of the baselines we consider show effective unlearning motivating continued efforts to develop approaches for unlearning that effectively tune models so that they truly behave as if they were never trained on the forget data at all.

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

↑