Online Data Selection Is Implicit Alignment
Summary
Online data selection during supervised fine-tuning (SFT) acts as an implicit alignment mechanism, influencing model behavioral preferences even without explicit preference optimization. This research compares random, loss-based, quality-based, and diversity-based online selectors, measuring their induced behavioral drift across metrics like helpfulness, refusal rate, verbosity, truthfulness, sycophancy, calibration, and jailbreak robustness. The study formalizes online selection as a reweighted SFT objective, where the online scorer functions similarly to a reward model, predicting that high-scoring data can favor specific response styles or safety postures. Empirically, selectors showing comparable task accuracy diverge sharply in refusal rate, verbosity, and sycophancy, with shifts directly correlating to the selected data's attribute mixture. The authors introduce Alignment Drift Auditing (ADA) for quantifying selection-induced behavioral movement and Alignment-Aware Selection (AAS), a diagnostic selector that maintains data efficiency while constraining drift on safety and style axes.
Key takeaway
For Machine Learning Engineers fine-tuning models, recognize that your online data selection choices implicitly align model behavior, even without explicit preference optimization. Carefully define your data scoring functions, as they act like a reward model, predictably shifting refusal rates, verbosity, and sycophancy. Implement Alignment Drift Auditing (ADA) to quantify these shifts and consider Alignment-Aware Selection (AAS) to maintain data efficiency while constraining unwanted behavioral drift in your models.
Key insights
Online data selection during SFT implicitly aligns models by shaping behavioral preferences, acting like a reward model.
Principles
- Online data selection reweights SFT objectives.
- Scorer definition predicts behavioral shifts.
- Task accuracy doesn't imply behavioral alignment.
Method
The paper formalizes online selection as a reweighted SFT objective. It proposes Alignment Drift Auditing (ADA) to quantify selection-induced behavioral movement and Alignment-Aware Selection (AAS) for constrained drift.
In practice
- Audit SFT data selection for implicit alignment.
- Define online scores to control response styles.
- Use AAS to balance efficiency and safety drift.
Topics
- Supervised Fine-tuning
- Online Data Selection
- Model Alignment
- Behavioral Drift
- Alignment Drift Auditing
- Alignment-Aware Selection
Best for: Research Scientist, AI Engineer, NLP Engineer, AI Scientist, Machine Learning Engineer, AI Ethicist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.