Search papers, labs, and topics across Lattice.
This study investigates the geometric structure of refusal behavior in AI models, revealing that refusal training shapes a low-dimensional refusal subspace that is critical for maintaining safety. By analyzing the training dynamics of the OLMo-2-0425-1B-Instruct model, the authors demonstrate that the activation updates from refusal-completion losses directly influence the model's refusal direction and subspace. The findings indicate that using diverse refusal prefixes can enhance the robustness of refusal mechanisms against vector ablation attacks, thereby improving model safety.
Diverse refusal prefixes can significantly bolster the stability of refusal mechanisms, making them more resilient against targeted 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.