PR 曲线
Parent: ai_keywords
PR 曲线(Precision-Recall Curve)
核心定义
PR 曲线(Precision-Recall Curve)是机器学习分类任务中评价模型性能的可视化工具。它以**精确率(Precision)**为纵轴、召回率(Recall)为横轴,刻画在不同分类阈值下模型的性能变化。特别适用于处理类别严重不平衡的问题(例如罕见病诊断),此时阳性样本数远少于阴性样本,PR 曲线比 ROC 曲线提供更敏感和真实的评估。
关键技术点
-
精确率与召回率的权衡
- 精确率 = TP / (TP + FP):预测为阳性的样本中真正阳性的比例,衡量模型“不出错”的能力。
- 召回率 = TP / (TP + FN):真实阳性样本中被正确识别出的比例,衡量模型“不漏诊”的能力。
PR 曲线越向右上方隆起,表示同时维持高精确率与高召回率。
-
平均精确率(Average Precision, AP)
PR 曲线下的面积(AUC-PR)即 AP 值,是一种数值化概括。AP 越高,模型对罕见阳性类的识别稳定性越好,常被用于目标检测(如 mAP)及医学诊断阈值优化。 -
与 ROC 曲线的区别
- ROC 曲线基于 TPR 和 FPR,受阴性样本数影响大;在不平衡数据中易虚假乐观(高 AUC-ROC 掩盖低精确率)。
- PR 曲线直接关注阳性类,更能揭示模型在少量正例下的实际表现,是临床辅助诊断系统的首选评估指标。
-
阈值选择策略
基于 PR 曲线可以确定操作点(Operating Point),根据临床需求选择最优阈值:例如在癫痫发作预警中,宁可牺牲部分精确率(允许少许误报)也要保障高召回率(不漏掉发作),此时选择召回率≥95%对应的最低阈值。
医学/神经科学应用场景
基于脑电图的癫痫发作检测(联合首都医科大学神经病学研究所背景)
- 问题:癫痫患者的长程脑电图(EEG)数据中,发作期(阳性)仅占 <5% 的记录时间,呈现极度不平衡。
- 方法:以卷积神经网络或 Transformer 模型对 EEG 片段分类,输出发作概率。调整分类阈值,绘制PR曲线并计算 AP。
- 临床价值:通过PR曲线评估模型在极低阳性率下的真实辨别力。研究发现,优质模型在召回率 >90% 时仍能保持精确率 >80%,而劣质模型精确率会骤降至30%以下。据此优化警觉系统,使得 ICU 医护人员可在误报率可接受范围内(如每24小时 2-3 次)捕获绝大多数亚临床发作,显著降低癫痫持续状态漏诊风险。