arXiv:cs.AI· Zixuan Weng, Haoqi He, Weifei Jin, Yue Huang, Yang Bai·· 13 小时前
Attention Sees, MLP Refuses:追踪 LLM 中从注意力到 MLP 的安全通路
Attention Sees, MLP Refuses: Tracing the Attention-to-MLP Safety Pathway in LLMs
AI 导读
研究将 LLM 安全失效重构为可测量的通路问题,在 18 组 Base-Instruct 配对(覆盖 Qwen、Llama、Gemma)中发现,有害性更易从 attention 更新中解码,而 MLP 更新在对齐模型中更强地写入拒绝。
正文
Abstract:Safety-aligned large language models (LLMs) remain vulnerable to jailbreaks and can over-refuse benign requests. Understanding these failures requires distinguishing harmfulness recognition from the computation that writes refusal. We reframe this as a pathway question and make it measurable on residual-stream updates in decoder-only Transformers, separating the harmfulness information a module update exposes from the change it produces in readable refusal evidence. Across 18 Base-Instruct pairs spanning Qwen, Llama, and Gemma, harmfulness is decoded more readily from attention updates, particularly in early layers, whereas MLP updates contribute more strongly to harm-selective refusal writing in aligned models. Post-training strengthens MLP refusal writing in all 18 pairs without systematically improving harmfulness decodability, while positive MLP writing becomes more concentrated in deeper layers after post-training. Across six reasoning-enabled models, harmfulness remains decodable during reasoning while strong refusal writing concentrates at answer onset, a pattern that persists with reasoning disabled. Causal interventions further support this pathway: resetting downstream MLP updates to their clean values substantially reduces the refusal shift caused by upstream attention interventions. Building on these findings, we introduce Sentinel, which combines an attention-derived perception score P with an MLP-derived decision score D. Their joint profiles characterize jailbreaks and over-refusals, and conditional routing with both signals achieves a lower aggregate error rate than single-signal routing and evaluated fixed reminders across eight models.
| Comments: | Preprint |
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.13078 [cs.AI] |
| (or arXiv:2610.13078v1 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2610.13078 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: ZIxuan Weng [view email]
[v1]
Fri, 9 Oct 2026 06:34:26 UTC (897 KB)
来源:arXiv:cs.AI · arxiv.org