Fairness-Aware Interpretable Modeling (FAIM) for Trustworthy Machine Learning in Healthcare

Mingxuan Liu, Yilin Ning, Yuhe Ke, Yuqing Shang, Bibhas Chakraborty +3 more
3/8/2024
cs.LGcs.AIcs.CY

No official code found. Best implementation: nliulab/FAIM (5 stars, MIT, last commit 2024-06-06).

Abstract

The escalating integration of machine learning in high-stakes fields such as healthcare raises substantial concerns about model fairness. We propose an interpretable framework - Fairness-Aware Interpretable Modeling (FAIM), to improve model fairness without compromising performance, featuring an interactive interface to identify a "fairer" model from a set of high-performing models and promoting the integration of data-driven evidence and clinical expertise to enhance contextualized fairness. We demonstrated FAIM's value in reducing sex and race biases by predicting hospital admission with two real-world databases, MIMIC-IV-ED and SGH-ED. We show that for both datasets, FAIM models not only exhibited satisfactory discriminatory performance but also significantly mitigated biases as measured by well-established fairness metrics, outperforming commonly used bias-mitigation methods. Our approach demonstrates the feasibility of improving fairness without sacrificing performance and provides an a modeling mode that invites domain experts to engage, fostering a multidisciplinary effort toward tailored AI fairness.

View on arXivView PDF

Code Implementations(1)

nliulab/FAIMCommunity100%
MIT

Fairness-Aware Interpretable Modelling (FAIM) for Trustworthy Machine Learning

50HTML, PythonJun 21, 20232 years agoMIT

Similar Papers

Cite this paper

@article{liu2024fairnessaware,
  title  = {Fairness-Aware Interpretable Modeling (FAIM) for Trustworthy Machine Learning in Healthcare},
  author = {Mingxuan Liu and Yilin Ning and Yuhe Ke and Yuqing Shang and Bibhas Chakraborty and Marcus Eng Hock Ong and Roger Vaughan and Nan Liu},
  year   = {2024},
  eprint = {2403.05235},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.05235v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.05235.svg)](https://codeofpaper.com/paper/2403.05235)

Discussion