Fair Wasserstein Coresets

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Xiong, Zikai, Dalmasso, Niccolò, Sharma, Shubham, Lecue, Freddy, Magazzeni, Daniele, Potluru, Vamsi K., Balch, Tucker, Veloso, Manuela
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909370768424960
author Xiong, Zikai
Dalmasso, Niccolò
Sharma, Shubham
Lecue, Freddy
Magazzeni, Daniele
Potluru, Vamsi K.
Balch, Tucker
Veloso, Manuela
author_facet Xiong, Zikai
Dalmasso, Niccolò
Sharma, Shubham
Lecue, Freddy
Magazzeni, Daniele
Potluru, Vamsi K.
Balch, Tucker
Veloso, Manuela
contents Data distillation and coresets have emerged as popular approaches to generate a smaller representative set of samples for downstream learning tasks to handle large-scale datasets. At the same time, machine learning is being increasingly applied to decision-making processes at a societal level, making it imperative for modelers to address inherent biases towards subgroups present in the data. While current approaches focus on creating fair synthetic representative samples by optimizing local properties relative to the original samples, their impact on downstream learning processes has yet to be explored. In this work, we present fair Wasserstein coresets (FWC), a novel coreset approach which generates fair synthetic representative samples along with sample-level weights to be used in downstream learning tasks. FWC uses an efficient majority minimization algorithm to minimize the Wasserstein distance between the original dataset and the weighted synthetic samples while enforcing demographic parity. We show that an unconstrained version of FWC is equivalent to Lloyd's algorithm for k-medians and k-means clustering. Experiments conducted on both synthetic and real datasets show that FWC: (i) achieves a competitive fairness-utility tradeoff in downstream models compared to existing approaches, (ii) improves downstream fairness when added to the existing training data and (iii) can be used to reduce biases in predictions from large language models (GPT-3.5 and GPT-4).
format Preprint
id arxiv_https___arxiv_org_abs_2311_05436
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Fair Wasserstein Coresets
Xiong, Zikai
Dalmasso, Niccolò
Sharma, Shubham
Lecue, Freddy
Magazzeni, Daniele
Potluru, Vamsi K.
Balch, Tucker
Veloso, Manuela
Machine Learning
Computers and Society
Data distillation and coresets have emerged as popular approaches to generate a smaller representative set of samples for downstream learning tasks to handle large-scale datasets. At the same time, machine learning is being increasingly applied to decision-making processes at a societal level, making it imperative for modelers to address inherent biases towards subgroups present in the data. While current approaches focus on creating fair synthetic representative samples by optimizing local properties relative to the original samples, their impact on downstream learning processes has yet to be explored. In this work, we present fair Wasserstein coresets (FWC), a novel coreset approach which generates fair synthetic representative samples along with sample-level weights to be used in downstream learning tasks. FWC uses an efficient majority minimization algorithm to minimize the Wasserstein distance between the original dataset and the weighted synthetic samples while enforcing demographic parity. We show that an unconstrained version of FWC is equivalent to Lloyd's algorithm for k-medians and k-means clustering. Experiments conducted on both synthetic and real datasets show that FWC: (i) achieves a competitive fairness-utility tradeoff in downstream models compared to existing approaches, (ii) improves downstream fairness when added to the existing training data and (iii) can be used to reduce biases in predictions from large language models (GPT-3.5 and GPT-4).
title Fair Wasserstein Coresets
topic Machine Learning
Computers and Society
url https://arxiv.org/abs/2311.05436