English

Algorithms with Calibrated Machine Learning Predictions

Machine Learning 2026-03-26 v4 Data Structures and Algorithms Machine Learning

Abstract

The field of algorithms with predictions incorporates machine learning advice in the design of online algorithms to improve real-world performance. A central consideration is the extent to which predictions can be trusted -- while existing approaches often require users to specify an aggregate trust level, modern machine learning models can provide estimates of prediction-level uncertainty. In this paper, we propose calibration as a principled and practical tool to bridge this gap, demonstrating the benefits of calibrated advice through two case studies: the ski rental and online job scheduling problems. For ski rental, we design an algorithm that achieves near-optimal prediction-dependent performance and prove that, in high-variance settings, calibrated advice offers more effective guidance than alternative methods for uncertainty quantification. For job scheduling, we demonstrate that using a calibrated predictor leads to significant performance improvements over existing methods. Evaluations on real-world data validate our theoretical findings, highlighting the practical impact of calibration for algorithms with predictions.

Keywords

Cite

@article{arxiv.2502.02861,
  title  = {Algorithms with Calibrated Machine Learning Predictions},
  author = {Judy Hanwen Shen and Ellen Vitercik and Anders Wikum},
  journal= {arXiv preprint arXiv:2502.02861},
  year   = {2026}
}

Comments

Matches the camera-ready version accepted at ICML 2025