English

Why Alignment Must Precede Distillation: A Minimal Working Explanation

Machine Learning 2025-09-30 v1

Abstract

For efficiency, preference alignment is often performed on compact, knowledge-distilled (KD) models. We argue this common practice introduces a significant limitation by overlooking a key property of the alignment's reference model: its distributional recall. We show that the standard KD -> Align workflow diminishes the model's capacity to align rare yet desirable behaviors, even under strong preference signals. We instead demonstrate that reversing the pipeline (i.e., Align -> KD) is essential: alignment must first be performed on a high-recall reference before distillation. Our contributions are threefold. First, we provide a minimal working explanation of how the reference model constrains preference alignment objectives at a fundamental level. Second, we validate this theory in a controllable Mixture-of-Gaussians experiment, where low-recall anchoring consistently results in suboptimal model performance. Finally, we demonstrate that the same phenomenon holds in LLM alignment with the SmolLM2 family: models aligned after KD fail to effectively align target behaviors, resulting in substantially lower reward and target precision. In contrast, our proposed Align -> KD pipeline robustly aligns these behaviors, yielding models with superior target-oriented metrics and lower variance. Together, these results establish reference-model recall as a first-order design choice in alignment, offering a clear principle: alignment must precede distillation.

Keywords

Cite

@article{arxiv.2509.23667,
  title  = {Why Alignment Must Precede Distillation: A Minimal Working Explanation},
  author = {Sungmin Cha and Kyunghyun Cho},
  journal= {arXiv preprint arXiv:2509.23667},
  year   = {2025}
}

Comments

Preprint

R2 v1 2026-07-01T06:02:03.012Z