arXiv:2504.12996cs.CLcs.AI2025-04ACL

通过分层优化实现大模型精准删忆,保护隐私同时保持性能

SHA256 at SemEval-2025 Task 4: Selective Amnesia -- Constrained Unlearning for Large Language Models via Knowledge Isolation

  • 基于因果分析定位关键记忆层,仅优化底层模块
  • 在1B模型上达成88%基准准确率,2nd名次表现优异
  • 适合关注大模型隐私安全与可控删忆的研究者

大语言模型在训练中常会记忆敏感信息,公开部署时带来风险。现有机器删忆方法难以选择性移除特定数据关联而不损害整体能力。本文针对SemEval-2025 Task 4提出两阶段方法:结合因果中介分析与分层优化,在OLMo架构(1B和7B参数)上系统追踪发现,前5层Transformer中的MLP模块是存储主体-属性关联的关键区域。基于此,我们设计约束优化策略:冻结上层,对下层使用联合损失函数——通过输出词元交叉熵惩罚最大化遗忘集损失,同时以自适应正则化最小化保留集偏差。该方法在1B模型赛道取得第2名,任务表现优秀的同时维持了88%的基线MMLU准确率。结果表明,基于因果洞察的分层优化为高效、精准的删忆提供了新范式,显著推进了大模型数据隐私保护进程。

原文摘要 · Abstract (English)

Large language models (LLMs) frequently memorize sensitive information during training, posing risks when deploying publicly accessible models. Current machine unlearning methods struggle to selectively remove specific data associations without degrading overall model capabilities. This paper presents our solution to SemEval-2025 Task 4 on targeted unlearning, which introduces a two-stage methodology that combines causal mediation analysis with layer-specific optimization. Through systematic causal tracing experiments on OLMo architectures (1B and 7B parameters), we identify the critical role of the first few transformer layers (layers 0-5) in storing subject-attribute associations within MLP modules. Building on this insight, we develop a constrained optimization approach that freezes upper layers while applying a novel joint loss function to lower layers-simultaneously maximizing forget set loss via output token cross-entropy penalties and minimizing retain set deviation through adaptive regularization. Our method achieves 2nd place in the 1B model track, demonstrating strong task performance while maintaining 88% of baseline MMLU accuracy. These results establish causal-informed layer optimization as a promising paradigm for efficient, precise unlearning in LLMs, offering a significant step forward in addressing data privacy concerns in AI systems.

大模型隐私删忆技术因果分析知识隔离

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