Online Data Selection Is Implicit Alignment

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

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

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

Topics

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.