Why Alignment Must Precede Distillation: A Minimal Working Explanation
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
- 2024Eliminating Biased Length Reliance of Direct Preference Optimization via Down-Sampled KL Divergence16 Jun
- 2024Preference Learning Algorithms Do Not Learn Preference Rankings29 May
- 2022Training language models to follow instructions with human feedback4 Mar
Other references · 22
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2017Deep reinforcement learning from human preferencescited by 3,912
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
- 2020Learning to summarize from human feedbackcited by 2,588
- 2019HuggingFace's Transformers: State-of-the-art Natural Language Processingcited by 2,218
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,150
- 2023PaLM 2 Technical Reportcited by 1,162
10 more
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
- 2023RRHF: Rank Responses to Align Language Models with Human Feedback without tearscited by 463
- 2023Understanding the Effects of RLHF on LLM Generalisation and Diversitycited by 396
- 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 187
- 2025RL's Razor: Why Online Reinforcement Learning Forgets Lesscited by 176
- 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 152
- 2022RL with KL penalties is better viewed as Bayesian inferencecited by 135
- 2025On the Design of KL-Regularized Policy Gradient Algorithms for LLM Reasoningcited by 30
- 2025Why Knowledge Distillation Works in Generative Models: A Minimal Working Explanationcited by 9
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026How Sampling Shapes LLM Alignment: From One-Shot Optima to Iterative Dynamics
- 2025Advantage-Guided Distillation for Preference Alignment in Small Language Models
- 2024Elephant in the Room: Unveiling the Impact of Reward Model Quality in Alignment
- 2026Reward Shaping for (Inference-Time) Alignment: A Stackelberg Game Perspective
- 2025Capturing Nuanced Preferences: Preference-Aligned Distillation for Small Language Models
- 2025Beyond Ordinal Preferences: Why Alignment Needs Cardinal Human Feedback
- 2024Understanding the Learning Dynamics of Alignment with Human Feedback
- 2025The Reward Model Selection Crisis in Personalized Alignment