Data filtering works a lot worse than you would expect
Summary
A study investigating data filtering for supervised fine-tuning (SFT) of large language models found that removing specific training data points is largely ineffective at eliminating undesirable behaviors. Researchers used a "speed-run" OLMo SFT model, a rank 64 LoRA on OLMo 3 7B mid-train with a 1% stratified sample of the Dolci-Think-SFT-7B dataset. They evaluated seven behaviors, including bold formatting, liberal-lean, and "validate feelings." Across various Training Data Attribution (TDA) methods—EKFAC, Probe, LLM Judge, and Activation-based—filtering out 10-25% of identified "proponent" documents often performed no better than random removal. For instance, filtering 10% of documents did not reduce the "Your feelings are valid" behavior, despite relevant phrases appearing in less than 0.2% of documents. The only exception was "refusal," which proved filterable using Probes and LLM Judges. This suggests many SFT behaviors are elicited from the base model's existing personas rather than being taught by specific SFT data.
Key takeaway
For machine learning engineers aiming to mitigate undesirable behaviors in LLMs via supervised fine-tuning, recognize that data filtering is generally ineffective for broad traits. Your efforts should focus on "refusal" behaviors, where LLM Judges or Probes show efficacy. For other behaviors, consider that they may be inherent "personas" elicited from the base model, requiring alternative intervention strategies beyond simple data removal.
Key insights
Data filtering is largely ineffective for removing broad undesirable LLM behaviors acquired during SFT, except for refusal.
Principles
- Broad SFT behaviors are often elicited, not taught.
- Many behaviors are bundled into model personas.
- Random data removal often matches targeted filtering.
Method
The study involved fine-tuning OLMo 3 7B mid-train with LoRA, identifying behaviors, scoring training examples with TDA methods, filtering top examples, and retraining to evaluate behavior reduction.
In practice
- Prioritize filtering for "refusal" behaviors.
- Use LLM Judges or Probes for refusal filtering.
- Re-evaluate assumptions about SFT behavior origins.
Topics
- Data Filtering
- Supervised Fine-Tuning
- Large Language Models
- OLMo 3 7B
- Training Data Attribution
- Model Behavior
- LoRA
Code references
Best for: Research Scientist, AI Engineer, NLP Engineer, AI Scientist, Machine Learning Engineer, MLOps Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by AI Alignment Forum.