Data filtering works a lot worse than you would expect

· Source: AI Alignment Forum · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, extended

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

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

Topics

Code references

Best for: Research Scientist, AI Engineer, NLP Engineer, AI Scientist, Machine Learning Engineer, MLOps Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by AI Alignment Forum.