用表示空间指导参数高效遗忘,提升大模型删敏效果与保留性能。
Representation-Guided Parameter-Efficient LLM Unlearning

- 基于表示空间几何特性初始化LoRA,定位精准遗忘子空间。
- 引入正则化损失,使更新输出正交于保留集表示空间。
- 在TOFU和WMDP上优于现有方法,忘得更彻底且不影响原有能力。
大型语言模型常记忆敏感或有害信息,亟需高效的机器遗忘技术。现有参数高效遗忘方法仍面临遗忘与保留之间的权衡问题,根源在于其依赖参数重要性度量来识别仅与遗忘数据相关的参数,而这一方法受参数叠加现象制约。由于大模型参数具有多义性,重要性度量难以区分与遗忘集和保留集相关的参数。为此,本文提出表示引导的低秩遗忘(REGLU),利用表示空间的几何特性实现鲁棒且精确的遗忘。首先,设计一种表示引导的LoRA初始化,识别选择性遗忘的最优子空间;其次,引入正则化损失,约束LoRA更新后的输出位于保留集表示子空间的正交补空间,从而最小化对保留任务性能的干扰。我们在多个模型上的TOFU和WMDP基准测试中评估了REGLU,结果表明其持续优于当前最佳基线,在保持更高模型效用的同时实现了更优的遗忘质量。
原文摘要 · Abstract (English)
Large Language Models (LLMs) often memorize sensitive or harmful information, necessitating effective machine unlearning techniques. While existing parameter-efficient unlearning methods have shown promise, they still struggle with the forget-retain trade-off. This can be attributed to their reliance on parameter importance metrics to identify parameters that are important exclusively for the forget set, which is fundamentally limited by the superposition phenomenon. Due to the polysemantic nature of LLM parameters, such an importance metric may struggle to disentangle parameters associated with the forget and retain sets. In this work, we propose Representation-Guided Low-rank Unlearning (REGLU), a novel approach that leverages the geometric properties of representation spaces to achieve robust and precise unlearning. First, we develop a representation-guided initialization for LoRA that identifies the optimal subspace for selective forgetting. Second, we introduce a regularization loss that constrains the outputs of the LoRA update to lie in the orthogonal complement of the retain set's representation subspace, thereby minimizing interference with the model's performance on the retain set. We evaluate REGLU on the TOFU and WMDP benchmarks across multiple models. Our results demonstrate that REGLU consistently outperforms state-of-the-art baselines, achieving superior unlearning quality while maintaining higher model utility.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。