LGAICLCRApr 1, 2024

What is in Your Safe Data? Identifying Benign Data that Breaks Safety

arXiv:2404.01099v2107 citationsh-index: 23
Originality Incremental advance
AI Analysis

This addresses a critical safety vulnerability in LLMs for developers and users, though it is incremental as it builds on prior findings of safety degradation from benign fine-tuning.

The paper investigates why fine-tuning aligned large language models with benign data can degrade safety, identifying specific data patterns that lead to jailbreaking and showing that training on just 100 such datapoints increases harmful responses from <20% to >70%.

Current Large Language Models (LLMs), even those tuned for safety and alignment, are susceptible to jailbreaking. Some have found that just further fine-tuning an aligned model with benign data (i.e., data without harmful content) surprisingly leads to substantial degradation in safety. We delve into the data-centric aspects of why benign fine-tuning inadvertently contributes to jailbreaking. First, we represent fine-tuning data through two lenses: representation and gradient spaces. Additionally, we propose a bi-directional anchoring method that, during the selection process, prioritizes data points that are close to harmful examples and far from benign ones. Our approach effectively identifies subsets of benign data that are more likely to degrade the model's safety after fine-tuning. Training on just 100 of these seemingly benign datapoints surprisingly leads to the fine-tuned model affirmatively responding to >70% of tested harmful requests, compared to <20% after fine-tuning on randomly selected data. We also observe that the selected data frequently appear as lists, bullet points, or math questions, indicating a systematic pattern in fine-tuning data that contributes to jailbreaking.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes