Attention Is Where You Attack
On the map: LLM alignment / RLHF / preference optimisation → Safety alignment and jailbreak attacks
also in 1 more
Abstract (arXiv)
Safety-aligned large language models rely on RLHF and instruction tuning to refuse harmful requests, yet the internal mechanisms implementing safety behavior remain poorly understood. We introduce the Attention Redistribution Attack (ARA), a white-box adversarial attack that identifies safety-critical attention heads and crafts nonsemantic adversarial tokens that redirect attention away from safety-relevant positions. Unlike prior jailbreak methods operating at the semantic or output-logit level, ARA targets the geometry of softmax attention on the probability simplex using Gumbel-softmax optimization over targeted heads. Across LLaMA-3-8B-Instruct, Mistral-7B-Instruct-v0.1, and Gemma-2-9B-it, ARA bypasses safety alignment with as few as 5 tokens and 500 optimization steps, achieving 36% ASR on Mistral-7B and 30% on LLaMA-3 against 200 HarmBench prompts, while Gemma-2 remains at 1%. Our principal mechanistic finding is a dissociation between ablation and redistribution: zeroing out the top-ranked safety heads produces at most 1 flip among 39 to 50 baseline refusals, while ARA targeting the corresponding safety-heavy layers flips 72/200 prompts on Mistral-7B and 60/200 on LLaMA-3. This suggests that safety is not localized in these heads as removable components, but emerges from the attention routing they perform. Removing a head allows compensation through the residual stream, while redirecting its attention propagates a corrupted signal downstream.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Safety alignment and jailbreak attacks, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 36 of its 42 references that are in the map. First posted on arXiv 30 Apr 2026.
Builds on
36 of 42 references.
Closest ancestors · 3
Same direction, newest first
- 2023AutoDAN: Generating Stealthy Jailbreak Prompts on Aligned Large Language Models3 Oct
- 2023Baseline Defenses for Adversarial Attacks Against Aligned Language Models1 Sep
- 2023Universal and Transferable Adversarial Attacks on Aligned Language Models27 Jul
Other references · 33
- 2017Attention Is All You Needcited by 59,932
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
- 2017Deep reinforcement learning from human preferencescited by 3,846
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,634
- 2023Mistral 7Bcited by 3,065
- 2016Towards Evaluating the Robustness of Neural Networkscited by 2,976
- 2022Constitutional AI: Harmlessness from AI Feedbackcited by 2,950
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
21 more
- 2023Jailbroken: How Does LLM Safety Training Fail?cited by 1,565
- 2021Multitask Prompted Training Enables Zero-Shot Task Generalizationcited by 1,462
- 2024HarmBench: A Standardized Evaluation Framework for Automated Red Teaming and Robust Refusalcited by 1,330
- 2023Jailbreaking Black Box Large Language Models in Twenty Queriescited by 1,326
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,239
- 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
- 2024Refusal in Language Models Is Mediated by a Single Directioncited by 960
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 700
- 2023SmoothLLM: Defending Large Language Models Against Jailbreaking Attackscited by 434
- 2023Detecting Language Model Attacks with Perplexitycited by 411
- 2023Low-Resource Languages Jailbreak GPT-4cited by 290
- 2023Certifying LLM Safety against Adversarial Promptingcited by 248
- 2023A Wolf in Sheep's Clothing: Generalized Nested Jailbreak Prompts can Fool Large Language Models Easilycited by 242
- 2023Multilingual Jailbreak Challenges in Large Language Modelscited by 222
- 2024AmpleGCG: Learning a Universal and Transferable Generative Model of Adversarial Suffixes for Jailbreaking Both Open and Closed LLMscited by 160
- 2024Improved Techniques for Optimization-Based Jailbreaking on Large Language Modelscited by 98
- 2024Coercing LLMs to do and reveal (almost) anythingcited by 88
- 2024CodeAttack: Revealing Safety Generalization Challenges of Large Language Models via Code Completioncited by 70
- 2025Attention Eclipse: Manipulating Attention to Bypass LLM Safety-Alignmentcited by 6
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Safety Alignment Should Be Made More Than Just A Few Attention Heads
- 2026Seeing No Evil: Blinding Large Vision-Language Models to Safety Instructions via Adversarial Attention Hijacking
- 2025Spot Risks Before Speaking! Unraveling Safety Attention Heads in Large Vision-Language Models
- 2026Robust Harmful Features Under Jailbreak Attacks: Mechanistic Evidence from Attention Head Specialization in Large Language Models
- 2026Understanding the Effects of Safety Unalignment on Large Language Models
- 2025Safety Alignment Can Be Not Superficial With Explicit Safety Signals
- 2025SHIELD: Classifier-Guided Prompting for Robust and Safer LVLMs
- 2026Beyond Routine Compliance: Cunning Data Cultivates Safety Vigilance in Large Language Models