arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.03421cs.LGstat.ML

分类器的概率分数:仅校准并不足够

Probabilistic Scores of Classifiers, Calibration is not Enough

Université du Québec à Montréal(蒙特利尔大学(魁北克大学蒙特利尔分校));Aix Marseille Univ(艾克斯-马赛大学);CNRS(法国国家科学研究中心);AMSE(马赛经济与管理科学研究所);Milliman France(美世法国公司)

Agathe Fernandes Machado, Arthur Charpentier, Emmanuel Flachaire, Ewen Gallic, François Hu

AI总结:

本文研究二分类中概率分数与真实分布的对齐问题,提出基于KL散度优化随机森林和XGBoost等树模型,并在10个UCI数据集及模拟实验中验证其优于传统校准指标,且无明显性能损失。

摘要:

在二分类任务中,准确表示概率预测对于预测支付违约或评估医疗风险等各类现实应用至关重要。模型必须经过良好校准,以确保预测概率与实际结果之间保持一致。然而,当分数异质性与底层数据概率分布发生偏离时,传统校准指标会失去可靠性,无法使分数分布与实际概率对齐。在本研究中,我们重点介绍那些优先优化预测分数与真实概率分布之间对齐程度、而非最小化传统性能或校准指标的方法。在使用随机森林(Random Forest)和XGBoost等基于树的模型时,我们的分析强调了这些模型在调整超参数以最小化预测分布与真实分布之间的Kullback-Leibler(KL)散度方面所提供的灵活性。通过在10个UCI数据集和模拟实验上进行广泛的实证分析,我们证明基于KL散度优化树模型能够在没有显著性能损失的情况下,实现预测分数与实际概率之间更优的对齐。在现实场景中,参考概率被先验地确定为通过最大似然估计得到的Beta分布。相反,最小化传统校准指标可能导致次优结果,其特征是性能明显下降以及KL值较差。我们的发现揭示了传统校准指标的局限性,这些局限性可能削弱预测模型在关键决策中的可靠性。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑