How Is Uncertainty Propagated in Knowledge Distillation?
On the map: Knowledge distillation → General knowledge distillation methods
Abstract (arXiv)
Knowledge distillation transfers behavior from a teacher to a student model, but the process is inherently stochastic: teacher outputs, student training, and student inference can all be random. Collapsing these uncertainties to a single point estimate can distort what is learned. We systematically study how uncertainty propagates through knowledge distillation across three representative model classes--linear regression, feed-forward neural networks, and large language models (LLMs)--and propose simple corrections. We distinguish inter-student uncertainty (variance across independently distilled students) from intra-student uncertainty (variance of a single student's predictive distribution), showing that standard single-response knowledge distillation suppresses intra-student variance while leaving substantial inter-student variability. To address these mismatches, we introduce two variance-aware strategies: averaging multiple teacher responses, which reduces noise at rate O(1/k), and variance-weighting, which combines teacher and student estimates via inverse-variance weighting to yield a minimum-variance estimator. We provide formal guarantees in linear regression, validate the methods in neural networks, and demonstrate empirical gains in LLM distillation, including reduced systematic noise and hallucination. These results reframe knowledge distillation as an uncertainty transformation and show that variance-aware distillation produces more stable students that better reflect teacher uncertainty.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in General knowledge distillation methods, one of 12 directions in Knowledge distillation. Builds on 19 of its 48 references that are in the map. First posted on arXiv 26 Jan 2026.
Builds on
19 of 48 references.
Closest ancestors · 2
Same direction, newest first
- 2021Does Knowledge Distillation Really Work?10 Jun
- 2021Knowledge distillation: A good teacher is patient and consistent9 Jun
Other references · 17
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
- 2016Sequence-Level Knowledge Distillationcited by 801
- 2019On the Efficacy of Knowledge Distillationcited by 381
- 2022Efficient Methods for Natural Language Processing: A Surveycited by 78
- 2021Mitigating Membership Inference Attacks by Self-Distillation Through a Novel Ensemble Architecturecited by 67
- 2025A Survey of Large Language Model Agents for Question Answeringcited by 48
- 2024Beyond Answers: Transferring Reasoning Capabilities to Smaller LLMs Using Multi-Teacher Knowledge Distillationcited by 47
5 more
- 2022A Survey on Model Compression and Acceleration for Pretrained Language Modelscited by 45
- 2020Meta-KD: A Meta Knowledge Distillation Framework for Language Model Compression across Domainscited by 33
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Surveycited by 26
- 2025Smoothing Out Hallucinations: Mitigating LLM Hallucination with Smoothed Knowledge Distillationcited by 10
- 2025On the Generalization vs Fidelity Paradox in Knowledge Distillationcited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026A Statistical Perspective on Knowledge Distillation: Foundations, Classical Methods, and Large Language Model Extensions
- 2023On student-teacher deviations in distillation: does it pay to disobey?
- 2026Distilling What Matters: Confidence-Aware Selective Distillation for Large Language Models
- 2026Uncertainty-Aware Knowledge Distillation for Multimodal Large Language Models
- 2026On the Generalization of Knowledge Distillation: An Information-Theoretic View
- 2026Knowledge Distillation During Mid-Training Favors Reasoning over Factual Recall
- 2023Knowledge Distillation Performs Partial Variance Reduction
- 2026SGD-Based Knowledge Distillation with Bayesian Teachers: Theory and Guidelines