机器学习任务的评价指标(以MSE,precision-recall,NDCG为例)

    评价指标是机器学习任务中非常重要的一环。不同的机器学习任务有着不同的评价指标,同时同一种机器学习任务也有着不同的评价指标,每个指标的着重点不一样。如分类、回归、排序、聚类、热门主题模型、推荐等。并且很多指标可以对多种不同的机器学习模型进行评价,如精确率-召回率(precision-recall),可以用在分类、推荐、排序等


在这里我分享均方误差(MSE)、精确率-召回率(precision-recall)和归一化折损累计增益(NDCG)三个不同用处的评价指标。

一、均方误差(MSE)

均方误差(MSE),检测模型预测值和真实值的偏差,可用在回归任务

对于回归任务,我们得到的是结果是数值,不能像分类任务一样要求与真实值完全一致,只能是比较与真实值的偏差

比如,一个人实际身高173.5,有人说他174,他也不会否认,因为与实际身高只差0.5而已

式子:

范围[0,+∞),当预测值与真实值完全吻合时等于0,即完美模型;误差越大,该值越大


二、 精确率-召回率(precision-recall)

精确率-召回率(precision-recall),可以用在分类、推荐、排序任务等

混淆矩阵

如上图所示,要了解各个评价指标,首先需要知道混淆矩阵,混淆矩阵中的P表示Positive,即正例或者阳性,N表示Negative,即负例或者阴性,T表示预测正确的,F表示预测错误的

为比较精确率和召回率的区别,我们举一个识别垃圾邮件的例子

2.1 召回率=TP/(TP+FN):正确判为垃圾邮件占所有垃圾邮件的比例

我们可以想象一种情况:

如果我们把所有邮件都判为垃圾邮件,那真正的垃圾邮件确实会被百分百预测正确,所有垃圾邮件都被删除了,召回率就为100%!但我们凭直觉看,这不能是一个好的模型,因为如果按照这个识别,你等于一辈子收不到邮件

2.2 精确率=TP(TP+FP):所有判为垃圾邮件中,真正的垃圾邮件的比例

如果像上面判定所有都是垃圾邮件,那精确率就是为0的情况,精确率低的时候我们会失去重要的邮件,这就有点本末倒置了

如果把仅把一封邮件正确判定为垃圾邮件,精确率为1,召回率就为0。但因为实际上还存在其他许多垃圾邮件,就相当于没解决识别垃圾邮件的任务,这样也不好

2.3 精确度和召回率权衡

不难发现,精度和召回率成负相关,我们不能仅追求其中一个的指标,此时就需要根据任务需求权衡精度—召回率

例如在逃犯信息检索系统中,更希望尽可能少漏掉逃犯,此时召回率更重要。可使用F1和Fβ度量调和平均,使用ROC和AUC曲线可视化

三、归一化折损累计增益(NDCG)

这个指标通常用来衡量和评价搜索结果算法,可也用于排序、推荐任务

3.1 累计增量(CG)

它是一个搜索结果相关性分数的总和

[原因]:

我们搜索都希望能找到想要的东西,于是把我们期望结果和实际结果的相关性数值化

[式子]:

rel_{i} 即为单个的相关性分数,分数越大,相关性越大

3.2 折损累计增益(DCG)

Discounted的CG,就是在每一个CG的结果上处以一个折损值

[原因]:

对于搜索来说,都是优先先查看前面几个,前面的相关性大小很大影响我们能否找到期望结果

[式子]:

\log_2 (i+1) 即为折损值,排序越往后,价值越低

3.3 归一化折损累计增益(NDCG)

Normalized的DCG,就是每一个DCG结果除以一个IDCG

[原因]:

搜索时,我们搜索词不一样,搜索结果的数量是不一样的,使得DCG之间无法比较,于是需要归一化处理,使得所有nDCG在[0,1]之间比较

[式子]:

IDCG就是理想的DCG,即把最相关的排到最前面的情况

IDCG为排序的最大值,所以分子一定不大于分母,使得不同结果能在[0,1]之间比较,nDCG越大越好

3.4 例子

查询一:淘宝搜索飞机模型,出现四个搜索结果(取两位小数)

[1] 衡量搜索算法结果的相关性

CG_{4}  =1+2+3+0=6

[2] 强调排序在前的更被关注

DCG_{4}  =1+1.27+1.5+0=3.77

[3] 比较搜索引擎在一个查询和下一个查询中的性能

查询二:淘宝搜索牛仔裤,出现三个搜索结果

同理可得DCG_{3}  =3.5

因为以上面的搜索结果数量不一样不好比较,所以我们引入NDCG

①飞机模型搜索理想排序是D3 D2 D1 D4:

IDCG_{4}  =3+1.27+0.5+0=4.77

nDCG_{4}  =3.77÷4.77=0.79

②牛仔裤搜索理想排序是D3 D1 D2:

IDCG_{3}  =3+1.27+0=4.27

nDCG_{3}  =3.5÷4.27=0.82

这样标准化后就可以很好的比较了两者的性能

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

推荐阅读更多精彩内容