A Sub-4 Approximation for Fair k-Means
cs.CG, cs.LG
Submitted: 2026-09-07
Updated: 2026-09-07
License: http://creativecommons.org/licenses/by/4.0/
The gist: Fairness in clustering has attracted sustained research interest, motivated by the need to ensure equitable representation of protected groups in machine learning applications.
Terminology
Abstract
Fairness in clustering has attracted sustained research interest, motivated by the need to ensure equitable representation of protected groups in machine learning applications. We study fair k-means clustering in Euclidean space, where the proportion of each protected group in every cluster must lie within specified lower and upper bounds. These constraints make it challenging to determine both cluster centers and point assignments. We propose an approximation algorithm that combines a linear programming relaxation with geometric transformations of the input to construct candidate center sets. Given a ρ-approximate algorithm for weighted k-means and any ε>0, our algorithm returns a fractional solution whose cost is at most 1+(3-1/Γ)ρ+O(ε) times the optimal integral fair cost, where Γ about6.357 is an upper bound on the integrality gap of the standard Euclidean k-means LP. With a PTAS as the subroutine, the approximation ratio becomes 3.8427+O(ε), improving the previous factor of 5+O(ε) to below 4. The solution satisfies all fairness constraints exactly and can be rounded to an integral assignment with a bounded additive violation of fairness and no increase in cost. The same approximation guarantee extends to the k-sparse Wasserstein barycenter problem.
Related papers
- A Unified Dual Method for Matching Problems
- It's the Geometry, Not the Model: Effective Rank and Subspace Alignment in Functional Connectivity Classification
- Exact and Approximate Range Queries in Ball Mapper
- The optimal-transport cartogram: world population as a Brenier map
- Low-Dimensional Embeddings for Gaussian Kernels on Manifolds
- Direct Topology Tracking in Continuous Implicit Models