Salvato in:
Dettagli Bibliografici
Autori principali: Zhou, Guancheng, Xu, Haiping, Xu, Hongkang, Li, Chenyu, Yan, Donghui
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:https://arxiv.org/abs/2505.22984
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917255881687040
author Zhou, Guancheng
Xu, Haiping
Xu, Hongkang
Li, Chenyu
Yan, Donghui
author_facet Zhou, Guancheng
Xu, Haiping
Xu, Hongkang
Li, Chenyu
Yan, Donghui
contents The popular K-means clustering algorithm potentially suffers from a major weakness for further analysis or interpretation. Some cluster may have disproportionately more (or fewer) points from one of the subpopulations in terms of some sensitive variable, e.g., gender or race. Such a fairness issue may cause bias and unexpected social consequences. This work attempts to improve the fairness of K-means clustering with a two-stage optimization formulation--clustering first and then adjust cluster membership of a small subset of selected data points. Two computationally efficient algorithms are proposed in identifying those data points that are expensive for fairness, with one focusing on nearest data points outside of a cluster and the other on highly 'mixed' data points. Experiments on benchmark datasets show substantial improvement on fairness with a minimal impact to clustering quality. The proposed algorithms can be easily extended to a broad class of clustering algorithms or fairness metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2505_22984
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A Computational Approach to Improving Fairness in K-means Clustering
Zhou, Guancheng
Xu, Haiping
Xu, Hongkang
Li, Chenyu
Yan, Donghui
Machine Learning
Computers and Society
Methodology
The popular K-means clustering algorithm potentially suffers from a major weakness for further analysis or interpretation. Some cluster may have disproportionately more (or fewer) points from one of the subpopulations in terms of some sensitive variable, e.g., gender or race. Such a fairness issue may cause bias and unexpected social consequences. This work attempts to improve the fairness of K-means clustering with a two-stage optimization formulation--clustering first and then adjust cluster membership of a small subset of selected data points. Two computationally efficient algorithms are proposed in identifying those data points that are expensive for fairness, with one focusing on nearest data points outside of a cluster and the other on highly 'mixed' data points. Experiments on benchmark datasets show substantial improvement on fairness with a minimal impact to clustering quality. The proposed algorithms can be easily extended to a broad class of clustering algorithms or fairness metrics.
title A Computational Approach to Improving Fairness in K-means Clustering
topic Machine Learning
Computers and Society
Methodology
url https://arxiv.org/abs/2505.22984