English

Distribution Calibration for Regression

Machine Learning 2019-05-16 v1 Artificial Intelligence Machine Learning

Abstract

We are concerned with obtaining well-calibrated output distributions from regression models. Such distributions allow us to quantify the uncertainty that the model has regarding the predicted target value. We introduce the novel concept of distribution calibration, and demonstrate its advantages over the existing definition of quantile calibration. We further propose a post-hoc approach to improving the predictions from previously trained regression models, using multi-output Gaussian Processes with a novel Beta link function. The proposed method is experimentally verified on a set of common regression models and shows improvements for both distribution-level and quantile-level calibration.

Keywords

Cite

@article{arxiv.1905.06023,
  title  = {Distribution Calibration for Regression},
  author = {Hao Song and Tom Diethe and Meelis Kull and Peter Flach},
  journal= {arXiv preprint arXiv:1905.06023},
  year   = {2019}
}

Comments

ICML 2019, 10 pages

R2 v1 2026-06-23T09:07:03.528Z