Optimal Transport under Group Fairness Constraints
By: Linus Bleistein , Mathieu Dagréou , Francisco Andrade and more
Potential Business Impact:
Makes sure computer matches are fair for everyone.
Ensuring fairness in matching algorithms is a key challenge in allocating scarce resources and positions. Focusing on Optimal Transport (OT), we introduce a novel notion of group fairness requiring that the probability of matching two individuals from any two given groups in the OT plan satisfies a predefined target. We first propose \texttt{FairSinkhorn}, a modified Sinkhorn algorithm to compute perfectly fair transport plans efficiently. Since exact fairness can significantly degrade matching quality in practice, we then develop two relaxation strategies. The first one involves solving a penalised OT problem, for which we derive novel finite-sample complexity guarantees. This result is of independent interest as it can be generalized to arbitrary convex penalties. Our second strategy leverages bilevel optimization to learn a ground cost that induces a fair OT solution, and we establish a bound guaranteeing that the learned cost yields fair matchings on unseen data. Finally, we present empirical results that illustrate the trade-offs between fairness and performance.
Similar Papers
Optimal Transport Learning: Balancing Value Optimization and Fairness in Individualized Treatment Rules
Machine Learning (Stat)
Makes computer decisions fair for everyone.
Scalable Approximate Algorithms for Optimal Transport Linear Models
Machine Learning (Stat)
Helps computers learn from data faster and better.
Federated Sinkhorn
Distributed, Parallel, and Cluster Computing
Helps computers learn from private data together.