mlsanityData to 14 Oct 2026

Distillation Traps and Guards: A Calibration Knob for LLM Distillability

arXiv 2604.18963 ↗First posted 21 Apr 2026cs.LG cs.AIcited by 1

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

  1. 2025Sparse Logit Sampling: Accelerating Knowledge Distillation in LLMs21 Mar
  2. 2023Baby Llama: knowledge distillation from an ensemble of teachers trained on a small dataset with no performance penalty3 Aug
  3. 2023f-Divergence Minimization for Sequence-Level Knowledge Distillation27 Jul
  4. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes23 Jun

Other references · 31

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  4. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  6. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  7. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  8. 2024MMLU-Pro: A More Robust and Challenging Multi-Task Language Understanding Benchmarkcited by 1,848
  9. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  10. 2020Knowledge Distillation: A Surveycited by 1,603
  11. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
  12. 2016Sequence-Level Knowledge Distillationcited by 801
19 more
  1. 2019Improved Knowledge Distillation via Teacher Assistantcited by 573
  2. 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
  3. 2018Born Again Neural Networkscited by 546
  4. 2025Skywork-Reward-V2: Scaling Preference Data Curation via Human-AI Synergycited by 214
  5. 2025Big-Math: A Large-Scale, High-Quality Math Dataset for Reinforcement Learning in Language Modelscited by 109
  6. 2024Revisiting Catastrophic Forgetting in Large Language Model Tuningcited by 99
  7. 2020Sample Efficient Reinforcement Learning with REINFORCEcited by 86
  8. 2025dParallel: Learnable Parallel Decoding for dLLMscited by 65
  9. 2025Part I: Tricks or Traps? A Deep Dive into RL for LLM Reasoningcited by 56
  10. 2024Revisiting Backdoor Attacks against Large Vision-Language Models from Domain Shiftcited by 49
  11. 2021Undistillable: Making A Nasty Teacher That CANNOT teach studentscited by 36
  12. 2024Unlearning Backdoor Threats: Enhancing Backdoor Defense in Multimodal Contrastive Learning via Local Token Unlearningcited by 31
  13. 2025Wait, Wait, Wait... Why Do Reasoning Models Loop?cited by 26
  14. 2025T2VShield: Model-Agnostic Jailbreak Defense for Text-to-Video Modelscited by 18
  15. 2025SafeMobile: Chain-level Jailbreak Detection and Automated Evaluation for Multimodal Mobile Agentscited by 15
  16. 2025DOGe: Defensive Output Generation for LLM Protection Against Knowledge Distillationcited by 14
  17. 2025Detoxifying Large Language Models via Autoregressive Reward Guided Representation Editingcited by 10
  18. 2025ICLShield: Exploring and Mitigating In-Context Learning Backdoor Attackscited by 9
  19. 2024TrapFlow: Controllable Website Fingerprinting Defense via Dynamic Backdoor Learningcited by 5

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. 2026Distilling What Matters: Confidence-Aware Selective Distillation for Large Language Models
  2. 2022Hard Gate Knowledge Distillation -- Leverage Calibration for Robust and Reliable Language Model
  3. 2024Pre-training Distillation for Large Language Models: A Design Space Exploration
  4. 2026Coupled Calibration and Learning: Mitigating Teacher Bias in LLM Distillation without Target-Domain Reward Feedback
  5. 2025UNDO: Understanding Distillation as Optimization
  6. 2023Can a student Large Language Model perform as well as it's teacher?
  7. 2026Curriculum Learning-Guided Progressive Distillation in Large Language Models
  8. 2026Disentangling Self-Distillation: Measuring and Modeling Acquisition and Retention