了解曲线下面积 (AUC) 如何衡量模型性能。探索其在使用 Ultralytics YOLO26 进行分类、医疗健康和金融领域中的作用。
曲线下面积 (AUC) 是一种综合性能指标,用于机器学习 (ML)中评估分类模型的判别能力。具体而言,它测量受试者工作特征 (ROC) 曲线下方的二维面积,提供一个介于0到1之间的单一标量值。AUC为1.0表示一个完美的分类器,而AUC为0.5则表明模型性能不优于随机猜测。由于它聚合了所有可能分类阈值下的性能,AUC在评估预测建模能力方面特别有效,尤其是在最佳决策边界未知或可变的情况下。
为充分理解AUC,必须理解其底层的ROC曲线。该图表绘制了在不同阈值设置下,真阳性率(召回率) 与 假阳性率 之间的关系。AUC本质上量化了模型将随机选择的阳性实例排名高于随机选择的阴性实例的概率。
AUC是处理关键决策和不平衡数据集的行业中首选的指标,其中一个类别比另一个类别显著稀有。
区分 AUC 与其他模型评估指标至关重要,以便为您的项目选择正确的工具。
以下示例演示了如何加载预训练的YOLO26分类模型并运行验证。虽然YOLO模型主要报告top-1和top-5准确率,但验证过程会生成分析基于曲线的指标所需的预测数据。
from ultralytics import YOLO
# Load a pre-trained YOLO26 classification model
model = YOLO("yolo26n-cls.pt")
# Validate the model on a standard dataset (e.g., imagenet10)
# This generates precision, recall, and accuracy metrics
results = model.val(data="imagenet10")
# Access top-1 accuracy, a key point on the ROC curve
print(f"Top-1 Accuracy: {results.top1:.4f}")
print(f"Top-5 Accuracy: {results.top5:.4f}")
为了实现全面的生命周期管理,包括数据集标注和云训练(在这些场景中,指标会自动可视化),开发人员可以利用Ultralytics Platform。这简化了无需手动计算即可解释AUC等复杂指标的过程。
开启您的机器学习未来之旅