arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.08099cs.LGcs.AIcs.ET
收藏

AutoML背景下预测性能的置信区间估计

Confidence Interval Estimation of Predictive Performance in the Context of AutoML

JADBio Gnosis DA S.A.(JADBio Gnosis DA 股份有限公司);Honda Research Institute Europe GmbH(本田欧洲研究中心有限公司)

Konstantinos Paraschakis, Andrea Castellani, Giorgos Borboudakis, Ioannis Tsamardinos

AI总结:

本文针对AutoML中因赢家诅咒导致的置信区间估计难题,系统比较了9种CI估计方法,并提出计算更高效的BBC-F变体,实验表明BBC-F和BBC在包含率、紧密度和时间上均占优。

摘要:

任何监督机器学习分析都需要提供样本外预测性能的估计。然而,还必须以置信区间或可信区间(CI)的形式量化该性能的不确定性,而不仅仅是给出点估计。在AutoML环境中,由于“赢家诅咒”(即由于对多个机器学习流水线进行交叉验证并选择获胜流水线而导致的估计偏差),估计CI具有挑战性。在本工作中,我们在真实和模拟数据集语料库上,对AutoML环境下CI估计的9种最先进方法及其变体进行了比较评估。这些方法在包含百分比(95% CI是否至少在95%的时间内包含真实性能)、CI紧密度(更紧的CI因信息量更大而更受欢迎)和执行时间方面进行比较。该评估是首个覆盖大多数(即便不是全部)此类方法的评估,并将先前工作扩展到不平衡和小样本任务。此外,我们提出了现有方法(Bootstrap偏差校正,即BBC)的一种变体,称为BBC-F,它保持了BBC的统计特性,但计算效率更高。结果支持BBC-F和BBC在所有测量指标上均优于其他方法。

英文摘要:

Any supervised machine learning analysis is required to provide an estimate of the out-of-sample predictive performance. However, it is imperative to also provide a quantification of the uncertainty of this performance in the form of a confidence or credible interval (CI) and not just a point estimate. In an AutoML setting, estimating the CI is challenging due to the ``winner's curse", i.e., the bias of estimation due to cross-validating several machine learning pipelines and selecting the winning one. In this work, we perform a comparative evaluation of 9 state-of-the-art methods and variants in CI estimation in an AutoML setting on a corpus of real and simulated datasets. The methods are compared in terms of inclusion percentage (does a 95\% CI include the true performance at least 95\% of the time), CI tightness (tighter CIs are preferable as being more informative), and execution time. The evaluation is the first one that covers most, if not all, such methods and extends previous work to imbalanced and small-sample tasks. In addition, we present a variant, called BBC-F, of an existing method (the Bootstrap Bias Correction, or BBC) that maintains the statistical properties of the BBC but is more computationally efficient. The results support that BBC-F and BBC dominate the other methods in all metrics measured.

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

↑