通过分布式编码提升模型抗神经劫持能力
No Single Neuron of Failure: Distributed Safety Alignment Against White-Box Attacks

- 将安全机制分布于多个神经元,避免单点失效
- 在输入层定位关键神经元,实现精准干扰防御
- 适合关注模型安全与鲁棒性的研究者
随着开源大模型的快速发展,安全威胁正从黑盒越狱转向白盒攻击,直接定位并操控安全相关神经元。现有对齐方法通常只依赖少数神经元,存在脆弱的单点失效问题。为此,本文提出分布式安全对齐(DSA),将安全能力冗余编码至多个计算神经元中,确保关键安全神经元被破坏时仍能维持安全基线。具体地,将语言侧前馈网络中下投影层的输入作为个体神经元激活,并结合神经元激活值与损失梯度,计算方向感知的一阶泰勒得分,全局识别对当前拒绝行为贡献最大的神经元。随后,采用确定性掩码与随机丢弃相结合的方式进行定向干扰,迫使模型放弃依赖单一安全神经元,转而将安全行为冗余分布在多个补偿性神经元上。大量实验表明,DSA显著提升了对抗白盒神经元级安全攻击的鲁棒性,同时保持了模型在通用语言和多模态任务中的性能。
原文摘要 · Abstract (English)
With the rapid release of open-weight large foundation models, safety threats are shifting from black-box jailbreaks to neuron-level white-box attacks that directly identify and manipulate safety-related neurons. Existing alignment methods often investigate the safety behavior on a small number of neurons, creating fragile single point of failure with limited redundancy. To address this issue, we propose distributed safety alignment (DSA), which redundantly encodes safety capabilities across multiple computational neurons, ensuring that the model maintains its safety baseline even when critical safety neurons are disrupted. Specifically, we localize the intervention to the inputs of the down-projection layers in language-side feed-forward networks and treat each feature coordinate as the activation of an individual neuron. DSA then combines neuron activations with loss gradients to compute a direction-aware first-order Taylor score that globally identifies the neurons that contribute most to the current refusal behavior of the model. Finally, targeted disruption via deterministic masking and stochastic dropout is coupled, forcing the model to abandon narrow safety neurons and redundantly encode safety behavior across multiple compensatory neurons. Extensive experiments show that DSA substantially improves robustness against white-box neuron-level safety attacks while preserving the model's general language and multimodal utility.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。