← 返回信息流

arXiv cs.CL论文

通过拒绝别名缓解消融

arxiv.org作者:Nathan Truong论文AI评分:70/100

本文针对大语言模型的安全对齐漏洞——消融(abliteration)提出防御方法。消融通过将权重矩阵投影到与拒绝方向正交的子空间来移除模型的拒绝能力,现有防御常忽视其根源:拒绝方向易于提取。作者提出AMRA方法,通过对残差流写入矩阵施加低秩更新,用随机别名替换触发拒绝的激活,并修正下游读取矩阵以保持原行为。在Llama-3-8B上,AMRA将消融后拒绝分数提升2.16点,MMLU下降不足0.5个百分点;在Gemma-2-9B上提升14.70点,但效用成本更高。

这篇是正式发表的长论文,站内提供中文解读,全文请到原文阅读 PDF。

Abstract:Abliteration, the removal of refusal capabilities from large language models by projecting weight matrices orthogonal to an extracted refusal direction, has emerged as a prominent safety concern through its ability to bypass post-training alignment using only a small set of contrastive prompts. We find that existing defenses commonly overlook the cause of abliteration; that is, how easily the refusal direction can be extracted. To hinder this process, we introduce a weight-editing method that obscures the refusal signal by applying rank-$k$ updates to residual stream writer matrices while replacing refusal-inducing activations with random aliases and correcting downstream reader matrices to preserve the model's original behavior. On Llama-3-8B, AMRA improves post-abliteration refusal scores by $2.16$ points over the undefended baseline with less than $0.5$ percentage points of MMLU degradation. On Gemma-2-9B, it improves the post-abliteration refusal by $14.70$ points over the baseline while keeping harmful output rates similar to the baseline, albeit at a greater utility cost.

阅读原文