文章

常见的论文中模型评价指标

常见的论文中模型评价指标

在机器学习与深度学习的研究中,如何科学地评估模型性能是一个核心问题。本文整理了论文中常见的几类评价指标:混淆矩阵相关指标(Accuracy、Precision、Recall、Fβ-Score 等)、ROC 曲线与 AUC、以及推荐系统中广泛使用的 HR@K、MAP、NDCG、MRR 等。

混淆矩阵

混淆矩阵是分类模型评估中最基础的工具,它以 $n \times n$ 矩阵的形式直观地呈现模型预测结果与真实标签之间的关系。在二分类问题中,混淆矩阵的结构如下:

混淆矩阵表示图

基于混淆矩阵的四个基本量(TP、TN、FP、FN),可以派生出多个评价指标:

准确率(Accuracy)

准确率衡量预测正确的样本占全部样本的比例,适用于类别分布均衡的场景。

\[Acc=\dfrac{TP+TN}{TP+TN+FP+FN}\]

精确率(Precision)

精确率表示在所有被模型预测为正类的样本中,真实为正类的比例。它关注的是预测的可靠性。

\[Precision=\dfrac{TP}{TP+FP}\]

召回率(Recall)

召回率表示在所有真实为正类的样本中,被模型正确预测出来的比例。它关注的是对正类的覆盖能力。

\[Recall=\dfrac{TP}{TP+FN}\]

Fβ-Score

Precision 和 Recall 之间存在天然的权衡关系。Fβ-Score 通过参数 $\beta$ 调节两者的相对重要性:$\beta < 1$ 偏重 Precision,$\beta > 1$ 偏重 Recall。通常取 $\beta = 1$,称为 F1-Score,即两者的调和平均。

\[F_{\beta}=(1+\beta^{2})\cdot\dfrac{Precision\cdot{Recall}}{(\beta^{2}\cdot{Precision})+Recall}\]

ROC 曲线与 AUC

ROC 曲线以假正例率(FPR)为横轴、真正例率(TPR)为纵轴,通过改变分类阈值绘制出一系列 $(FPR, TPR)$ 点,从而展现模型在不同阈值下的整体表现。AUC 即 ROC 曲线下的面积:

ROC 曲线与 AUC 示意图

AUC 的取值范围及其含义如下:

AUC 范围含义
$AUC=1$完美分类器,实际应用中极为罕见
$0.5<AUC<1$优于随机猜测,具有预测价值
$AUC=0.5$与随机猜测无异
$0<AUC<0.5$劣于随机猜测(反向使用可优于随机)

Hit Ratio(HR)

HR 是推荐系统中常用的召回指标,衡量在 Top-K 推荐列表中命中测试集物品的比例:

\[HR@K=\dfrac{Num@K}{GT}\]

其中 $Num@K$ 为所有用户的 Top-K 推荐列表中命中 Ground-Truth 物品的总数,$GT$ 为测试集中所有 Ground-Truth 物品的总数。

Mean Average Precision(MAP)

AP(Average Precision)用于评估排序质量。以搜索引擎为例:给定一个查询,系统返回 10 个结果,若只有 5 个相关,那么这 5 个结果排得越靠前,AP 越高;反之,若相关结果排在列表末尾,AP 就会很低。AP 本质上是一种对顺序敏感的召回率。

对于用户 $u$,其平均准确率为:

\[AP_{u}=\dfrac{1}{\Omega_{u}}\sum\limits_{i\in{\Omega_{u}}}\dfrac{\sum\limits_{j\in{\Omega_{u}}}h(p_{uj}<p_{ui})+1}{p_{ui}}\]

其中 $\Omega_{u}$ 表示用户 $u$ 的 Ground-Truth 集合,$p_{ui}$ 表示物品 $i$ 在推荐列表中的位置,$p_{uj}<p_{ui}$ 表示物品 $j$ 排在物品 $i$ 之前。

MAP 是所有用户 AP 的算术平均:

\[MAP=\dfrac{\sum\limits_{u\in{U}}AP_{u}}{|U|}\]

Normalized Discounted Cumulative Gain(NDCG)

NDCG 是排序质量评估中最常用的指标之一,它依次建立在 CG、DCG 的概念之上。

累积增益(CG)

CG 直接将推荐列表中每个位置的相关性分数累加:

\[CG_{k}=\sum\limits_{i=1}^{k}rel_{i}\]

其中 $rel_i$ 表示位置 $i$ 推荐结果的相关性,$k$ 为推荐列表长度。CG 的缺点在于不考虑位置因素——无论相关结果排在前面还是后面,累加值相同。

折损累积增益(DCG)

DCG 在 CG 的基础上引入位置折扣,位置越靠后,对总分的贡献越小:

\[DCG_{k}=\sum\limits_{i=1}^{k}\dfrac{2^{rel_{i}}-1}{\log_{2}(i+1)}\]

下表展示了折扣因子 $1/\log_2(i+1)$ 随位置的变化,可以直观地看到位置越靠后衰减越快:

NDCG 位置折扣因子示意图

归一化折损累积增益(NDCG)

不同用户的推荐列表长度和相关性分布不同,直接比较 DCG 没有意义。NDCG 通过除以理想情况下的 DCG(即 IDCG)进行归一化:

\[NDCG_{u}@K=\dfrac{DCG_{u}@K}{IDCG_{u}}\]

其中 IDCG 为将 Ground-Truth 按相关性降序排列后计算的 DCG。NDCG@K 的取值范围为 $(0, 1]$,值越大表示排序越接近理想排序。

对所有用户取平均得到整体的 NDCG@K:

\[NDCG@K=\dfrac{\sum\limits_{u\in{U}}NDCG_{u}@K}{|U|}\]

Mean Reciprocal Rank(MRR)

MRR 关注第一个正确结果在推荐列表中的排名,适用于只需要找到一个正确答案的场景(如问答系统、知识图谱链接预测):

\[MRR=\dfrac{1}{|Q|}\sum\limits_{i=1}^{|Q|}\dfrac{1}{rank_{i}}\]
其中 $Q$ 为查询(用户)的数量,$rank_i$ 为第 $i$ 个用户的推荐列表中第一个命中 Ground-Truth 的物品所在的位置。

指标对比总结

指标关注点是否顺序敏感典型场景
Accuracy分类正确比例—均衡分类任务
Precision预测正类的可靠性—搜索结果去噪
Recall正类的覆盖程度—医疗筛查、信息召回
F1-ScorePrecision 与 Recall 的调和—类别不平衡
AUC阈值无关的排序能力✓二分类整体评估
HR@KTop-K 命中率部分敏感推荐召回
MAP相关结果的整体排序✓推荐排序
NDCG@K考虑位置折损的排序✓推荐排序、检索排序
MRR首个正确结果的排名✓问答、知识图谱补全
本文由作者按照 CC BY 4.0 进行授权