Abstract:In binary classification tasks, accurate representation of probabilistic predictions is essential for various real-world applications such as predicting payment defaults or assessing medical risks. The model must then be well-calibrated to ensure alignment between predicted probabilities and actual outcomes. However, when score heterogeneity deviates from the underlying data probability distribution, traditional calibration metrics lose reliability, failing to align score distribution with actual probabilities. In this study, we highlight approaches that prioritize optimizing the alignment between predicted scores and true probability distributions over minimizing traditional performance or calibration metrics. When employing tree-based models such as Random Forest and XGBoost, our analysis emphasizes the flexibility these models offer in tuning hyperparameters to minimize the Kullback-Leibler (KL) divergence between predicted and true distributions. Through extensive empirical analysis across 10 UCI datasets and simulations, we demonstrate that optimizing tree-based models based on KL divergence yields superior alignment between predicted scores and actual probabilities without significant performance loss. In real-world scenarios, the reference probability is determined a priori as a Beta distribution estimated through maximum likelihood. Conversely, minimizing traditional calibration metrics may lead to suboptimal results, characterized by notable performance declines and inferior KL values. Our findings reveal limitations in traditional calibration metrics, which could undermine the reliability of predictive models for critical decision-making.

Predicting accurate probabilities with a ranking loss

Adversarial Prediction Games for Multivariate Losses

Predicting Rare Events by Shrinking Towards Proportional Odds

Constrained Classification and Ranking via Quantiles

Trustworthy Classification through Rank-Based Conformal Prediction Sets

From Classification Accuracy to Proper Scoring Rules: Elicitability of Probabilistic Top List Predictions

Omnipredictors for Regression and the Approximate Rank of Convex Functions

Personalized Ranking with Importance Sampling.

QuantProb: Generalizing Probabilities along with Predictions for a Pre-trained Classifier

Predictive Multiplicity in Probabilistic Classification

Obtaining Calibrated Probabilities with Personalized Ranking Models

Distribution-free risk assessment of regression-based machine learning algorithms

Probabilistic Ranking over Relations

Ranking Regularization for Critical Rare Classes: Minimizing False Positives at a High True Positive Rate

Probabilistic Scores of Classifiers, Calibration is not Enough

Conformal Prediction for Deep Classifier via Label Ranking

Predicting the Performance of Ensemble Classification Using Conditional Joint Probability

A New Probabilistic Model for Rank Aggregation.

Leveraging Uncertainty Estimates To Improve Classifier Performance

A Multiclass Classification Approach to Label Ranking

Confidence Ranking for CTR Prediction