To Trust or Not to Trust: On Calibration in ML-based Resource Allocation for Wireless Networks

TL;DR

This paper analyzes the calibration of machine learning models in resource allocation for wireless networks, establishing theoretical properties of outage probability and demonstrating calibration techniques through simulations on realistic channel models.

Contribution

It provides new theoretical insights into outage probability under calibration, guides threshold selection, and shows limitations of post-processing calibration in wireless resource management.

Findings

Perfect calibration aligns outage probability with expected output as resources increase.

Post-processing calibration cannot reduce the minimum outage probability.

Well-calibrated models can improve outage probability under certain conditions.

Abstract

In next-generation communications and networks, machine learning (ML) models are expected to deliver not only accurate predictions but also well-calibrated confidence scores that reflect the true likelihood of correct decisions. This paper studies the calibration performance of an ML-based outage predictor within a single-user, multi-resource allocation framework. We first establish key theoretical properties of this system's outage probability (OP) under perfect calibration. Importantly, we show that as the number of resources grows, the OP of a perfectly calibrated predictor approaches the expected output conditioned on it being below the classification threshold. In contrast, when only one resource is available, the system's OP equals the model's overall expected output. We then derive the OP conditions for a perfectly calibrated predictor. These findings guide the choice of the classification threshold to achieve a desired OP, helping system designers meet specific reliability requirements. We also demonstrate that post-processing calibration cannot improve the system's minimum achievable OP, as it does not introduce new information about future channel states. Additionally, we show that well-calibrated models are part of a broader class of predictors that necessarily improve OP. In particular, we establish a monotonicity condition that the accuracy-confidence function must satisfy for such improvement to occur. To demonstrate these theoretical properties, we conduct a rigorous simulation-based analysis using post-processing calibration techniques: Platt scaling and isotonic regression. As part of this framework, the predictor is trained using an outage loss function specifically designed for this system. Furthermore, this analysis is performed on Rayleigh fading channels with temporal correlation captured by Clarke's 2D model, which accounts for receiver mobility.