arXiv:2605.16776cs.LGcs.AI2026-05中稿 · ed被引 1

提出新方法让大模型安全删除敏感知识并拒绝生成有害内容。

Distinguishable Deletion: Unifying Knowledge Erasure and Refusal for Large Language Model Unlearning

论文配图:Distinguishable Deletion: Unifying Knowledge Erasure and Refusal for Large Language Model Unlearning
图 1 · 摘自论文原文
  • 在隐空间限制响应分布,实现知识删除与拒绝机制统一。
  • 能量指数可精准量化知识存在度,训练时控制删除边界。
  • 兼顾删除效果与安全性,适合需高可靠性的模型部署场景。

缓解敏感和有害输出是保障大语言模型安全部署的基础。现有方法主要分为知识删除(KD)和可区分拒绝(DR)两类:前者在训练中擦除不良信息,后者在推理中引导模型避开敏感知识。然而,基于KD的遗忘方法因以抑制特定词元序列代替完整知识清除,易导致偏差删除;而基于DR的方法因底层知识仍保留,存在有害知识重新出现的风险。为此,本文提出可区分删除(D²),通过在潜在表示空间中限制响应分布而非特定词元,实现对不良知识的清除,并区分未学习与保留知识,从而在推理时安全、连贯地触发拒绝机制。为实现D²,我们引入能量指数,量化知识存在程度及未学习与保留内容间的分离度。数学与实证分析表明,该能量指标兼具准确性和高效性,支持训练阶段的能量边界对齐(EUA)与推理阶段的能量拒绝机制。大量实验表明,EUA显著优于此前方法,验证了D²的优越性。代码已公开于https://github.com/Puning97/EUA-for-LLM-Unlearning。

原文摘要 · Abstract (English)

Mitigating sensitive and harmful outputs is fundamental to ensuring safe deployment of LLMs. Existing approaches typically follow two paradigms: Knowledge Deletion (KD), which erases undesirable information during training, and Distinguishable Refusal (DR), which steers models away from using sensitive knowledge during inference. Despite rapid progress, KD-based unlearning struggles with biased deletion due to suppressing specific token sequences as a substitute for complete knowledge removal, whereas DR-based unlearning risks the re-emergence of harmful knowledge because the underlying knowledge remains intact. To address these issues, we propose Distinguishable Deletion ($\mathrm{D^2}$), a paradigm that restricts the response distribution in the latent representation rather than specific tokens to erase undesirable knowledge, while distinguishing it from retained knowledge, enabling a refusal mechanism to handle unlearned inputs safely and coherently. To implement $\mathrm{D^2}$, we introduce an energy index that quantifies the presence of knowledge and the separation between unlearned and retained content. Mathematical and empirical analyses show that energy is both accurate and efficient, enabling Energy-based Unlearning Alignment (EUA) to enforce energy-boundary unlearning during training and apply an energy-based refusal mechanism at inference. Extensive experiments demonstrate that EUA significantly outperforms previous methods, indicating the superiority of $\mathrm{D^2}$. Our code is available at https://github.com/Puning97/EUA-for-LLM-Unlearning.

大模型知识删除安全生成模型遗忘

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