AICRAug 2

No Single Neuron of Failure: Distributed Safety Alignment Against White-Box Attacks

arXiv:2608.0141411.1
Predicted impact top 16% in AI · last 90 daysOriginality Highly original
AI Analysis

This work is significant for the safety of large foundation models, particularly against sophisticated white-box attacks, by mitigating single points of failure in safety mechanisms.

This paper addresses the vulnerability of large foundation models to white-box neuron-level attacks by proposing Distributed Safety Alignment (DSA). DSA redundantly encodes safety capabilities across multiple neurons, ensuring safety even when critical neurons are disrupted, and substantially improves robustness against these attacks while preserving model utility.

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.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes