信息检索常用的评价指标(MAP,NDCG,ERR)

一、MAP

Precision(p):

准确率(精确度)是指检索得到的文档中相关文档所占的比例,公式如下:

p r e c i s i o n = | { r e l e v a n t   d o c u m e n t s } { r e t r i e v e d   d o c u m e n t s } | | { r e t r i e v e d   d o c u m e n t s } |

p r e c i s i o n @ 10 表示检索10个文档中包含相关所占的比例,即: | { r e t r i e v e d   d o c u m e n t s } | = 10

Recall(R):

召回率是指所有相关文档中被检索到的比例,公式如下:

r e c a l l = | { r e l e v a n t   d o c u m e n t s } { r e t r i e v e d   d o c u m e n t s } | | { r e l e v a n t   d o c u m e n t s } |

Average precision(AveP):

由前面可知,准确率和召回率都只能衡量检索性能的一个方面,最理想的情况肯定是准确率和召回率都比较高。当我们想提高召回率的时候,肯定会影响准确率,所以可以把准确率看做是召回率的函数,即: P = f ( R ) ,也就是随着召回率从0到1,准确率的变化情况。那么就可以对函数 P = f ( R ) R 上进行积分,可以求 P 的期望均值。公式如下:

A v e P = 0 1 P ( r ) d r = k = 1 n P ( k ) Δ ( k ) = k = 1 n ( P ( k ) × r e l ( k ) ) n u m b e r   o f   r e l e v a n t   d o c u m e n t s

其中 r e l ( k ) 表示第 k 个文档是否相关,若相关则为1,否则为0, P ( k ) 表示前 k 个文档的准确率。 A v e P 的计算方式可以简单的认为是:

A v e P = 1 R × r = 1 R r p o s i t i o n ( r )

其中 R 表示相关文档的总个数, p o s i t i o n ( r ) 表示,结果列表从前往后看,第 r 个相关文档在列表中的位置。比如,有三个相关文档,位置分别为1、3、6,那么 A v e P = 1 3 × ( 1 1 + 2 3 + 3 6 ) 。在编程的时候需要注意,位置和第i个相关文档,都是从1开始的,不是从0开始的。

A v e P 意义是在召回率从0到1逐步提高的同时,对每个 R 位置上的 P 进行相加,也即要保证准确率比较高,才能使最后的 A v e P 比较大。

Mean average precision(MAP):

通常会用多个查询语句来衡量检索系统的性能,所以应该对多个查询语句的AveP求均值(the mean of average precision scores),即公式:

M A P = q = 1 Q A v e P ( q ) Q

二、NDCG

在MAP计算公式中,文档只有相关不相关两种,而在nDCG中,文档的相关度可以分多个等级进行打分。

Cumulative Gain(CG):

表示前p个位置累计得到的效益,公式如下:

C G n = i = 1 p r e l i

其中 r e l i 表示第 i 个文档的相关度等级,如:2表示非常相关,1表示相关,0表示无关,-1表示垃圾文件。

Discounted cumulative gain(DCG):

由于在 C G p 的计算中对位置信息不敏感,比如检索到了三个文档相关度依次是{3,-1,1}和{-1,1,3},显然前面的排序更优,但是它们的 C G 相同,所以要引入对位置信息的度量计算,既要考虑文档的相关度等级,也要考虑它所在的位置信息。假设每个位置按照从小到大的排序,它们的价值依次递减,如:可以假设第i个位置的价值是 1 l o g 2 ( i + 1 ) ,那么排在第i个位置的文档所产生的效益就是 r e l i × 1 l o g 2 ( i + 1 ) = r e l i l o g 2 ( i + 1 ) 。公式如下:

D C G p = i = 1 p r e l i l o g 2 ( i + 1 ) = r e l 1 + i = 2 p r e l i l o g 2 ( i + 1 )

另一种比较常用的,用来增加相关度影响比重的 D C G 计算方式是:

D C G p = i = 1 p 2 r e l i 1 l o g 2 ( i + 1 )

Ideal DCG(IDCG):

IDCG是理想情况下的DCG,即对于一个查询语句和p来说,DCG的最大值。公式如下:

I D C G p = i = 1 | R E L | 2 r e l i 1 l o g 2 ( i + 1 )

其中 | R E L | 表示,文档按照相关性从大到小的顺序排序,取前p个文档组成的集合。也就是按照最优的方式对文档进行排序。

Normalize DCG(NDCG):

由于每个查询语句所能检索到的结果文档集合长度不一, p 值的不同会对 D C G 的计算有较大的影响。所以不能对不同查询语句的 D C G 进行求平均,需要进行归一化处理。 N D C G 就是用 I D C G 进行归一化处理,表示当前 D C G I D C G 还差多大的距离。公式如下:

N D C G p = D C G p I D C G p

这样每个查询语句的 N D C G p 就是从0到1,不同查询语句之间就可以做比较,就可以求多个查询语句的平均 N D C G p
N D C G @ 10 N D C G @ 20 分别表示求 p 为10和20的时候的 N D C G

三、ERR

Mean reciprocal rank (MRR) :

reciprocal rank是指,第一个正确答案的排名的倒数。MRR是指多个查询语句的排名倒数的均值。公式如下:

M R R = 1 | Q | i = 1 | Q | 1 r a n k k

其中 r a n k k 表示第 i 个查询语句的第一个正确答案的排名。

Cascade Models:

之前的评分模型虽然考虑了位置自身的价值信息和位置上文档的相关度信息,但是没有考虑文档之间的相关性信息。一种考虑是,一个文档是否被用户点击和排在它前面的文档有很大的关系,比如排在前面的文档都是不相关文档,那么它被点击的概率就高,如果排它前面的文档都是非常相关的文档,那么它被点击的概率就很低。Cascade Models假设用户从排名由高到底依次查看文档,一旦文档满足了用户的需求,则停止查看后续的文档。用 R i 表示用户只看在位置 i 上的文档后就不在需要查看其它文档的概率,显然文档的相关度越高, R i 越大。那么用户在位置 i 停止的概率公式如下:

P P r = i = 1 r 1 ( 1 R i ) R r

Expected reciprocal rank (ERR) :

区别 R R 是计算第一个相关文档的位置的倒数, E R R 表示用户的需求被满足时停止的位置的倒数的期望。首先是计算用户在位置 r 停止的概率 P P r ,如下所示:

P P r = i = 1 r 1 ( 1 R i ) R r

其中 R i 是关于文档相关度等级的函数,可以选取如下的函数:

R i = R ( g i ) = 2 g 1 2 g m a x , g { 0 , 1 , , g m a x }

那么 E R R 的计算公式如下:

E R R = r = 1 n φ ( r ) P P r = r = 1 n 1 r P P r = r = 1 n 1 r i = 1 r 1 ( 1 R i ) R r

更通用一点, E R R 不一定计算用户需求满足时停止的位置的倒数的期望,可以是其它基于位置的函数 φ ( r ) ,只要满足 φ ( 0 ) = 1 ,且 φ ( r ) 0 随着 r 。比如 D C G 中的 φ ( r ) = 1 l o g 2 ( r + 1 )

四、F-score/F-measure

F-score/F-measure :

一种同时考虑准确率和召回率的指标。公式如下:

F = 2 × p r e c i s i o n × r e c a l l ( p r e c i s i o n + r e c a l l )

可以看出F的取值范围从0到1。另外还有一种F的变体如下所示:

F β = ( 1 + β 2 ) × ( p r e c i s i o n × r e c a l l ) ( β 2 × p r e c i s i o n + r e c a l l )

常用的两种设置是 F 2 F 0.5 ,前者中 r e c a l l 重要程度是 p r e c i s i o n 的两倍,后者则相反, p r e c i s i o n 重要程度是 r e c a l l 的两倍。

转自:https://blog.csdn.net/u010138758/article/details/69936041

相关文章
相关标签/搜索