Safety Alignment in LLMs Localized to MLP Layers

Mingyu Zong, Sampad Mohanty, Bhaskar Krishnamachari· August 13, 2026 View original

Key takeaways

  • LLM safety alignment, specifically refusal behavior, is localized rather than distributed.
  • MLP weights, not attention weights, primarily encode safety-aligned refusal.
  • Refusal-relevant parameters are concentrated in mid-network MLP blocks (e.g., layers 8-11).
  • Targeted interventions on specific MLP layers could improve LLM safety and reduce over-refusal.

Who benefits

AI DevelopmentCybersecurityContent ModerationSoftware Development

Summary

This research reveals that safety-aligned refusal behavior in large language models is not distributed throughout the network but is primarily encoded within the MLP (Multi-Layer Perceptron) weights, particularly concentrated in mid-network blocks. Transplanting these specific MLP weights from aligned models into unaligned ones significantly recovers refusal capabilities.

The safety alignment mechanisms within large language models (LLMs) are often perceived as a property distributed across the entire network. However, the observed fragility of these safety features suggests that refusal behavior might be concentrated in a more specific set of parameters. This study investigates the precise location where safety-aligned refusal is encoded by selectively transferring weights from aligned models to unaligned base models. Experiments conducted across two open-weight model families and four safety benchmarks consistently showed that MLP (Multi-Layer Perceptron) weights are the dominant factor in transferring refusal capabilities. Replacing MLP parameters recovered significantly more malicious-prompt refusal than replacing attention parameters. Furthermore, within the MLP stack, the parameters responsible for refusal behavior exhibited a clear concentration in the mid-network layers, specifically blocks spanning layers 8-11. The findings also indicate that the composition of these safety-relevant components is non-additive, meaning that simply adding more aligned blocks can sometimes reduce refusal performance, suggesting a complex interplay of parameters.

Why it matters

Understanding where safety alignment is encoded allows professionals to develop more targeted and efficient interventions for improving LLM safety, potentially reducing over-refusal and enhancing model robustness against malicious prompts.

How to implement this in your domain

  1. 1Analyze existing LLM safety protocols to identify areas for targeted improvement.
  2. 2Investigate methods for fine-tuning or patching specific MLP layers to enhance safety alignment.
  3. 3Develop tools to monitor and evaluate refusal behavior in LLMs, distinguishing between malicious and benign prompts.
  4. 4Explore techniques for selectively transferring safety knowledge between different LLM versions or models.
  5. 5Collaborate with researchers to apply these findings in developing more robust and less brittle safety mechanisms for deployed LLMs.

Original post by Mingyu Zong, Sampad Mohanty, Bhaskar Krishnamachari

"arXiv:2608.11583v1 Announce Type: new Abstract: Safety alignment in large language models is often treated as a distributed property of the entire network, yet its practical brittleness suggests that refusal behavior may be concentrated in a smaller set of parameters. This work a…"

View on X

Originally posted by Mingyu Zong, Sampad Mohanty, Bhaskar Krishnamachari on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses

More in AI Research