Skip to content
InfoResearchPreprintLLM-specific

Follow the Gradient: Tracing Safety Degradation Back to Training Data

Published
Record updated
View JSON

Summary

This paper presents SimpleLoRIF, a gradient embedding method that flags instruction-tuning examples whose gradients align with unsafe behavior, even when the examples look benign. Applied to Tulu 3, removing the 20% most aligned examples to produce SafeTulu reduced GCG attack success rate by up to 22% across Qwen, Llama, and Olmo, with capability largely preserved. On emergent misalignment, the authors report AUC as high as 0.865 in the gradient space of misaligned models.

Mitigation

The source describes removing the 20% of training examples most gradient-aligned with unsafe responses, producing the SafeTulu dataset, and releases SafeTulu along with a Hugging Face Space that lets researchers customize the removal fraction or filter subsets.