Individually Fair Gradient Boosting
Authors
Authors
- Alexander Vargo
- Fan Zhang
- Mikhail Yurochkin
- Yuekai Sun
We consider the task of enforcing individual fairness in gradient boosting. Gradient boosting is a popular method for machine learning from tabular data, which arise often in applications where algorithmic fairness is a concern. At a high level, our approach is a functional gradient descent on a (distributionally) robust loss function that encodes our intuition of algorithmic fairness for the ML task at hand. Unlike prior approaches to individual fairness that only work with smooth ML models, our approach also works with non-smooth models such as decision trees. We show that our algorithm converges globally and generalizes. We also demonstrate the efficacy of our algorithm on three ML problems susceptible to algorithmic bias.
Please cite our work using the BibTeX below.
@misc{vargo2021individually,
title={Individually Fair Gradient Boosting},
author={Alexander Vargo and Fan Zhang and Mikhail Yurochkin and Yuekai Sun},
year={2021},
eprint={2103.16785},
archivePrefix={arXiv},
primaryClass={cs.LG}
}