VOOZH about

URL: https://pubmed.ncbi.nlm.nih.gov/37990734/

⇱ Improving Fairness in AI Models on Electronic Health Records: The Case for Federated Learning Methods - PubMed


Clipboard, Search History, and several other advanced features are temporarily unavailable.
Skip to main page content
👁 Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

👁 Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation

Add to Collections

Add to My Bibliography

Your saved search

Create a file for external citation management software

Your RSS Feed

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.

PubMed Disclaimer

Figures

👁 Figure 1:
Figure 1:. Federated Learning architecture where hospitals collaborate to learn a joint model. At each round, the clients send their trained parameters to the server (blue arrows). The server then aggregates them and sends the parameters back to the clients (orange arrows). The clients then update the model based on their local dataset.
👁 Figure 2:
Figure 2:. Dipole’s architecture with adversarial debiasing.
👁 Figure 3:
Figure 3:. Impact of the fairness budget on the Accuracy, TPSD, Worst TPR, and APSD for the MIMIC-III IID cohort. The shaded areas represent the standard deviation.

References

    1. 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
    1. 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
    1. Aurangzeb Ahmad Muhammad, Patel Arpit, Eckert Carly, Kumar Vikas, and Teredesai Ankur. 2020. Fairness in Machine Learning for Healthcare. 10.1145/3394486.3406461 - DOI
    1. 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).
    1. 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.

LinkOut - more resources

Cite

NCBI Literature Resources

MeSH PMC Bookshelf Disclaimer

The PubMed wordmark and PubMed logo are registered trademarks of the U.S. Department of Health and Human Services (HHS). Unauthorized use of these marks is strictly prohibited.