提出分层优化的遗忘算法,让大模型高效删知识又不丢能力。
BLUR: A Bi-Level Optimization Approach for LLM Unlearning
- 用上下层优化框架,优先删知识再保性能
- 在多个任务和模型上超越现有最优算法
- 适合需要合规删除数据的AI应用
让大语言模型(LLMs)能够删除训练中获取的知识与能力,对符合数据法规和推动生成式AI伦理至关重要。尽管已有多种遗忘算法,但如何正确定义遗忘问题仍不清晰。当前主流方法采用遗忘与保留损失的加权和,常因两者冲突导致性能下降。本文认为应建模遗忘问题的层次结构:遗忘(删除特定知识或能力)优先于保留(维持模型实用性)。由此提出双层优化框架,下层最小化遗忘损失,上层保持模型效用。基于此,我们提出新算法 exttt{BLUR},不仅具备强理论保证,且性能显著更优。大量实验表明, exttt{BLUR} 在多种遗忘任务、模型和评估指标下均持续优于现有最先进算法。代码已开源:https://github.com/OptimAI-Lab/BLURLLMUnlearning。
原文摘要 · Abstract (English)
Enabling large language models (LLMs) to unlearn knowledge and capabilities acquired during training has proven vital for ensuring compliance with data regulations and promoting ethical practices in generative AI. Although there are growing interests in developing various unlearning algorithms, it remains unclear how to best formulate the unlearning problem. The most popular formulation uses a weighted sum of forget and retain loss, but it often leads to performance degradation due to the inherent trade-off between forget and retain losses. In this work, we argue that it is important to model the hierarchical structure of the unlearning problem, where the forget problem (which \textit{unlearns} certain knowledge and/or capabilities) takes priority over the retain problem (which preserves model utility). This hierarchical structure naturally leads to a bi-level optimization formulation where the lower-level objective focuses on minimizing the forget loss, while the upper-level objective aims to maintain the model's utility. Based on this new formulation, we propose a novel algorithm, termed Bi-Level UnleaRning (\texttt{BLUR}), which not only possesses strong theoretical guarantees but more importantly, delivers superior performance. In particular, our extensive experiments demonstrate that \texttt{BLUR} consistently outperforms all the state-of-the-art algorithms across various unlearning tasks, models, and metrics. Codes are available at https://github.com/OptimAI-Lab/BLURLLMUnlearning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。