机器学习中的评价指标

在人工智能领域,机器学习的效果需要用各种指标来评价。本文将阐述机器学习中的常用性能评价指标,矢量卷积与神经网格的评价指标不包括在内。

一、训练与识别

当一个机器学习模型建立好了之后,即模型训练已经完成,我们就可以利用这个模型进行分类识别。

比如,给模型输入一张电动车的照片,模型能够识别出这是一辆电动车;输入一辆摩托车的照片,模型能够识别出这是一辆摩托车。前提是:在模型训练过程中,进行了大量电动车照片、摩托车照片的反复识别训练。

image

但即便模型具备了识别电动车、摩托车的能力,并不代表每次都能百分百正确识别。当然,我们肯定希望识别正确率越高越好。识别正确率越高,代表模型性能越良好。

具体有哪些指标可以评价模型性能的优良呢?我们从下面的例子来详细了解。

例如,一个测试样本集S总共有100张照片,其中,电动车的照片有60张,摩托车的照片是40张。给模型(二分类模型)输入这100张照片进行分类识别,我们的目标是:要模型找出这100张照片中的所有电动车。这里所说的目标即为正例(Positives),非目标即为负例(Negatives)。

假设模型给出的识别结果如下图:

image

从上表结果可以看出,在100张照片中,模型识别给出了50个电动车目标,剩下50个则是摩托车。这与实际的情况有出入(实际是:电动车60个,摩托车40个),因而有些识别是错误的。正确的识别数据体现在TP和TN(T代表True),错误的识别数据则体现在FP和FN(F代表False)。

在识别给出的50个电动车目标中,其中只有40个是对的(TP:真的电动车),另外10个则识别错了(FP:假的电动车,实际是摩托车)。

以上四个识别结果数值(TP、FP 、TN、FN)就是常用的评估模型性能优良的基础参数。在进一步详细说明TP、FP 、TN、FN各符号的含义之前,我们先来了解正例(正样本)、负例(负样本)的概念。

二、正例与负例

正例(Positives):你所关注的识别目标就是正例。

负例(Negatives):正例以外的就是负例。

例如,在上面的例子中,我们关注的目标是电动车,那么电动车就是正例,剩下摩托车则是负例。

再如,假设在一个森林里,有羚羊、驯鹿、考拉三种动物,我们的目标是识别出羚羊,那么羚羊就是正例,驯鹿和考拉则是负例。

image

▲正例与负例图示1

又如,有一堆数字卡片,我们的目标是要找出含有数字8的卡片,那么含有数字8的卡片就是正例,剩于其他的都是负例。

image

▲正例与负例图示2
混淆矩阵
了解了正例(Positives)和负例(Negatives)的概念,我们就可以很好地理解TP、FN、TN、FP的各自含义(其中T代表True,F代表False,P即Positives,N即Negatives):

image

在以上四个基础参数中,真正例与真负例就是模型给出的正确的识别结果,比如电动车识别成电动车(真正例),摩托车识别成摩托车(真负例);伪正例与伪负例则是模型给出的错误的识别结果,比如摩托车识别成电动车(伪正例),电动车识别成摩托车(伪负例)。其中,真正例(TP)是评价模型性能非常关键的参数,因为这是我们所关注的目标的有用结果,该值越高越好。

可以看出,在一个数据集里,模型给出的判断结果关系如下:

image

接下来,我们就来了解模型性能的各类评价指标。

三、模型性能指标

1.正确率(Accuracy)

正确率(Accuracy):也即准确率,识别对了的正例(TP)与负例(TN)占总识别样本的比例。

即:

A=(TP+ TN)/S

在上述电动车的例子中,从上表可知,TP+ TN =70,S= 100,则正确率为:

A=70/100=0.7

通常来说,正确率越高,模型性能越好。

2. 错误率(Error-rate)

错误率(Error-rate):识别错了的正例(FP)与负例(FN)占总识别样本的比例。

即:

E=( FP+FN)/S

在上述电动车的例子中,从上表可知,FP+ FN =30,S= 100,则错误率为:

E=30/100=0.3

可见,正确率与错误率是分别从正反两方面进行评价的指标,两者数值相加刚好等于1。正确率高,错误率就低;正确率低,错误率就高。

3. 精度(Precision)

精度(Precision):识别对了的正例(TP)占识别出的正例的比例。其中,识别出的正例等于识别对了的正例加上识别错了的正例。

即:

P=TP/(TP+ FP)

在上述电动车的例子中,TP=40,TP+ FP=50。也就是说,在100张照片识别结果中,模型总共给出了50个电动车的目标,但这50个目标当中只有40个是识别正确的,则精度为:

P=40/50=0.8

因此,精度即为识别目标正确的比例。精度也即查准率,好比电动车的例子来说,模型查出了50个目标,但这50个目标中准确的比率有多少。

4.召回率(Recall)

召回率(Recall):识别对了的正例(TP)占实际总正例的比例。其中,实际总正例等于识别对了的正例加上识别错了的负例(真正例+伪负例)。

即:

R=TP/(TP+ FN)

同样,在上述电动车的例子中,TP=40,TP+FN =60。则召回率为:

R=40/60=0.67

在一定意义上来说,召回率也可以说是“找回率”,也就是在实际的60个目标中,找回了40个,找回的比例即为:40/60。同时,召回率也即查全率,即在实际的60个目标中,有没有查找完全,查找到的比率是多少。

从公式可以看出,精度与召回率都与TP值紧密相关,TP值越大,精度、召回率就越高。理想情况下,我们希望精度、召回率越高越好。但单独的高精度或高召回率,都不足以体现模型的高性能。

例如下面的例子:

高精度模型
image

从上表可以看出,该模型识别结果给出正例50个,负例200个。在识别给出的50个正例当中全部都正确(都是真正例,没有伪正例),因而精度P为100%,非常高。但是识别给出的200个负例全部都错误(都是伪负例),错误率非常高,这样的模型性能其实非常低。

高召回率模型
image

上表可以看出,该模型识别结果给出正例110个,负例0个。在110个正例当中,其中10个是真正例(识别正确),100个却是伪正例(识别错误)。在这个测试数据集中,计算的召回率R为100%,非常好,也就是说,在这个数据集里总共有10个目标,已全部找到(召回)。但同时,计算得出模型识别结果的错误率E也很高,高达91%,所以这个模型性能也很低,基本不可靠。

5.精度-召回率曲线(PR曲线)

实际中,精度与召回率是相互影响的。通常,精度高时,召回率就往往偏低,而召回率高时,精度则会偏低。这其实也很好理解,前面我们说了,精度即查准率,召回率即查全率,要想查得精准(一查一个准),即模型给出的目标都正确,那就得提高阈值门槛,阈值一提高,符合要求的目标就会减少,那必然会导致漏网之鱼增多,召回率降低。

相反,若想召回率高,没有漏网之鱼(目标都找到),就要降低阈值门槛,才能把所有目标收入囊中,与此同时会揽入一些伪目标,从而导致精度降低。

例如,在不同的阈值下(分别为0.6和0.5),模型给出15张图片的识别结果如下:

image

上表中1、0分别代表正例和负例。通过设定一个阈值(T),当置信度分数大于阈值则识别为正例,小于阈值则识别为负例。上表识别结果中当阈值T=0.6,模型给出的正例有8个,当阈值T=0.5,模型给出的正例则有10个。

通过与真实属性值核对,我们可以得出这两个阈值下的各个参数(TP、FP、FN)以及计算得出召回率(R)和精度(P)如下:

image

可以看出,设定的阈值不同,得出的召回率(R)和精度(P)也不相同。因此,对于每一个阈值可得到对应的一组(R,P),例如,上述的两个阈值可得出两组(R,P),分别为:(0.86,0.75)和(1,0.7)。如果取多个不同的阈值,就可以得到多组(R,P)。将这些坐标点(R,P)绘制在坐标上,然后将各坐标点用曲线连起来,即可得到PR曲线。

因此,PR曲线即是以召回率R为横轴,精度P为纵轴画出的曲线,如下图:

image
6. AP(Average Precision)值

PR曲线下的面积称为AP(Average Precision),表示召回率从0-1的平均精度值。如何计算AP呢?很显然,根据数学知识,可用积分进行计算,公式如下:

image

显然,这个面积的数值不会大于1。PR曲线下的面积越大,模型性能则越好。性能优的模型应是在召回率(R)增长的同时保持精度(P)值都在一个较高的水平,而性能较低的模型往往需要牺牲很多P值才能换来R值的提高。如下图所示,有两条PR曲线,可以看出,PR1曲线为性能较优的模型表现形式,PR1曲线下的面积明显大于PR2曲线下的面积。对于PR1曲线,随着R值的增长,P值仍能保持在一个较高的水平;而对于PR2曲线,随着R值的增长,P值则不断下降,因此是通过牺牲P值才能换得R值的提高。

image

除了使用积分方法计算AP值,实际应用中,还常使用插值方法进行计算。常见的一种插值方法是:选取11个精度点值,然后计算出这11个点的平均值即为AP值。

怎样选取11个精度点值呢?通常先设定一组阈值,例如[0,0.1,0.2…,1], 对于R大于每一个阈值(R>0, R>0.1,…, R>1),会得到一个对应的最大精度值Pmax,这样就会得到11个最大精度值(Pmax1, Pmax2,…, Pmax11)。

则:

AP=(Pmax1+ Pmax2+…+ Pmax11)/11

7.mAP(Mean Average Precision)值

AP是衡量模型在单个类别上平均精度的好坏,mAP则是衡量模型在所有类别上平均精度的好坏,每一个类别对应有一个AP,假设有n个类别,则有n个AP,分别为:AP1,AP2,…,APn, mAP就是取所有类别 AP 的平均值,即:

mAP= (AP1+ AP2+…+ APn)/n

8. 综合评价指标F-Measure

F-Measure又称F-Score,是召回率R和精度P的加权调和平均,顾名思义即是为了调和召回率R和精度P之间增减反向的矛盾,该综合评价指标F引入了系数α对R和P进行加权调和,表达式如下:

image

而我们最常用的F1指标,就是上式中系数α取值为1的情形,即:

F1=2P.R/(P+R)

F1的最大值为1,最小值为0。

9.ROC曲线与AUC

ROC(Receiver Operating Characteristic)曲线与AUC(Area Under the Curver)

ROC曲线,也称受试者工作特征。ROC曲线与真正率(TPR,True Positive Rate)和假正率(FPR, False Positive Rate)密切相关。

真正率(TPR): 识别对了的正例(TP)占实际总正例的比例,实际计算值跟召回率相同。即:

TPR =TP/(TP+ FN)

假正率(FPR): 识别错了的正例(FP)占实际总负例的比例。也可以说,误判的负例(实际是负例,没有判对)占实际总负例的比例。计算式如下:

FPR =FP/(FP+ TN)

以FPR为横轴,TPR为纵轴,绘制得到的曲线就是ROC曲线,绘制方法与PR曲线类似。绘制得到的ROC曲线示例如下:

image

一般来说,ROC曲线越靠近左上方越好。

ROC曲线下的面积即为AUC。面积越大代表模型的分类性能越好。如上图所示,绿线分类模型AUC=0.83大于红线分类模型AUC=0.65,因此,绿线分类模型的分类性能更优。并且,绿线较红线更光滑。通常来说,ROC曲线越光滑,过拟合程度越小。绿线分类模型的整体性能要优于红线分类模型。

10. IoU(Intersection-over-Union)指标——交并比

IoU简称交并比,顾名思义数学中交集与并集的比例。假设有两个集合A与B, IoU即等于A与B的交集除以A与B的并集,表达式如下:

IoU=A∩B/A∪B

在目标检测中,IoU为预测框(Prediction)和真实框(Ground truth)的交并比。如下图所示,在关于小猫的目标检测中,紫线边框为预测框(Prediction),红线边框为真实框(Ground truth)。

image

将预测框与真实框提取如下图,两者的交集区域为左下图斜线填充的部分,两者的并集区域为右下图蓝色填充的区域。IoU即为:

左边斜线填充的面积/右边蓝色填充的总面积。

image

预测框与真实框交集与并集示例

在目标检测任务中,通常取IoU≥0.5,认为召回。如果IoU阈值设置更高,召回率将会降低,但定位框则更加精确。

理想的情况,当然是预测框与真实框重叠越多越好,如果两者完全重叠,则交集与并集面积相同,此时IoU等于1。

11. Top1与TopK

Top1:对一张图片,模型给出的识别概率中(即置信度分数),分数最高的为正确目标,则认为正确。这里的目标也就是我们说的正例。

TopK: 对一张图片,模型给出的识别概率中(即置信度分数),分数排名前K位中包含有正确目标(正确的正例),则认为正确。

K的取值一般可在100以内的量级,当然越小越实用。比如较常见的,K取值为5,则表示为Top5,代表置信度分数排名前5当中有一个是正确目标即可;如果K取值100,则表示为Top100,代表置信度分数排名前100当中有一个是正确目标(正确的正例)即可。可见,随着K增大,难度下降。

例如,在一个数据集里,我们对前5名的置信度分数进行排序,结果如下:

image

上表中,取阈值T=0.45,排名前5的置信度分数均大于阈值,因此都识别为正例。对于Top1来说,即ID号为4的图片,实际属性却是负例,因此目标识别错误。而对于Top5来说,排名前5的置信度分数中,有识别正确的目标,即ID号为2、20的图片,因此认为正确。

在常见的人脸识别算法模型中,正确率是首当其冲的应用宣传指标。事实上,对同一个模型来说,各个性能指标也并非一个静止不变的数字,会随着应用场景、人脸库数量等变化而变化。因此,实际应用场景下的正确率跟实验室环境下所得的正确率一定是存在差距的,某种程度上说,实际应用场景下的正确率更具有评价意义。

四、相关文档推荐

前段时间在看机器学习中的相关东西,其中有些术语如精确率、召回率、平均精度和交并比,于是在网上查呀,一大堆资料,看的是有些绕,感觉主要是好多中文blog和文章说的不清楚,最后查看了wiki和一篇外文结合一些中文blog算是比较清楚了。由于这些blog和文献都说的比较好也全,在此就不重复抄袭了,我把参阅的文章和blog链接一并放置在此并做些说明。

1、https://blog.csdn.net/hysteric314/article/details/54093734#commentsedit

这篇blog是翻译这篇外文blog:https://sanchom.wordpress.com/tag/average-precision/

大体看了下,感觉翻译的很好(除个别翻译的有些别扭,感觉那篇外文有些地方写的也有点别扭),这篇介绍的很全面也很详细,而且举了一个生动的示例,很容易理解

2、http://www.cnblogs.com/xuexuefirst/p/8858274.html

这篇blog讲的很不错,特别是陈述了一些不平衡数据的问题,还有混淆矩阵等

3、https://en.wikipedia.org/wiki/Precision_and_recall

这是wiki上关于精度和召回率的一些解释说的很好,blog 2中的混淆矩阵感觉就来自其中。说实话我也不是太想wiki一下毕竟一大堆英文,但百度关于这个是在说的不好

上一张wiki有关这两个定义的截图:

image

作者 | 我的智慧生活
来源 | 咪付
原文 | 机器学习中的评价指标

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,362评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,330评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,247评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,560评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,580评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,569评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,929评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,587评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,840评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,596评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,678评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,366评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,945评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,929评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,165评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,271评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,403评论 2 342