arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.02191cs.LGcs.CVstat.MEstat.ML

在不确定的真实标签下评估AI系统:皮肤病学中的案例研究

Evaluating AI systems under uncertain ground truth: a case study in dermatology

Google DeepMind(谷歌DeepMind);Google(谷歌);Bogazici University(博加齐奇大学)
David Stutz, Ali Taylan Cemgil, Abhijit Guha Roy, Tatiana Matejovicova, Melih Barsbey, Patricia Strachan, Mike Schaekermann, Jan Freyberg, Rajeev Rikhye, Beverl… 展开作者

David Stutz, Ali Taylan Cemgil, Abhijit Guha Roy, Tatiana Matejovicova, Melih Barsbey, Patricia Strachan, Mike Schaekermann, Jan Freyberg, Rajeev Rikhye, Beverly Freeman, Javier Perez Matos, Umesh Telang, Dale R. Webster, Yuan Liu, Greg S. Corrado, Yossi Matias, Pushmeet Kohli, Yun Liu, Arnaud Doucet, Alan Karthikesalingam

AI总结:

针对医疗AI评估中忽略真实标签不确定性(鉴别诊断个体不确定性与专家分歧)导致性能高估的问题,提出统计聚合方法捕捉真实标签不确定性,在皮肤病分类中证明其能提供更准确的性能及不确定性估计,指出医疗诊断需更细致的评估协议。

摘要:

为确保安全性,医疗AI系统在部署前需经过全面评估,将其预测结果与被假定为固定且确定的真实标签进行验证。然而,这种真实标签通常以鉴别诊断的形式构建。单一鉴别诊断反映了一位专家评估中的不确定性,而多位专家则通过意见分歧引入另一层不确定性。标准评估会将这些鉴别诊断聚合为单一标签,从而忽略了这两种形式的不确定性。在本文中,我们表明忽略不确定性会导致对模型性能的过度乐观估计,进而低估特定诊断决策相关的风险。为此,我们提出一种统计聚合方法,基于观察到的标注推断潜在医疗状况候选者自身概率的分布。该公式自然地考虑了不同专家之间的潜在分歧,以及源自个体鉴别诊断的不确定性,捕捉了整个真实标签的不确定性。我们的方法可归结为生成医疗状况概率的多个样本,然后基于这些采样概率评估并平均性能指标。在皮肤状况分类中,我们发现数据集的很大一部分存在显著的真实标签不确定性,标准评估在不提供不确定性估计的情况下严重高估了性能。相比之下,我们的框架对常见的关注指标(如top-k准确率和平均重叠度)提供了不确定性估计,表明性能可能会变化多个百分点。我们的结论是,虽然在许多AI应用中假定明确的真实标签是可接受的,但在医疗诊断中应采用更细致的评估协议。

英文摘要:

For safety, medical AI systems undergo thorough evaluations before deployment, validating their predictions against a ground truth which is assumed to be fixed and certain. However, this ground truth is often curated in the form of differential diagnoses. While a single differential diagnosis reflects the uncertainty in one expert assessment, multiple experts introduce another layer of uncertainty through disagreement. Both forms of uncertainty are ignored in standard evaluation which aggregates these differential diagnoses to a single label. In this paper, we show that ignoring uncertainty leads to overly optimistic estimates of model performance, therefore underestimating risk associated with particular diagnostic decisions. To this end, we propose a statistical aggregation approach, where we infer a distribution on probabilities of underlying medical condition candidates themselves, based on observed annotations. This formulation naturally accounts for the potential disagreements between different experts, as well as uncertainty stemming from individual differential diagnoses, capturing the entire ground truth uncertainty. Our approach boils down to generating multiple samples of medical condition probabilities, then evaluating and averaging performance metrics based on these sampled probabilities. In skin condition classification, we find that a large portion of the dataset exhibits significant ground truth uncertainty and standard evaluation severely over-estimates performance without providing uncertainty estimates. In contrast, our framework provides uncertainty estimates on common metrics of interest such as top-k accuracy and average overlap, showing that performance can change multiple percentage points. We conclude that, while assuming a crisp ground truth can be acceptable for many AI applications, a more nuanced evaluation protocol should be utilized in medical diagnosis.

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

↑