Calibrating Student Models via Uncertainty-Integrated Distillation

Ruixuan Zhang, Hongye Tan, Zhiwei Hu, XiaoQi Han, Ru Li, Jeff Z. Pan, Victor Gutierrez Basulto


Abstract
Model calibration aligns a model’s confidence with the accuracy of its predictions, directly influencing reliability and user trust. Surprisingly, existing research on knowledge distillation has paid little attention to the calibration of student models. In this paper, we evaluate calibration across diverse knowledge distillation settings and demonstrate that student models distilled from black-box teachers generally exhibit poor calibration. We attribute this to the fact that distillation fails to convey well-calibrated confidence, which serves as a crucial expression of predictive uncertainty. To address this limitation, we propose an ***U**ncertainty-**I**ntegrated **K**nowledge **D**istillation* (**UIKD**) method comprising two key components: (1) *uncertainty estimation*, which leverages the collective intelligence of multiple LLMs via a debate mechanism to quantify uncertainty in the distillation data; and (2) *uncertainty integration*, which employs an adaptive weighted loss function to integrate this data uncertainty into the student model effectively. Experiments on common-sense reasoning and natural language inference benchmarks demonstrate that UIKD consistently improves student model calibration while maintaining predictive accuracy. Our code is available at https://github.com/RuixuanZhang127/UIKD.
Anthology ID:
2026.emnlp-main.1930
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
43923–43938
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1930/
DOI:
Bibkey:
Cite (ACL):
Ruixuan Zhang, Hongye Tan, Zhiwei Hu, XiaoQi Han, Ru Li, Jeff Z. Pan, and Victor Gutierrez Basulto. 2026. Calibrating Student Models via Uncertainty-Integrated Distillation. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 43923–43938, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Calibrating Student Models via Uncertainty-Integrated Distillation (Zhang et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1930.pdf