English

Diversity regularization in deep ensembles

Machine Learning 2018-02-23 v1

Abstract

Calibrating the confidence of supervised learning models is important for a variety of contexts where the certainty over predictions should be reliable. However, it has been reported that deep neural network models are often too poorly calibrated for achieving complex tasks requiring reliable uncertainty estimates in their prediction. In this work, we are proposing a strategy for training deep ensembles with a diversity function regularization, which improves the calibration property while maintaining a similar prediction accuracy.

Keywords

Cite

@article{arxiv.1802.07881,
  title  = {Diversity regularization in deep ensembles},
  author = {Changjian Shui and Azadeh Sadat Mozafari and Jonathan Marek and Ihsen Hedhli and Christian Gagné},
  journal= {arXiv preprint arXiv:1802.07881},
  year   = {2018}
}

Comments

6 pages

R2 v1 2026-06-23T00:29:39.338Z