arXiv:2509.22483cs.LGcs.AI2025-09被引 3

提出新框架实现高效模型删忆,兼顾遗忘效果与性能保留

OFMU: Optimization-Driven Framework for Machine Unlearning

  • 用双层优化结构,先强制遗忘再恢复性能
  • 在多个视觉语言任务上遗忘率超90%,性能下降低于5%
  • 适合需合规删忆的AI系统开发者使用

部署于敏感场景的大语言模型需具备删除特定知识(如用户请求、版权内容或过时信息)的能力,以满足监管、隐私和安全要求。机器删忆旨在移除目标数据的影响(遗忘),同时保持对其他数据的性能(保留)。现有方法常将此问题转化为加权求和的单目标优化,但因梯度方向冲突导致训练不稳定且模型效用下降。为此,我们提出OFMU——一种基于惩罚的双层优化框架,通过内层最大化步骤引入相似性感知惩罚,解耦遗忘与保留目标的梯度,外层最小化步骤恢复模型性能。所提两循环算法在凸与非凸条件下均具有收敛性保证,理论分析表明其在遗忘效率与模型效用间取得更优平衡。大量实验显示,OFMU在多个视觉与语言基准上持续优于现有方法,遗忘率超过90%,性能损失低于5%。

原文摘要 · Abstract (English)

Large language models deployed in sensitive applications increasingly require the ability to unlearn specific knowledge, such as user requests, copyrighted materials, or outdated information, without retraining from scratch to ensure regulatory compliance, user privacy, and safety. This task, known as machine unlearning, aims to remove the influence of targeted data (forgetting) while maintaining performance on the remaining data (retention). A common approach is to formulate this as a multi-objective problem and reduce it to a single-objective problem via scalarization, where forgetting and retention losses are combined using a weighted sum. However, this often results in unstable training dynamics and degraded model utility due to conflicting gradient directions. To address these challenges, we propose OFMU, a penalty-based bi-level optimization framework that explicitly prioritizes forgetting while preserving retention through a hierarchical structure. Our method enforces forgetting via an inner maximization step that incorporates a similarity-aware penalty to decorrelate the gradients of the forget and retention objectives, and restores utility through an outer minimization step. To ensure scalability, we develop a two-loop algorithm with provable convergence guarantees under both convex and non-convex regimes. We further provide a rigorous theoretical analysis of convergence rates and show that our approach achieves better trade-offs between forgetting efficacy and model utility compared to prior methods. Extensive experiments across vision and language benchmarks demonstrate that OFMU consistently outperforms existing unlearning methods in both forgetting efficacy and retained utility.

机器删忆双层优化大模型安全

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