arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.10085cs.LGcs.AIcs.CGstat.ML
收藏

社会公平的k-Means聚类

Socially Fair k-Means Clustering

Georgia Tech(佐治亚理工学院);MPI for Intelligent Systems(马克斯·普朗克智能系统研究所)

Mehrdad Ghadiri, Samira Samadi, Santosh Vempala

摘要:

我们表明,流行的k均值聚类算法(劳埃德启发式),用于各种科学数据,可能会导致结果是不利的数据分组(例如,人口群体)。这种有偏见的聚类可能对以人为中心的应用程序(如资源分配)产生有害影响。我们提出了一个公平的k-means目标和算法来选择聚类中心,为不同的群体提供公平的成本。Fair-Lloyd算法是对Lloyd的k-means启发式算法的改进,继承了其简单性、高效性和稳定性。与标准劳埃德相比,我们发现,在基准数据集上,Fair-Lloyd通过确保所有组在输出k-聚类中具有相等的成本来表现出无偏的性能,同时在运行时间上产生可忽略不计的增加,从而使其成为当前使用k-均值的可行的公平选项。

英文摘要:

We show that the popular k-means clustering algorithm (Lloyd's heuristic), used for a variety of scientific data, can result in outcomes that are unfavorable to subgroups of data (e.g., demographic groups). Such biased clusterings can have deleterious implications for human-centric applications such as resource allocation. We present a fair k-means objective and algorithm to choose cluster centers that provide equitable costs for different groups. The algorithm, Fair-Lloyd, is a modification of Lloyd's heuristic for k-means, inheriting its simplicity, efficiency, and stability. In comparison with standard Lloyd's, we find that on benchmark datasets, Fair-Lloyd exhibits unbiased performance by ensuring that all groups have equal costs in the output k-clustering, while incurring a negligible increase in running time, thus making it a viable fair option wherever k-means is currently used.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑