Optimal and Provable Calibration in High-Dimensional Binary Classification: Angular Calibration and Platt Scaling
Not on the map. Why?
Abstract (arXiv)
We study the fundamental problem of calibrating a linear binary classifier of the form σ(hatw^top x), where the feature vector x is Gaussian, σ is a link function, and hatw is an estimator of the true linear weight w^star. By interpolating with a noninformative textitchance classifier, we construct a well-calibrated predictor whose interpolation weight depends on the angle angle(hatw, w_star) between the estimator hatw and the true linear weight w_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 angle(hatw, w_star) can be consistently estimated. Furthermore, the resulting predictor is uniquely textitBregman-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.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 53 of its 95 references that are in the map. First posted on arXiv 21 Feb 2025.
Builds on
53 of 95 references.
References · 53
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,463
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,051
- 2014OpenML: networked science in machine learningcited by 526
- 2019Surprises in High-Dimensional Ridgeless Least Squares Interpolationcited by 513
- 2019The generalization error of random features regression: Precise asymptotics and double descent curvecited by 413
- 2020BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learningcited by 371
- 2019Beyond temperature scaling: Obtaining well-calibrated multiclass probabilities with Dirichlet calibrationcited by 364
- 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 337
38 more
- 2019Verified Uncertainty Calibrationcited by 309
- 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 231
- 2018Just Interpolate: Kernel "Ridgeless" Regression Can Generalizecited by 190
- 2019Ensemble Distribution Distillationcited by 158
- 2015Statistical physics of inference: Thresholds and algorithmscited by 128
- 2017Optimal Errors and Phase Transitions in High-Dimensional Generalized Linear Modelscited by 116
- 2020The Neural Tangent Kernel in High Dimensions: Triple Descent and a Multi-Scale Theory of Generalizationcited by 101
- 2018A modern maximum-likelihood theory for high-dimensional logistic regressioncited by 81
- 2020Distribution-free binary classification: prediction sets, confidence intervals and calibrationcited by 79
- 2015Posterior calibration and exploratory analysis for natural language processing modelscited by 76
- 2022Universality of empirical risk minimizationcited by 75
- 2020Moment Multicalibration for Uncertainty Estimationcited by 63
- 2018Evaluating Uncertainty Quantification in End-to-End Autonomous Driving Controlcited by 55
- 2021A unifying tutorial on Approximate Message Passingcited by 50
- 2018The phase transition for the existence of the maximum likelihood estimate in high-dimensional logistic regressioncited by 46
- 2020A Precise High-Dimensional Asymptotic Theory for Boosting and Minimum-ℓ1-Norm Interpolated Classifierscited by 44
- 2020The Lasso with general Gaussian designs with applications to hypothesis testingcited by 40
- 2021Don't Just Blame Over-parametrization for Over-confidence: Theoretical Analysis of Calibration in Binary Classificationcited by 39
- 2015De-biasing the Lasso: Optimal Sample Size for Gaussian Designscited by 35
- 2020Hydra: Preserving Ensemble Diversity for Model Distillationcited by 31
- 2019Isotonic Distributional Regressioncited by 28
- 2020Sample Complexity of Uniform Convergence for Multicalibrationcited by 24
- 2020Approximate Message Passing with Spectral Initialization for Generalized Linear Modelscited by 22
- 2023Classifier Calibration with ROC-Regularized Isotonic Regressioncited by 19
- 2024Optimal Ridge Regularization for Out-of-Distribution Predictioncited by 19
- 2024Fundamental computational limits of weak learnability in high-dimensional multi-index modelscited by 19
- 2022Theoretical characterization of uncertainty in high-dimensional linear classificationcited by 18
- 2020Universality of Approximate Message Passing Algorithmscited by 16
- 2024Generalization error of min-norm interpolators in transfer learningcited by 16
- 2017The Likelihood Ratio Test in High-Dimensional Logistic Regression Is Asymptotically a Rescaled Chi-Squarecited by 15
- 2020The Asymptotic Distribution of the MLE in High-dimensional Logistic Models: Arbitrary Covariancecited by 15
- 2019De-Biasing The Lasso With Degrees-of-Freedom Adjustmentcited by 12
- 2023Expectation consistency for calibration of neural networkscited by 10
- 2023Minimum-Risk Recalibration of Classifierscited by 10
- 2023Online Platt Scaling with Calibeatingcited by 8
- 2022Observable adjustments in single-index models for regularized M-estimatorscited by 5
- 2022A New Central Limit Theorem for the Augmented IPW Estimator: Variance Inflation, Cross-Fit Covariance and Beyondcited by 5
- 2024Using Platt's scaling for calibration after undersampling -- limitations and how to address themcited by 5
50 most cited of 53 All 53 (JSON)
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.