Search papers, labs, and topics across Lattice.
This study investigates the geometric properties of refusal behavior in AI models, revealing that the refusal direction and subspace emerge from activation updates during refusal-completion training. By analyzing the training dynamics and the impact of diverse refusal prefixes, the authors demonstrate that these prefixes can enhance the stability of refusal features, making them more resilient to vector ablation attacks. The findings indicate that incorporating varied refusal starts can significantly strengthen the model's safety mechanisms against misuse.
Diverse refusal prefixes can significantly enhance the stability of AI models' refusal behaviors, making them more resistant to adversarial attacks.
Refusal training protects AI models from jailbreaks by training models to decline unsafe queries, reducing the risk of misuse. Recent work finds that refusal behavior in aligned language models can be mediated by a single activation direction or a low-dimensional refusal subspace shared across harmful prompts: ablating those directions suppresses refusals while largely preserves other model capabilities. Yet it remains unclear why safety-critical features in a wide range of models emerge and concentrated, low-dimensional structure. In a case study of OLMo-2-0425-1B-Instruct we find that the refusal geometry reflects refusal training: activation updates resulting from refusal-completion first-token losses explain the resulting refusal direction and refusal subspace. We study refusal directions through the training dynamics across refusal datasets and reveal that their brittleness is associated with repetitive refusal starts, which in turn is linked to concentration of gradients and refusal features in a low-dimensional subspace. Across frozen-model analyses and controlled synthetic fine-tuning, we find evidence of a hardening lever: diverse refusal starts can raise stable ranks of gradients and activation changes, making refusals harder to remove with a vector ablation attack.