Back to AI News
arxiv

Addition by Subtraction: Mitigating Fragile Safety Alignment in LLMs

This paper introduces Addition by Subtraction, a finetuning method that removes safety-critical neurons and uses preference optimization to redistribute safety behavior across the model. The resulting models are significantly more robust to pruning-based attacks, refusal-direction ablation, and various black-box jailbreaks, with attack success rates reduced by up to 90% across four models. The approach also improves truthfulness while largely preserving general capabilities.

AI Brief

Provides a concrete, novel method to strengthen LLM safety alignment against sparse-component vulnerabilities, with broad empirical validation.

arXiv:2611.06553v1 Announce Type: new Abstract: Large language models (LLMs) undergo safety alignment to prevent harmful generations, yet they remain vulnerable to jailbreak attacks. Recent work reveals a deeper structural weakness: safety alignment can be highly sparse, such that pruning a small number of safety-critical neurons or ablating a single refusal direction can substantially undermine aligned behavior. In this work, we take the first steps toward strengthening safety alignment such that it is less dependent on sparse model components. We propose Addition by Subtraction, which gradually removes identified safety-critical neurons during finetuning and uses preference optimization to re-establish safety-aligned behavior through other parts of the model. Our results show that the resulting models are substantially more robust to pruning-based attacks. Remarkably, these benefits generalize beyond the white-box attacks motivating our design: Addition by Subtraction improves robustness against other sparsity-based interventions, including refusal-direction ablation, as well as a broad range of black-box jailbreak attacks. The finetuned models also exhibit improved truthfulness while largely preserving general capabilities. Comprehensive experiments across four models from three vendors demonstrate consistent improvements in safety robustness, reducing attack success rates by up to 90% for some attacks.
Be the first to rate this article.

Comments & Questions (0)

Captcha
Please be respectful — let's keep the conversation friendly.

No comments yet

Be the first to comment!