提出排序向量聚类新方法,高效解决偏好数据分组难题
Ranking Vectors Clustering: Theory and Applications
- 将聚类中心和数据都限制为排序向量,构建KRC模型
- 单簇情况下可线性时间求出最优中心,算法整体快速有效
- 适合个性化推荐、大规模决策等需要排序聚类的场景
我们研究排序向量聚类问题,其中每个向量代表一个由不同整数构成的有序偏好列表。重点关注k-中心点排序向量聚类(KRC)问题,旨在将一组排序向量划分为k个簇并确定每个簇的中心。与经典k均值聚类(KMC)不同,KRC要求观测值和中心均为排序向量。我们证明了KRC的NP-hard性并刻画其可行集。在单簇情况下,推导出最优中心的闭式解析解,可在线性时间内计算。为应对计算挑战,提出高效近似算法KRCA,通过迭代优化基于KMC的初始解。此外,引入分支定界(BnB)算法,在KRCA中实现高效簇重构,利用决策树框架降低计算时间,并通过控制参数平衡解的质量与效率。建立了KRCA和BnB的理论误差界。在合成和真实数据集上的大量实验表明,KRCA始终优于基线方案,显著提升解质量且计算速度快。该工作凸显了KRC在个性化和大规模决策中的实际意义,提供了可拓展的方法论与洞见。
原文摘要 · Abstract (English)
We study the problem of clustering ranking vectors, where each vector represents preferences as an ordered list of distinct integers. Specifically, we focus on the k-centroids ranking vectors clustering problem (KRC), which aims to partition a set of ranking vectors into k clusters and identify the centroid of each cluster. Unlike classical k-means clustering (KMC), KRC constrains both the observations and centroids to be ranking vectors. We establish the NP-hardness of KRC and characterize its feasible set. For the single-cluster case, we derive a closed-form analytical solution for the optimal centroid, which can be computed in linear time. To address the computational challenges of KRC, we develop an efficient approximation algorithm, KRCA, which iteratively refines initial solutions from KMC, referred to as the baseline solution. Additionally, we introduce a branch-and-bound (BnB) algorithm for efficient cluster reconstruction within KRCA, leveraging a decision tree framework to reduce computational time while incorporating a controlling parameter to balance solution quality and efficiency. We establish theoretical error bounds for KRCA and BnB. Through extensive numerical experiments on synthetic and real-world datasets, we demonstrate that KRCA consistently outperforms baseline solutions, delivering significant improvements in solution quality with fast computational times. This work highlights the practical significance of KRC for personalization and large-scale decision making, offering methodological advancements and insights that can be built upon in future studies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。