Efficient k-means with Individual Fairness via Exponential Tilting

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhu, Shengkun, Zeng, Jinshan, Sun, Yuan, Wang, Sheng, Li, Xiaodong, Peng, Zhiyong
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909230247706624
author Zhu, Shengkun
Zeng, Jinshan
Sun, Yuan
Wang, Sheng
Li, Xiaodong
Peng, Zhiyong
author_facet Zhu, Shengkun
Zeng, Jinshan
Sun, Yuan
Wang, Sheng
Li, Xiaodong
Peng, Zhiyong
contents In location-based resource allocation scenarios, the distances between each individual and the facility are desired to be approximately equal, thereby ensuring fairness. Individually fair clustering is often employed to achieve the principle of treating all points equally, which can be applied in these scenarios. This paper proposes a novel algorithm, tilted k-means (TKM), aiming to achieve individual fairness in clustering. We integrate the exponential tilting into the sum of squared errors (SSE) to formulate a novel objective function called tilted SSE. We demonstrate that the tilted SSE can generalize to SSE and employ the coordinate descent and first-order gradient method for optimization. We propose a novel fairness metric, the variance of the distances within each cluster, which can alleviate the Matthew Effect typically caused by existing fairness metrics. Our theoretical analysis demonstrates that the well-known k-means++ incurs a multiplicative error of O(k log k), and we establish the convergence of TKM under mild conditions. In terms of fairness, we prove that the variance generated by TKM decreases with a scaled hyperparameter. In terms of efficiency, we demonstrate the time complexity is linear with the dataset size. Our experiments demonstrate that TKM outperforms state-of-the-art methods in effectiveness, fairness, and efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2406_16557
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Efficient k-means with Individual Fairness via Exponential Tilting
Zhu, Shengkun
Zeng, Jinshan
Sun, Yuan
Wang, Sheng
Li, Xiaodong
Peng, Zhiyong
Machine Learning
Computers and Society
In location-based resource allocation scenarios, the distances between each individual and the facility are desired to be approximately equal, thereby ensuring fairness. Individually fair clustering is often employed to achieve the principle of treating all points equally, which can be applied in these scenarios. This paper proposes a novel algorithm, tilted k-means (TKM), aiming to achieve individual fairness in clustering. We integrate the exponential tilting into the sum of squared errors (SSE) to formulate a novel objective function called tilted SSE. We demonstrate that the tilted SSE can generalize to SSE and employ the coordinate descent and first-order gradient method for optimization. We propose a novel fairness metric, the variance of the distances within each cluster, which can alleviate the Matthew Effect typically caused by existing fairness metrics. Our theoretical analysis demonstrates that the well-known k-means++ incurs a multiplicative error of O(k log k), and we establish the convergence of TKM under mild conditions. In terms of fairness, we prove that the variance generated by TKM decreases with a scaled hyperparameter. In terms of efficiency, we demonstrate the time complexity is linear with the dataset size. Our experiments demonstrate that TKM outperforms state-of-the-art methods in effectiveness, fairness, and efficiency.
title Efficient k-means with Individual Fairness via Exponential Tilting
topic Machine Learning
Computers and Society
url https://arxiv.org/abs/2406.16557