K-means聚类算法的原理、应用与实例

极致人生-010

于 2024-04-10 11:02:52 发布

阅读量2.4k

点赞数 32

文章标签：算法 kmeans 聚类

本文链接：https://blog.csdn.net/qqrrjj2011/article/details/137588399

版权

文章目录

- 一个使用 K-means 聚类算法进行客户细分的简单实例

K-means 聚类算法：原理

K-means 算法是一种经典的无监督学习方法，用于对未标记的数据集进行分群，即将数据集中相似的对象划分为不同的簇。以下是其基本原理：

1. 初始化：

设定簇的数量（K）：由用户预先指定，表示希望得到的簇的数量。
选择初始聚类中心（Centroids）：通常随机从数据集中选取 K 个对象作为初始的聚类中心。

2. 分配对象到簇：

计算距离：对于数据集中每一个对象，计算其与 K 个聚类中心之间的距离（通常使用欧氏距离）。
分配归属：将每个对象分配到与其最近的聚类中心对应的簇中。

3. 更新聚类中心：

计算簇内平均值：对于每个簇，计算其包含的所有对象的特征均值，得到新的聚类中心。
移动中心：将簇的聚类中心更新为这个新的计算出的均值位置。

4. 判断收敛与迭代：

检查终止条件：比较当前迭代前后聚类中心的变化情况，如果变化小于某个预定阈值或达到最大迭代次数，则算法结束；否则，返回步骤2，继续进行新一轮的分配和更新。

上述过程反复进行，直到聚类中心的位置不再显著变化或达到预设的迭代次数上限。最终得到的簇即为数据集中的自然结构划分，每个簇内的对象在特征空间中较为接近，而不同簇之间的对象相对较远。

K-means 聚类算法的应用

K-means 聚类因其简单、高效的特点，在众多领域中有着广泛应用，包括但不限于：

1. 数据挖掘与分析：

市场细分：对消费者数据进行聚类，识别具有相似消费习惯或偏好特征的客户群体，以便制定针对性的营销策略。
文档分类：对文本数据（如新闻文章、网页等）进行聚类，自动划分主题相似的文章类别。
社交网络分析：对用户行为数据进行聚类，发现用户社区、兴趣小组或用户角色。

2. 图像处理与计算机视觉：

图像分割：对图像像素进行聚类，实现基于颜色、纹理等特征的图像区域划分。
对象识别与跟踪：对视频帧中的对象进行聚类，辅助进行运动对象的识别与跟踪。

3. 生物医学研究：

基因表达数据分析：对基因表达谱数据进行聚类，识别具有相似表达模式的基因组群，揭示潜在的生物学功能或疾病相关性。
医疗影像分析：对医学影像（如MRI、CT等）进行聚类，区分正常组织与异常区域，辅助诊断与治疗规划。

4. 其他领域：

地理信息系统（GIS）：对地理位置数据进行聚类，发现人口分布、商业热点等空间模式。
物联网（IoT）：对传感器数据进行聚类，识别设备工作状态模式或异常行为。

K-means 聚类算法的优化与改进

尽管 K-means 算法简单易用，但在实际应用中可能会遇到一些挑战，为此研究人员提出了多种优化与改进策略：

1. 初始聚类中心的选择：

K-means++：通过概率方法选择初始聚类中心，确保它们尽可能分散且能代表数据的整体分布，从而提高算法的稳定性和收敛速度。
其他策略：如基于密度的方法、基于层次的方法或使用智能优化算法（如遗传算法、模拟退火等）来确定初始聚类中心。

2. 距离度量与标准化：

非欧氏距离：根据数据特性选择更适合的距离度量，如曼哈顿距离、余弦相似度、马氏距离等。
特征缩放与标准化：对数据进行预处理，如归一化、标准化等，以消除特征间尺度差异对聚类结果的影响。

3. 处理不同类型数据与噪声：

模糊 C 均值（FCM）：允许对象属于多个簇，适用于边界模糊或含有噪声的数据。
DBSCAN 或 OPTICS：针对具有不同密度区域的数据，发现任意形状的簇，并能较好地处理噪声点和离群值。

4. 动态调整簇数量 K：

肘部法则：通过观察轮廓系数、 inertia（簇内平方和）等指标随 K 值变化的趋势，选择“肘部”处的 K 值作为最优簇数。
交叉验证或**贝叶斯信息准则（BIC）**等统计方法：用于评估不同 K 值下的聚类质量，选择最优 K。

5. 并行与分布式计算：

MapReduce 或 Spark 等框架：对大规模数据集进行分布式 K-means 聚类，利用多核处理器或集群的并行计算能力加速算法执行。

6. 异质聚类：

混合高斯模型（GMM）：将数据视为由多个高斯分布生成，每个高斯分布对应一个簇，适用于数据内部存在异质性的场景。GMM 通过 EM 算法进行参数估计和聚类。
概率潜在语义分析（PLSA）：适用于处理文本数据，假设每个文档是若干隐含主题的混合，每个主题对应一个簇，通过最大化似然函数进行参数估计和聚类。

7. 高维数据聚类：

子空间聚类（如 CLIQUE、SPEC、PROCLUS 等）：寻找数据中具有聚类结构的低维子空间，降低维度以改善 K-means 在高维空间中的性能。
稀疏编码或深度学习预处理：通过学习数据的潜在表示（如自编码器、深度神经网络等），将原始高维数据映射到低维、更利于聚类的特征空间。

8. 时间序列与流数据聚类：

在线 K-means 或 增量 K-means：适应数据流的实时更新，仅对新加入的数据点或发生变化的簇进行重新分配和中心更新，无需每次都遍历整个数据集。
动态聚类（如 DenStream、CluStream 等）：适用于数据分布随时间变化的场景，能够持续监控数据流，发现并跟踪动态出现和消失的簇。

9. 加权 K-means 聚类：

加权 K-means：为数据点赋予权重，反映其在聚类中的相对重要性，适用于处理带有不确定性的数据或含有噪声的数据集。
约束 K-means：引入先验知识或用户指定的约束条件（如必须将某些对象分到同一簇、某些对象不能分到同一簇等），引导聚类过程，提高结果的实用价值。

10. 聚类后处理与评估：

后处理方法：如对小簇合并、大簇分裂、边界对象重新分配等操作，以改善聚类的直观解释性和用户接受度。
聚类评估指标：如轮廓系数、Calinski-Harabasz 指数、Davies-Bouldin 指数等，定量评价聚类结果的质量，为算法选择和参数调优提供依据。

综上所述，通过对 K-means 聚类算法进行适当的优化与改进，我们可以应对更广泛的数据类型、规模、特性和应用场景，提高聚类的准确性和效率，使其在实际问题中发挥更大的作用。同时，结合领域知识和具体需求，灵活运用各种策略和方法，有助于获得更为满意的聚类结果。

一个使用 K-means 聚类算法进行客户细分的简单实例

以下是一个使用 K-means 聚类算法进行客户细分的简单实例。在这个例子中，我们假设有一家电商公司收集了其部分客户的购买历史数据，包括两个主要特征：年度消费金额（Annual_Spending）和购物频次（Purchase_Frequency）。公司希望通过 K-means 聚类算法将客户分为不同的群体，以便制定更具针对性的营销策略。

数据准备：

假设我们有一个包含 n 个客户的样本数据集 dataset，其中每个客户记录由两列组成：

dataset = [
    [Annual_Spending_1, Purchase_Frequency_1],
    [Annual_Spending_2, Purchase_Frequency_2],
    ...
    [Annual_Spending_n, Purchase_Frequency_n]
]

实施 K-means 聚类：

初始化聚类中心：选择 k 个初始聚类中心，这里假设 k=3，可以随机从数据集中抽取 k 个样本作为初始聚类中心。

import numpy as np

k = 3
initial_centers = np.random.choice(dataset, k, replace=False)

迭代过程：

分配样本到最近的聚类中心：计算每个样本到各个聚类中心的距离（通常使用欧氏距离），将其分配到距离最近的聚类。

def euclidean_distance(sample, center):
    return np.sqrt(np.sum((sample - center)**2))

clusters = [[] for _ in range(k)]
for sample in dataset:
    distances = [euclidean_distance(sample, center) for center in initial_centers]
    closest_cluster_index = np.argmin(distances)
    clusters[closest_cluster_index].append(sample)

更新聚类中心：重新计算每个聚类内所有样本的均值，作为新的聚类中心。

new_centers = []
for cluster in clusters:
    if cluster:
        mean = np.mean(cluster, axis=0)
        new_centers.append(mean)
    else:
        # 如果某个聚类为空，可以重新随机选择一个样本作为中心，或者使用前一轮的中心
        new_centers.append(initial_centers[np.random.randint(k)])

收敛判断：比较新旧聚类中心的变化，如果变化小于某个设定阈值或达到最大迭代次数，则停止迭代；否则，用新聚类中心替换旧中心，继续下一轮迭代。

convergence_threshold = 0.01
max_iterations = 100
iteration = 0

while iteration < max_iterations:
    old_centers = initial_centers.copy()
    initial_centers = new_centers
    # ... 重复步骤 2 中的分配样本和更新聚类中心操作
    # ... 计算新旧聚类中心之间的差异，并判断是否满足收敛条件
    # 如果未达到收敛条件，增加迭代计数器并继续下一轮
    iteration += 1

# 当算法收敛或达到最大迭代次数时，输出最终的聚类结果

结果解释与应用：

经过 K-means 聚类后，我们得到了三个客户群体（假设 k=3）。每个群体代表一类具有相似消费行为的客户，可以根据聚类结果分析：

低消费、低频次群体：可能对应价格敏感型客户，可推送优惠券或特价商品以刺激消费。
中等消费、中频次群体：可能是忠诚但并不频繁购物的客户，可以通过定期促销活动保持其活跃度。
高消费、高频次群体：可能为VIP客户，应提供专属客户服务、优先权益等，以维持其高价值贡献。

以上实例展示了如何使用 K-means 聚类算法对客户数据进行细分。实际应用中，可能还需要对数据进行预处理（如标准化或归一化）、选择合适的 k 值（可以通过肘部法则、轮廓系数等方法确定）、以及对聚类结果进行可视化（如使用散点图展示不同群体在特征空间中的分布）等步骤。

上述实例展示了 K-means 聚类算法的基本应用流程。接下来，我们将对实例进行补充和完善，包括数据预处理、确定 K 值、结果可视化以及对聚类结果的解释与应用。

1. 数据预处理：

在实际应用中，数据往往需要进行预处理以提高聚类效果。在这个例子中，由于年度消费金额和购物频次的量纲和数值范围可能存在较大差异，我们对其进行标准化处理：

from sklearn.preprocessing import StandardScaler

scaler = StandardScaler()
scaled_dataset = scaler.fit_transform(dataset)

标准化后的数据将具有零均值和单位方差，有助于消除特征间的尺度差异，使得距离计算更加公平。

2. 确定 K 值：

K-means 聚类算法需要预先指定簇的数量 k。在这里，我们使用肘部法则来帮助确定一个合适的 k 值。肘部法则通过观察随着 k 增大，轮廓系数或簇内平方和（inertia）的变化趋势，选择拐点（即“肘部”）处的 k 值。

from sklearn.cluster import KMeans
from sklearn.metrics import silhouette_score

inertias = []
silhouette_scores = []

for k in range(2, 11):  # 测试 2 到 10 个簇
    kmeans = KMeans(n_clusters=k).fit(scaled_dataset)
    inertias.append(kmeans.inertia_)
    silhouette_scores.append(silhouette_score(scaled_dataset, kmeans.labels_))

plt.figure(figsize=(10, 6))
plt.subplot(1, 2, 1)
plt.plot(range(2, 11), inertias, marker='o')
plt.title('Elbow Method: Inertia vs. Number of Clusters')
plt.xlabel('Number of Clusters (k)')
plt.ylabel('Inertia')

plt.subplot(1, 2, 2)
plt.plot(range(2, 11), silhouette_scores, marker='o')
plt.title('Silhouette Score vs. Number of Clusters')
plt.xlabel('Number of Clusters (k)')
plt.ylabel('Silhouette Score')

plt.tight_layout()
plt.show()

# 根据图形判断“肘部”位置，选择合适的 k 值

3. 结果可视化：

使用散点图将聚类结果可视化，可以帮助我们直观理解不同客户群体在年度消费金额和购物频次特征空间中的分布。

def plot_clusters(dataset, labels, centers):
    plt.figure(figsize=(8, 6))
    plt.scatter(dataset[:, 0], dataset[:, 1], c=labels, cmap='viridis', alpha=0.8)
    plt.scatter(centers[:, 0], centers[:, 1], marker='X', s=150, color='red', label='Cluster Centers')
    plt.xlabel('Annual Spending (Standardized)')
    plt.ylabel('Purchase Frequency (Standardized)')
    plt.legend()
    plt.show()

k = 3  # 假设通过肘部法则确定 k=3
kmeans = KMeans(n_clusters=k).fit(scaled_dataset)
labels = kmeans.labels_
centers = kmeans.cluster_centers_

plot_clusters(scaled_dataset, labels, centers)