中文

一种可扩展的个体公平 K-means 聚类算法

数据结构与算法 2024-02-14 v1 计算机与社会 机器学习

摘要

我们提出了一种可扩展的算法,用于解决由 Jung 等人以及 Mahabadi 等人引入的个体公平 (pp, kk)-聚类问题。给定度量空间中的 nn 个点 PP,设 δ(x)\delta(x)(其中 xPx\in P)为围绕 xx 且至少包含 n/kn / k 个点的最小球的半径。如果对于每个 xPx\in P,聚类中心都在距离 xxδ(x)\delta(x) 的范围内,则该聚类被称为个体公平的。虽然已知该问题存在良好的近似算法,但尚未提出具有良好理论保证的高效实用算法。我们设计了首个快速局部搜索算法,其运行时间为 ~O(nk2)O(nk^2),并获得双标准 (O(1),6)(O(1), 6) 近似。随后,我们通过实证表明,我们的算法不仅比 prior work 快得多,而且能产生成本更低的解。

关键词

引用

@article{arxiv.2402.06730,
  title  = {A Scalable Algorithm for Individually Fair K-means Clustering},
  author = {MohammadHossein Bateni and Vincent Cohen-Addad and Alessandro Epasto and Silvio Lattanzi},
  journal= {arXiv preprint arXiv:2402.06730},
  year   = {2024}
}

备注

32 pages, 2 figures, to appear at the 27th International Conference on Artificial Intelligence and Statistics (AISTATS) 2024