mlsanityData to 14 Oct 2026

Why Alignment Must Precede Distillation: A Minimal Working Explanation

arXiv 2509.23667 ↗First posted 28 Sep 2025cs.LGcited by 1

On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods

Abstract (arXiv)

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.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 25 of its 29 references that are in the map. First posted on arXiv 28 Sep 2025.

Builds on

25 of 29 references.

Closest ancestors · 3

Same direction, newest first

  1. 2024Eliminating Biased Length Reliance of Direct Preference Optimization via Down-Sampled KL Divergence16 Jun
  2. 2024Preference Learning Algorithms Do Not Learn Preference Rankings29 May
  3. 2022Training language models to follow instructions with human feedback4 Mar

Other references · 22

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  5. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  6. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  7. 2017Deep reinforcement learning from human preferencescited by 3,912
  8. 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
  9. 2020Learning to summarize from human feedbackcited by 2,588
  10. 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
  11. 2019Fine-Tuning Language Models from Human Preferencescited by 2,150
  12. 2023PaLM 2 Technical Reportcited by 1,162
10 more
  1. 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
  2. 2023RRHF: Rank Responses to Align Language Models with Human Feedback without tearscited by 463
  3. 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 396
  4. 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
  5. 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 187
  6. 2025RL's Razor: Why Online Reinforcement Learning Forgets Lesscited by 176
  7. 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 152
  8. 2022RL with KL penalties is better viewed as Bayesian inferencecited by 135
  9. 2025On the Design of KL-Regularized Policy Gradient Algorithms for LLM Reasoningcited by 30
  10. 2025Why Knowledge Distillation Works in Generative Models: A Minimal Working Explanationcited by 9

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026How Sampling Shapes LLM Alignment: From One-Shot Optima to Iterative Dynamics
  2. 2025Advantage-Guided Distillation for Preference Alignment in Small Language Models
  3. 2024Elephant in the Room: Unveiling the Impact of Reward Model Quality in Alignment
  4. 2026Reward Shaping for (Inference-Time) Alignment: A Stackelberg Game Perspective
  5. 2025Capturing Nuanced Preferences: Preference-Aligned Distillation for Small Language Models
  6. 2025Beyond Ordinal Preferences: Why Alignment Needs Cardinal Human Feedback
  7. 2024Understanding the Learning Dynamics of Alignment with Human Feedback
  8. 2025The Reward Model Selection Crisis in Personalized Alignment