Distillation Traps and Guards: A Calibration Knob for LLM Distillability
On the map: Knowledge distillation → Knowledge distillation for LLMs
Abstract (arXiv)
Knowledge distillation (KD) transfers capabilities from large language models (LLMs) to smaller students, yet it can fail unpredictably and also underpins model leakage risks. Our analysis revealed several distillation traps: tail noise, off-policy instability, and, most fundamentally, the teacher-student gap, that distort training signals. These traps manifest as overconfident hallucinations, self-correction collapse, and local decoding degradation, causing distillation to fail. Motivated by these findings, we propose a post-hoc calibration method that, to the best of our knowledge, for the first time enables control over a teacher's distillability via reinforcement fine-tuning (RFT). Our objective combines task utility, KL anchor, and across-tokenizer calibration reward. This makes distillability a practical safety lever for foundation models, connecting robust teacher-student transfer with deployment-aware model protection. Experiments across math, knowledge QA, and instruction-following tasks show that students distilled from distillable calibrated teachers outperform SFT and KD baselines, while undistillable calibrated teachers retain their task performance but cause distilled students to collapse, offering a practical knob for both better KD and model IP protection.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Knowledge distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 35 of its 51 references that are in the map. First posted on arXiv 21 Apr 2026.
Builds on
35 of 51 references.
Closest ancestors · 4
Same direction, newest first
- 2025Sparse Logit Sampling: Accelerating Knowledge Distillation in LLMs21 Mar
- 2023Baby Llama: knowledge distillation from an ensemble of teachers trained on a small dataset with no performance penalty3 Aug
- 2023f-Divergence Minimization for Sequence-Level Knowledge Distillation27 Jul
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes23 Jun
Other references · 31
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 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
- 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,848
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2020Knowledge Distillation: A Surveycited by 1,603
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
- 2016Sequence-Level Knowledge Distillationcited by 801
19 more
- 2019Improved Knowledge Distillation via Teacher Assistantcited by 573
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
- 2018Born Again Neural Networkscited by 546
- 2025Skywork-Reward-V2: Scaling Preference Data Curation via Human-AI Synergycited by 214
- 2025Big-Math: A Large-Scale, High-Quality Math Dataset for Reinforcement Learning in Language Modelscited by 109
- 2024Revisiting Catastrophic Forgetting in Large Language Model Tuningcited by 99
- 2020Sample Efficient Reinforcement Learning with REINFORCEcited by 86
- 2025dParallel: Learnable Parallel Decoding for dLLMscited by 65
- 2025Part I: Tricks or Traps? A Deep Dive into RL for LLM Reasoningcited by 56
- 2024Revisiting Backdoor Attacks against Large Vision-Language Models from Domain Shiftcited by 49
- 2021Undistillable: Making A Nasty Teacher That CANNOT teach studentscited by 36
- 2024Unlearning Backdoor Threats: Enhancing Backdoor Defense in Multimodal Contrastive Learning via Local Token Unlearningcited by 31
- 2025Wait, Wait, Wait... Why Do Reasoning Models Loop?cited by 26
- 2025T2VShield: Model-Agnostic Jailbreak Defense for Text-to-Video Modelscited by 18
- 2025SafeMobile: Chain-level Jailbreak Detection and Automated Evaluation for Multimodal Mobile Agentscited by 15
- 2025DOGe: Defensive Output Generation for LLM Protection Against Knowledge Distillationcited by 14
- 2025Detoxifying Large Language Models via Autoregressive Reward Guided Representation Editingcited by 10
- 2025ICLShield: Exploring and Mitigating In-Context Learning Backdoor Attackscited by 9
- 2024TrapFlow: Controllable Website Fingerprinting Defense via Dynamic Backdoor Learningcited by 5
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Distilling What Matters: Confidence-Aware Selective Distillation for Large Language Models
- 2022Hard Gate Knowledge Distillation -- Leverage Calibration for Robust and Reliable Language Model
- 2024Pre-training Distillation for Large Language Models: A Design Space Exploration
- 2026Coupled Calibration and Learning: Mitigating Teacher Bias in LLM Distillation without Target-Domain Reward Feedback
- 2025UNDO: Understanding Distillation as Optimization
- 2023Can a student Large Language Model perform as well as it's teacher?
- 2026Curriculum Learning-Guided Progressive Distillation in Large Language Models
- 2026Disentangling Self-Distillation: Measuring and Modeling Acquisition and Retention