Abstract
Developing AI tools that preserve fairness is of critical importance, specifically in high-stakes applications such as those in healthcare. However, health AI models' overall prediction performance is often prioritized over the possible biases such models could have. In this study, we show one possible approach to mitigate bias concerns by having healthcare institutions collaborate through a federated learning paradigm (FL; which is a popular choice in healthcare settings). While FL methods with an emphasis on fairness have been previously proposed, their underlying model and local implementation techniques, as well as their possible applications to the healthcare domain remain widely underinvestigated. Therefore, we propose a comprehensive FL approach with adversarial debiasing and a fair aggregation method, suitable to various fairness metrics, in the healthcare domain where electronic health records are used. Not only our approach explicitly mitigates bias as part of the optimization process, but an FL-based paradigm would also implicitly help with addressing data imbalance and increasing the data size, offering a practical solution for healthcare applications. We empirically demonstrate our method's superior performance on multiple experiments simulating large-scale real-world scenarios and compare it to several baselines. Our method has achieved promising fairness performance with the lowest impact on overall discrimination performance (accuracy). Our code is available at https://github.com/healthylaife/FairFedAvg.
Keywords: Adversarial Fairness; Algorithmic Fairness; Federated Learning.
Figures
References
-
- Abay Annie, Zhou Yi, Baracaldo Nathalie, Rajamoni Shashank, Chuba Ebube, and Ludwig Heiko. 2020. Mitigating Bias in Federated Learning. 10.48550/ARXIV.2012.02447 - DOI
-
- Agarwal Alekh, Dudik Miroslav, and Steven Wu Zhiwei. 2019. Fair Regression: Quantitative Definitions and Reduction-Based Algorithms. In Proceedings of the 36th International Conference on Machine Learning (Proceedings of Machine Learning Research, Vol. 97), Kamalika Chaudhuri and Ruslan Salakhutdinov (Eds.).PMLR, 120–129. https://proceedings.mlr.press/v97/agarwal19d.html
-
- Aurangzeb Ahmad Muhammad, Patel Arpit, Eckert Carly, Kumar Vikas, and Teredesai Ankur. 2020. Fairness in Machine Learning for Healthcare. 10.1145/3394486.3406461 - DOI
-
- Beutel Daniel J, Topal Taner, Mathur Akhil, Qiu Xinchi, Parcollet Titouan, and Lane Nicholas D. 2020. Flower: A Friendly Federated Learning Research Framework. arXiv preprint arXiv:2007.14390 (2020).
-
- US Census. [n. d.]. Racial and Ethnic Diversity in the United States: 2010 Census and 2020 Census. https://www.census.gov/library/visualizations/interactive/racial-and-eth.... Accessed: 2022–11-17.
Grants and funding
LinkOut - more resources
Full Text Sources
