Coresets for clustering with fairness constraints

Lingxiao Huang, Shaofeng H.C. Jiang, Nisheeth K. Vishnoi

Research output: Contribution to journalConference articleScientificpeer-review

3 Citations (Scopus)

Abstract

In a recent work, [20] studied the following “fair” variants of classical clustering problems such as k-means and k-median: given a set of n data points in Rd and a binary type associated to each data point, the goal is to cluster the points while ensuring that the proportion of each type in each cluster is roughly the same as its underlying proportion. Subsequent work has focused on either extending this setting to when each data point has multiple, non-disjoint sensitive types such as race and gender [7], or to address the problem that the clustering algorithms in the above work do not scale well [42, 8, 6]. The main contribution of this paper is an approach to clustering with fairness constraints that involve multiple, non-disjoint types, that is also scalable. Our approach is based on novel constructions of coresets: for the k-median objective, we construct an e-coreset of size O(Gk2e-d) where G is the number of distinct collections of groups that a point may belong to, and for the k-means objective, we show how to construct an e-coreset of size O(Gk3e-d-1). The former result is the first known coreset construction for the fair clustering problem with the k-median objective, and the latter result removes the dependence on the size of the full dataset as in [42] and generalizes it to multiple, non-disjoint types. Plugging our coresets into existing algorithms for fair clustering such as [6] results in the fastest algorithms for several cases. Empirically, we assess our approach over the Adult, Bank, Diabetes and Athlete dataset, and show that the coreset sizes are much smaller than the full dataset; applying coresets indeed accelerates the running time of computing the fair clustering objective while ensuring that the resulting objective difference is small. We also achieve a speed-up to recent fair clustering algorithms [6, 7] by incorporating our coreset construction.

Original languageEnglish
JournalADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS
Volume32
Publication statusPublished - 2019
MoE publication typeA4 Article in a conference publication
EventConference on Neural Information Processing Systems - Vancouver, Canada
Duration: 8 Dec 201914 Dec 2019
Conference number: 33

Fingerprint Dive into the research topics of 'Coresets for clustering with fairness constraints'. Together they form a unique fingerprint.

Cite this