English

Credal Ensemble Distillation for Uncertainty Quantification

Machine Learning 2025-11-19 v1 Artificial Intelligence

Abstract

Deep ensembles (DE) have emerged as a powerful approach for quantifying predictive uncertainty and distinguishing its aleatoric and epistemic components, thereby enhancing model robustness and reliability. However, their high computational and memory costs during inference pose significant challenges for wide practical deployment. To overcome this issue, we propose credal ensemble distillation (CED), a novel framework that compresses a DE into a single model, CREDIT, for classification tasks. Instead of a single softmax probability distribution, CREDIT predicts class-wise probability intervals that define a credal set, a convex set of probability distributions, for uncertainty quantification. Empirical results on out-of-distribution detection benchmarks demonstrate that CED achieves superior or comparable uncertainty estimation compared to several existing baselines, while substantially reducing inference overhead compared to DE.

Keywords

Cite

@article{arxiv.2511.13766,
  title  = {Credal Ensemble Distillation for Uncertainty Quantification},
  author = {Kaizheng Wang and Fabio Cuzzolin and David Moens and Hans Hallez},
  journal= {arXiv preprint arXiv:2511.13766},
  year   = {2025}
}

Comments

An extended version for Credal Ensemble Distillation for Uncertainty Quantification, which has been accepted for publication at AAAI 2026

R2 v1 2026-07-01T07:41:57.175Z