NeurIPS2025

Optimal and Provable Calibration in High-Dimensional Binary Classification: Angular Calibration and Platt Scaling

Yufan Li, Pragya Sur

被引用 2 次

摘要

We study the fundamental problem of calibrating a linear binary classifier of the form σ(w^x)\sigma(\hat{w}^\top x), where the feature vector xx is Gaussian, σ\sigma is a link function, and w^\hat{w} is an estimator of the true linear weight ww^\star. By interpolating with a noninformative chance classifier\textit{chance classifier}, we construct a well-calibrated predictor whose interpolation weight depends on the angle (w^,w)\angle(\hat{w}, w_\star) between the estimator w^\hat{w} and the true linear weight ww_\star. We establish that this angular calibration approach is provably well-calibrated in a high-dimensional regime where the number of samples and features both diverge, at a comparable rate. The angle (w^,w)\angle(\hat{w}, w_\star) can be consistently estimated. Furthermore, the resulting predictor is uniquely Bregman-optimal\textit{Bregman-optimal}, minimizing the Bregman divergence to the true label distribution within a suitable class of calibrated predictors. Our work is the first to provide a calibration strategy that satisfies both calibration and optimality properties provably in high dimensions. Additionally, we identify conditions under which a classical Platt-scaling predictor converges to our Bregman-optimal calibrated solution. Thus, Platt-scaling also inherits these desirable properties provably in high dimensions.