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

边学习边遗忘:一种面向生成式语言模型的迭代遗忘框架

Learn while Unlearn: An Iterative Unlearning Framework for Generative Language Models

University of Science and Technology of China(中国科学技术大学);The Hong Kong University of Science and Technology(香港科技大学)

Haoyu Tang, Ye Liu, Xi Zhao, Xukai Liu, Yanghai Zhang, Kai Zhang, Xiaofang Zhou, Enhong Chen

AI总结:

本文提出迭代对比遗忘(ICU)框架,通过知识遗忘引导、对比学习增强和迭代遗忘精化三个模块,在不依赖原始训练数据的前提下实现敏感信息遗忘并保持模型表达能力。

摘要:

机器学习,尤其是自然语言处理(NLP)领域的最新进展,已经催生了在海量数据集上训练的强大模型。然而,这些模型存在泄露敏感信息的风险,引发了隐私方面的担忧。为此,欧盟《通用数据保护条例》(GDPR)等监管措施推动了对机器遗忘(Machine Unlearning)技术日益增长的兴趣,该技术使模型能够选择性地遗忘特定数据条目。早期的遗忘方法主要依赖预处理手段,而近年来的研究则转向基于训练的解决方案。尽管这些方法有效,但一个关键局限仍然存在:大多数方法需要访问原始训练数据,而这些数据往往不可得。此外,直接应用遗忘技术会带来削弱模型表达能力的代价。为应对这些挑战,我们提出了迭代对比遗忘(ICU)框架,该框架包含三个核心组件:一个知识遗忘引导模块,旨在利用遗忘损失针对特定知识进行移除;一个对比学习增强模块,用于在单纯遗忘目标下保持模型的表达能力;以及一个迭代遗忘精化模块,通过持续评估和更新动态调整遗忘过程。实验结果表明,我们的ICU方法能够在遗忘敏感信息的同时保持模型的整体性能,为注重隐私的机器学习应用提供了一种有前景的解决方案。

英文摘要:

Recent advances in machine learning, particularly in Natural Language Processing (NLP), have produced powerful models trained on vast datasets. However, these models risk leaking sensitive information, raising privacy concerns. In response, regulatory measures such as the European Union's General Data Protection Regulation (GDPR) have driven increasing interest in Machine Unlearning techniques, which enable models to selectively forget specific data entries. Early unlearning approaches primarily relied on pre-processing methods, while more recent research has shifted towards training-based solutions. Despite their effectiveness, a key limitation persists: most methods require access to original training data, which is often unavailable. Additionally, directly applying unlearning techniques bears the cost of undermining the model's expressive capabilities. To address these challenges, we introduce the Iterative Contrastive Unlearning (ICU) framework, which consists of three core components: A Knowledge Unlearning Induction module designed to target specific knowledge for removal using an unlearning loss; A Contrastive Learning Enhancement module to preserve the model's expressive capabilities against the pure unlearning goal; And an Iterative Unlearning Refinement module that dynamically adjusts the unlearning process through ongoing evaluation and updates. Experimental results demonstrate the efficacy of our ICU method in unlearning sensitive information while maintaining the model's overall performance, offering a promising solution for privacy-conscious machine learning applications.

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

↑