arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.13474cs.CLcs.LG
收藏

交替偏好优化用于大型语言模型中的事实知识遗忘

Alternate Preference Optimization for Unlearning Factual Knowledge in Large Language Models

University of Massachusetts Amherst(马萨诸塞大学阿默斯特分校);Microsoft(微软公司)

Anmol Mekala, Vineeth Dorna, Shreya Dubey, Abhishek Lalwani, David Koleczek, Mukund Rungta, Sadid Hasan, Elita Lobo

AI总结:

针对现有LLM遗忘方法仅依赖负面反馈导致输出质量差的问题,提出交替偏好优化(AltPO),结合负面与域内正面反馈,并引入新评估指标,实验证明其能有效遗忘且保持模型性能。

摘要:

机器遗忘旨在高效地消除特定训练数据(即遗忘集)对模型的影响。然而,现有的大型语言模型(LLMs)遗忘方法面临一个关键挑战:它们仅依赖负面反馈来抑制与遗忘集相关的响应,这常常导致无意义或不一致的输出,降低了模型的实用性并带来潜在的隐私风险。为解决这一局限,我们提出了一种名为交替偏好优化(AltPO)的新方法,该方法将负面反馈与遗忘集上的域内正面反馈相结合。此外,我们引入了新的评估指标来评估与遗忘集相关的响应质量。大量实验表明,我们的方法不仅能实现有效的遗忘,还能避免不良的模型行为,同时保持整体模型性能。我们的实现可在 https://github.com/molereddy/Alternate-Preference-Optimization 获取。

英文摘要:

Machine unlearning aims to efficiently eliminate the influence of specific training data, known as the forget set, from the model. However, existing unlearning methods for Large Language Models (LLMs) face a critical challenge: they rely solely on negative feedback to suppress responses related to the forget set, which often results in nonsensical or inconsistent outputs, diminishing model utility and posing potential privacy risks. To address this limitation, we propose a novel approach called Alternate Preference Optimization (AltPO), which combines negative feedback with in-domain positive feedback on the forget set. Additionally, we introduce new evaluation metrics to assess the quality of responses related to the forget set. Extensive experiments show that our approach not only enables effective unlearning but also avoids undesirable model behaviors while maintaining overall model performance. Our implementation can be found at https://github.com/molereddy/Alternate-Preference-Optimization.

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

↑