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

基于知识去学习的语言模型隐私风险缓解

Knowledge Unlearning for Mitigating Privacy Risks in Language Models

KAIST(韩国科学技术院);Konkuk University(建国大学);Seoul National University(首尔大学);LG AI Research(LG人工智能研究院);University of Illinois Chicago(伊利诺伊大学芝加哥分校)

Joel Jang, Dongkeun Yoon, Sohee Yang, Sungmin Cha, Moontae Lee, Lajanugen Logeswaran, Minjoon Seo

摘要:

预训练的语言模型(LM)在初始预训练期间记住大量知识,包括可能侵犯个人生活和身份隐私的信息。以前解决语言模型隐私问题的工作主要集中在数据预处理和差分隐私方法上,这两种方法都需要重新训练底层LM。我们提出知识unlearning作为一种替代方法,以减少隐私风险的LM事后。我们表明,简单地对目标令牌序列执行梯度上升可以有效地忘记它们,而对于较大的LM,一般语言建模性能几乎没有下降;有时甚至可以通过几次迭代来大幅提高底层LM。我们还发现,顺序遗忘比尝试一次性遗忘所有数据更好,并且遗忘高度依赖于哪种数据(域)被遗忘。通过与以前的数据预处理方法和已知用于减轻LM隐私风险的解码方法进行比较,我们表明,在易受提取攻击的数据先验已知的情况下,unlearning可以提供更强的经验隐私保证,同时更加有效和鲁棒。我们在https://github.com/joeljang/knowledge-unlearning上发布了复制结果所需的代码和数据集。

英文摘要:

Pretrained Language Models (LMs) memorize a vast amount of knowledge during initial pretraining, including information that may violate the privacy of personal lives and identities. Previous work addressing privacy issues for language models has mostly focused on data preprocessing and differential privacy methods, both requiring re-training the underlying LM. We propose knowledge unlearning as an alternative method to reduce privacy risks for LMs post hoc. We show that simply performing gradient ascent on target token sequences is effective at forgetting them with little to no degradation of general language modeling performances for larger LMs; it sometimes even substantially improves the underlying LM with just a few iterations. We also find that sequential unlearning is better than trying to unlearn all the data at once and that unlearning is highly dependent on which kind of data (domain) is forgotten. By showing comparisons with a previous data preprocessing method and a decoding method known to mitigate privacy risks for LMs, we show that unlearning can give a stronger empirical privacy guarantee in scenarios where the data vulnerable to extraction attacks are known a priori while being much more efficient and robust. We release the code and dataset needed to replicate our results at https://github.com/joeljang/knowledge-unlearning.

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

↑