arXiv:2608.16249cs.LG2026-08

提出可精确控制遗忘程度的LLM去知识方法

SAUL: Sharpness-Aware Augmented-Lagrangian Unlearning

论文配图:SAUL: Sharpness-Aware Augmented-Lagrangian Unlearning
图 1 · 摘自论文原文
  • 用约束优化显式控制遗忘量,避免过度遗忘
  • 在三个基准上实现更好遗忘与通用性平衡
  • 适合需要精准删除数据的AI安全场景

大型语言模型中的机器去知识面临遗忘程度与通用性能之间的关键权衡。我们提出SAUL(Sharpness-Aware Augmented-Lagrangian Unlearning),将去知识建模为遵循‘恰到好处地遗忘,不多不少’原则的约束最小化问题。核心在于将遗忘设定为带有预定满足标准的显式约束,而以往方法通常通过优化目标隐式指定遗忘水平。增广拉格朗日控制器根据约束违反情况自适应调节遗忘侧压力,并在满足预定标准时最终关闭遗忘更新。保留与遗忘目标上的锐度感知更新,结合双优化器设计维持角色分离状态,进一步稳定去知识动态。我们在TOFU、WMDP和MUSE基准上评估了SAUL,证明其在特定遗忘标准下优于代表性锐度与扰动基线方法。此外,我们还表明,将增广拉格朗日控制器作为即插即用模块应用于现有基线,可在TOFU上提升遗忘后的通用性能,验证了显式遗忘控制的实际价值。

原文摘要 · Abstract (English)

Machine unlearning in Large Language Models (LLMs) faces a critical trade-off between erasing target knowledge and preserving general utility. We propose SAUL (Sharpness-Aware Augmented-Lagrangian Unlearning), which formulates unlearning as a constrained minimization problem following the principle of "forget enough, but no more than necessary." At its core, SAUL formulates forgetting as an explicit constraint with a prescribed satisfaction criterion, whereas prior unlearning methods typically specify the desired level of forgetting implicitly through optimization objectives. An augmented Lagrangian controller adaptively adjusts forget-side pressure according to constraint violation and can eventually deactivate the forget-side update as the prescribed criterion remains satisfied. Sharpness-aware updates on both retain and forget objectives, together with a dual-optimizer design that maintains role-separated states, further stabilize the resulting unlearning dynamics. We evaluate SAUL on the TOFU, WMDP, and MUSE benchmarks, demonstrating favorable forgetting-utility trade-offs over representative sharpness- and perturbation-based baselines under benchmark-specific forgetting criteria. Beyond the complete SAUL framework, we further show on TOFU that applying the augmented-Lagrangian controller as a drop-in modifier to representative baselines improves their post-forgetting utility, demonstrating the practical value of explicit forgetting control.

大模型去知识约束优化增强拉格朗日

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。