使用 ROC 曲线评估和比较分类器性能

分类器性能最常报告的指标是准确率:即正确分类的百分比。

此指标的优点是易于理解,并且可以轻松地比较不同分类器的性能,但它忽略了许多在诚实评估分类器性能时应考虑的因素。

什么是分类器性能?

分类器性能不仅仅是正确分类的数量。

例如,考虑筛查宫颈癌等相对罕见疾病的问题,其患病率约为 10%(实际统计数据)。如果一个懒惰的巴氏涂片筛查员将他们看到的每个样本都归类为“正常”,那么他们的准确率将为 90%。非常令人印象深刻!但这个数字完全忽略了 10% 的患病女性根本没有被诊断出来的事实。

一些性能指标

在之前的一篇博文中,我们讨论了一些可用于评估分类器的其他性能指标。回顾一下

大多数分类器会产生一个分数,然后对该分数进行阈值处理以决定分类。如果分类器产生一个介于 0.0(肯定为阴性)和 1.0(肯定为阳性)之间的分数,通常认为大于 0.5 的任何值都为阳性。

但是,应用于数据集(其中 PP 是阳性人群,NP 是阴性人群)的任何阈值都将产生真阳性(TP)、假阳性(FP)、真阴性(TN)和假阴性(FN)(图 1)。我们需要一种方法来考虑所有这些数字。

ROC Curve Explaination

图 1. 重叠的数据集将始终产生假阳性和假阴性以及真阳性和真阴性

一旦您拥有了所有这些度量的数字,就可以计算出一些有用的度量。

  • 准确率 = (1 – 错误率) = (TP + TN)/(PP + NP) = Pr(C),正确分类的概率。
  • 灵敏度 = TP/(TP + FN) = TP/PP = 测试在患病人群中检测疾病的能力。
  • 特异度 = TN/(TN + FP) = TN / NP = 测试在无病人群中正确排除疾病的能力。

让我们为一些合理的实际数字计算这些度量。如果我们有 100,000 名患者,其中 200 名(20%)实际上患有癌症,我们可能会看到以下测试结果(表 1)

Table of sample data

表 1. 巴氏涂片筛查“合理”值的诊断测试性能说明

对于这些数据

  • 灵敏度 = TP/(TP + FN) = 160 / (160 + 40) = 80.0%
    特异度 = TN/(TN + FP) = 69,860 / (69,860 + 29,940) = 70.0%

换句话说,我们的测试将正确识别 80% 的患者,但 30% 的健康人将被错误地检测为阳性。仅考虑测试的灵敏度(或准确率),可能会丢失重要信息。

通过考虑我们错误的以及正确的答案,我们可以更深入地了解分类器的性能。

克服必须选择截止值的难题的一种方法是,从 0.0 的阈值开始,这样所有情况都被视为阳性。我们正确地分类了所有阳性病例,并错误地分类了所有阴性病例。然后,我们将阈值移动到 0.0 和 1.0 之间的每个值,逐渐减少假阳性数量并增加真阳性数量。

然后,可以将 TP(灵敏度)与 FP(1 – 特异度)绘制在每个使用的阈值上。生成的图称为接收者操作特征(ROC)曲线(图 2)。ROC 曲线是在 1950 年代用于雷达回波的信号检测,此后已应用于广泛的问题。

Example ROC Curves

图 2. ROC 曲线示例

对于完美的分类器,ROC 曲线将直接沿着 Y 轴向上延伸,然后沿着 X 轴向右延伸。没有区分能力的分类器将位于对角线上,而大多数分类器则介于两者之间。

ROC 分析提供了独立于(并且在指定)成本背景或类别分布的工具,用于选择可能的最佳模型并丢弃次优模型

接收者操作特征维基百科文章

使用 ROC 曲线

阈值选择

显而易见,ROC 曲线可用于选择分类器的阈值,该阈值可最大化真阳性,同时最小化假阳性。

但是,不同类型的问题具有不同的最佳分类器阈值。例如,对于癌症筛查测试,我们可能愿意接受相对较高的假阳性率以获得高真阳性,识别可能的癌症患者是最重要的。

然而,对于治疗后的随访测试,可能需要一个不同的阈值,因为我们希望最小化假阴性,我们不希望告诉患者他们已康复(如果事实并非如此)。

性能评估

ROC 曲线还使我们能够评估分类器在其整个操作范围内的性能。最广泛使用的度量是曲线下的面积(AUC)。从图 2 可以看出,没有区分能力、基本上随机猜测的分类器的 AUC 为 0.5,因为曲线沿着对角线。那个神话般的完美分类器的 AUC 为 1.0。大多数分类器的 AUC 介于这两个值之间。

AUC 小于 0.5 可能表明发生了一些有趣的事情。非常低的 AUC 可能表明问题设置错误,分类器发现了数据中的一种关系,这种关系本质上与预期相反。在这种情况下,检查整个 ROC 曲线可能会提供一些关于正在发生什么的线索:是阳性和阴性被错误标记了吗?

分类器比较

AUC 可用于比较两个或多个分类器的性能。可以选择一个阈值来比较该点的分类器性能,或者通过考虑 AUC 来比较整体性能。

大多数已发表的报告以绝对方式比较 AUC:“分类器 1 的 AUC 为 0.85,分类器 2 的 AUC 为 0.79,因此分类器 1 明显更好”。但是,可以计算 AUC 差异是否在统计学上显著。有关详细信息,请参阅下面的 Hanley & McNeil (1982) 论文。

ROC 曲线分析教程

何时使用 ROC 曲线分析

在这篇文章中,我使用了生物医学示例,ROC 曲线广泛用于生物医学领域。然而,该技术适用于任何产生每个案例分数而非二元决策的分类器。

神经网络和许多统计算法是合适的分类器的示例,而决策树等方法则不太适合。只有两个可能结果的算法(如这里使用的癌症/无癌症示例)最适合此方法。

任何可以输入到适当分类器中的数据都可以进行 ROC 曲线分析。

进一步阅读

对《使用 ROC 曲线评估和比较分类器性能》的 17 条回复

  1. Kevin Markham 2015年1月15日上午5:33 #

    对 ROC 曲线和 AUC 的非常有见地的总结!

    我制作了一个 14 分钟的视频,用简单的可视化解释了 ROC 曲线,我想分享它,因为我的许多学生都觉得它很有帮助:http://www.dataschool.io/roc-curves-and-auc-explained/

  2. Jeff 2016年8月26日上午7:56 #

    很棒的文章,信息量很大!我可以看到,当人群中有一个被用来区分两个类别的测量值时,从低到高递增决策阈值并获得每个增量的灵敏度和特异度结果的想法是很直观的。但是,当有两个或多个测量值被用来预测响应时,我们该怎么办?

    为了增加额外的关注点,我了解到各种分类器方程以各种方式生成决策阈值。多维决策阈值的方法将如何因分类器而异?

    谢谢!

  3. CiCi 2016年9月6日晚上7:19 #

    很棒的文章,信息量很大!您能告诉我您用什么工具绘制了这篇文章中的两个图吗?我喜欢线的风格,我想尝试一下。
    谢谢!

  4. Divya Nair 2017年2月10日晚上7:11 #

    非常感谢这篇文章。它确实很有信息量。如果能提供更多参考文献将很有用。

  5. edward cooper 2017年9月7日凌晨2:13 #

    我最近写了一篇关于 ROC 曲线与第一类错误和统计检验功效关系的博文。希望它能为这篇出色的 ROC 文章提供更多背景信息。

    https://phdstatsphys.wordpress.com/2017/09/04/roc-curve-with-type-i-and-ii-error-the-short-story/

  6. eden 2018年8月18日上午5:23 #

    你好,
    我正在为二元分类器(神经网络)考虑 ROC 曲线,但有点困惑。由于它是一个离散分类器,它只有 0 和 1 作为响应,所以我无法更改阈值。我也知道你可以使用任何协变量作为阈值,但在神经网络中没有单独的特征可以作为协变量使用。因此,据我理解,我只能获得灵敏度和特异度值,而不能得到一条曲线。我的理解正确吗?

  7. Royath 2018年11月2日上午11:30 #

    你好,

    随着我们将阈值从 0 增加到 1,在某个值之后,假阳性和真阳性都会减少,它们不像这里建议的那样具有反比关系:“然后,我们将阈值移动到 0.0 和 1.0 之间的每个值,逐渐减少假阳性数量并增加真阳性数量。” 或者是我理解错了。例如:如果阈值为 1,那么真阳性为零。

    谢谢。

  8. Ana 2018年12月6日上午9:00 #

    嗨,感谢这篇文章,有应用示例总是更容易理解。有几点说明:200 不是 100,000 的 20%,而是 0.2%。所有阳性表示阈值为 1,而不是 0。在该阈值(所有阳性)下,TP 为 1,FP 也为 1。通过降低阈值,TP 和 FP 都会降低。我认为更直观的解释是,从阈值 0(所有阴性)开始,此时没有阳性病例,TP 和 FP 都为 0。通过增加阈值,我们会获得更高的 TP,但也会获得更高的 FP(增益-损失平衡)。

  9. Martin 2019年1月26日凌晨2:44 #

    嗨,
    感谢您一如既往的文章。
    不过我有一个问题,如何自动选择一个阈值,既能最大化 TP 率又能最小化 FP 率?

    提前感谢!

    • Jason Brownlee 2019年1月26日上午6:16 #

      也许线性测试每个阈值,直到找到合适的权衡?

  10. Dip 2019年3月11日晚上8:28 #

    谢谢 Jason。这确实非常有用。一个快速的问题——这是否意味着 ROC 曲线只能用于逻辑回归等概率分类器,而不能用于决策树等离散分类器?

    • Jason Brownlee 2019年3月12日上午6:49 #

      是的。尽管可以使用方法从树/SVM 等中获取概率,然后进行校准并与 ROC 一起使用。

  11. Sharath 2021年3月9日晚上11:56 #

    1) 200 不是 100,000 的 20%

    2) 在计算灵敏度时,您提到

    灵敏度 = TP/(TP + FN) = 160 / (160 + 40) = 80.0%

    但 FN 应该是 29,940,对吧??

  12. JeanD 2021年8月4日晚上11:44 #

    嗨,Jason,
    在描述 ROC 曲线的构建方式时,您写道

    “然后,我们将阈值移动到 0.0 和 1.0 之间的每个值,逐渐减少假阳性数量并增加真阳性数量。”

    虽然我明白提高阈值会减少假阳性数量(因为总的被分类为阳性的点更少),但如何增加真阳性数量呢?您是不是指真阴性?

    提高阈值将使分类器对它分类为阳性的内容更加严格,所以我看不到这如何能增加真阳性数量。

    谢谢

留下回复

Machine Learning Mastery 是 Guiding Tech Media 的一部分,Guiding Tech Media 是一家领先的数字媒体出版商,专注于帮助人们了解技术。访问我们的公司网站以了解更多关于我们的使命和团队的信息。