提出新方法实现大模型精准删知识且不伤能力
UniErase: Towards Balanced and Precise Unlearning in Language Models
- 用可学习的遗忘令牌引导模型进入遗忘空间
- 仅改3.66%参数,遗忘能力提升4倍且保留更好性能
- 适合需要精确删除敏感或过时信息的场景
大语言模型需迭代更新以解决过时信息问题,而模型遗忘技术提供了选择性删除的方法。然而,主流遗忘方法依赖微调,往往难以精准定位目标知识,在大规模、连续场景下也难平衡遗忘效果与通用能力。为此,本文提出UniErase,一种新型遗忘框架,兼具精准性与平衡性。我们首先引入遗忘令牌(Unlearning Token),通过优化使其引导模型进入遗忘空间;进一步设计轻量级遗忘编辑(Unlearning Edit),高效关联目标知识与该元令牌。作为基于编辑的新范式,UniErase在批量、顺序及精准遗忘任务中均表现优异,涵盖虚构与真实知识场景。在TOFU基准上,相比8个基线,仅修改约3.66%参数,其模型能力比最优遗忘基线提升约4.01倍,同时保持更高遗忘效果;相较最优保留能力方法,遗忘效率提升35.96%,展现出当前领域双项顶尖的平衡表现。
原文摘要 · Abstract (English)
Large language models (LLMs) require iterative updates to address the outdated information problem, where LLM unlearning offers an approach for selective removal. However, mainstream unlearning methods primarily rely on fine-tuning techniques, which often lack precision in targeted unlearning and struggle to balance unlearning efficacy with general ability under massive and sequential settings. To bridge this gap, in this work, we introduce UniErase, a novel unlearning framework that demonstrates precision and balanced performances between knowledge unlearning and ability retaining. We first propose the Unlearning Token, which is optimized to steer LLMs toward a forgetting space. To achieve concrete unlearning behaviors, we further introduce the lightweight Unlearning Edit to efficiently associate the unlearning targets with this meta-token. Serving as a new unlearning paradigm via editing, UniErase achieves outstanding performances across batch, sequential, and precise unlearning tasks under fictitious and real-world knowledge scenarios. On the TOFU benchmark, compared with 8 baselines, UniErase, modifying only $\sim$ \textbf{3.66%} of the LLM parameters, outperforms the previous best-forgetting baseline by \textbf{$\sim$ 4.01$\times$} for \textbf{model ability} with even higher unlearning efficacy. Similarly, UniErase, with better ability retention, also surpasses the previous best-retaining method by \textbf{35.96%} for \textbf{unlearning efficacy}, showing balanced and dual top-tier performances in the current unlearning community.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。