《sklearn机器学习——聚类性能指标》Fowlkes-Mallows 得分

Fowlkes-Mallows (FM) 得分是一种用于评估聚类算法性能的外部指标,它衡量的是聚类结果与某个参考模型(通常是真实标签)之间的一致性。该指标特别适用于需要将聚类结果与已知类别进行比较的场景。

Fowlkes-Mallows 得分简介

1. 核心思想

Fowlkes-Mallows 得分的核心思想是衡量两个数据点集合(在聚类中通常指聚类结果和真实标签)之间的一致性。它通过计算在两个集合中都被分到同一类的数据点对(pair of points)的比例来实现。

更具体地说,它关注的是:

  • 真阳性 (True Positive, TP): 在真实标签中属于同一类,且在聚类结果中也被分到同一类的数据点对的数量。
  • 假阳性 (False Positive, FP): 在真实标签中不属于同一类,但在聚类结果中被错误地分到同一类的数据点对的数量。
  • 假阴性 (False Negative, FN): 在真实标签中属于同一类,但在聚类结果中被错误地分到不同类的数据点对的数量。

2. 计算方法

FM 得分的计算依赖于一个2x2的列联表,该表统计了在参考标签和聚类结果中,样本对的分配情况:

聚类结果相同 聚类结果不同
真实标签相同 TP FN
真实标签不同 FP TN

其中:

  • TP (True Positive): 在真实标签中属于同一类,并且在聚类结果中也被分到同一类的样本对数量。
  • FP (False Positive): 在真实标签中属于不同类,但在聚类结果中被错误地分到同一类的样本对数量。
  • FN (False Negative): 在真实标签中属于同一类,但在聚类结果中被错误地分到不同类的样本对数量。
  • TN (True Negative): 在真实标签中属于不同类,并且在聚类结果中也被分到不同类的样本对数量。

Fowlkes-Mallows 得分的计算公式为:

\\text{FM} = \\sqrt{\\text{Precision} \\times \\text{Recall}}

其中:

  • Precision (精确率) = TP / (TP + FP):衡量聚类结果中,被分到同一簇的样本对在真实标签中也属于同一类的比例。
  • Recall (召回率) = TP / (TP + FN):衡量在真实标签中属于同一类的样本对,在聚类结果中也被正确分到同一簇的比例。

因此,FM 得分是精确率和召回率的几何平均数。

3. 特点

  1. 取值范围 : FM 得分的取值范围在 [0, 1] 之间。

    • 1: 表示聚类结果与参考标签完全一致,是完美的聚类。
    • 0: 表示聚类结果与参考标签完全不一致。
    • 值越接近 1,说明聚类性能越好。
  2. 对簇数量不敏感: 与一些其他指标不同,FM 得分对聚类算法产生的簇的数量没有直接偏好。

  3. 基于成对比较: 它关注的是样本对的关系,而不是单个样本的分配,这使得它对标签的排列不敏感(即簇的标签名称不重要,重要的是样本的分组关系)。

  4. 外部指标: 它需要真实的类别标签作为参考,因此只能在有监督或半监督的评估场景中使用。

4. 与其他指标的关系

  • 与 Rand Index (RI) 的关系: FM 得分和 RI 都基于成对样本的比较。但 RI 是精确率和召回率的算术平均数(RI = (TP + TN) / Total Pairs),而 FM 是它们的几何平均数。FM 得分通常被认为对不平衡的聚类结果更稳健。
  • 与几何平均数: 使用几何平均数(而非算术平均数)可以更好地平衡精确率和召回率。如果其中一个值很低,即使另一个值很高,FM 得分也会被显著拉低,从而避免了只优化单一指标的倾向。

5.简单示例

python 复制代码
from sklearn.metrics import fowlkes_mallows_score
import numpy as np

# 真实标签 (ground truth labels)
true_labels = [0, 0, 1, 1, 2, 2]

# 聚类结果 (clustering result)
cluster_labels = [0, 0, 1, 2, 1, 2]

# 计算 Fowlkes-Mallows 得分
fm_score = fowlkes_mallows_score(true_labels, cluster_labels)

print(f"Fowlkes-Mallows Score: {fm_score:.4f}")

输出结果

python 复制代码
Fowlkes-Mallows Score: 0.4714

5. 总结

Fowlkes-Mallows 得分是一个直观且有效的聚类评估指标,尤其适用于需要与真实标签进行比较的场景。它通过计算精确率和召回率的几何平均数,综合反映了聚类结果在"不合并不同类"(高精确率)和"不拆分同类"(高召回率)两方面的能力。

相关推荐
谭欣辰7 小时前
C++快速幂完整实战讲解
算法·决策树·机器学习
AI周红伟7 小时前
周红伟:GPT-Image-2深度解析:从技术原理到实战教程,为什么它能让整个AI圈炸锅?
人工智能·gpt·深度学习·机器学习·语言模型·openclaw
Uopiasd1234oo8 小时前
上下文引导模块改进YOLOv26局部与全局特征融合能力双重提升
深度学习·yolo·机器学习
哥布林学者9 小时前
深度学习进阶(十四)ConvNeXt
机器学习·ai
ytttr87311 小时前
MATLAB SIFT图像配准实现
算法·机器学习·matlab
小饕11 小时前
从 Word2Vec 到多模态:词嵌入技术的演进全景
人工智能·算法·机器学习
吻等离子11 小时前
机器学习基本概念篇(含思维导图)
人工智能·机器学习
lwf00616411 小时前
FFM (Field-aware Factorization Machine) 学习日记
算法·机器学习
用AI赚一点12 小时前
AI落地不是造大模型:从概念到落地的核心差异
人工智能·深度学习·机器学习
新新学长搞科研12 小时前
【高质量能源会议推荐】第十一届能源与环境研究进展国际学术会议(ICAEER 2026)
人工智能·物联网·算法·机器学习·能源·环境·新能源