Back

Fairness-Aware Machine Learning for Heart Failure Prediction: Performance, Bias, and Clinical Deployment Insights

Eziama, E. U.; Eziama, S. I.; Agboli, V. I.; Amusa, T.; Edokpolor, H.; Adeoba, M. I.; Usman, I. A.; Eghujovbo, V.; Olasege, I.; Ogunnaike, O.; Ige, T. O.; Okunola, D.; Taiwo, F. T.; Anyanacho, J. N.; Olasege, R.; Adeoye, A.

2025-10-19 cardiovascular medicine
10.1101/2025.10.17.25338263 medRxiv
Show abstract

Heart failure (HF) prediction models using machine learning (ML) must achieve a balance between performance, fairness, and real-world clinical utility. This paper assesses the potential of ML and DL models in the context of heterogeneous databases (UCI, MIMIC) and aims to derive applicable schemes for equitable deployment in healthcare. Although the Transformer models depicted notable AUC-ROC in the UCI data (0.986), they suffered a considerable performance degradation in the MIMIC data, suggesting their generalizability issue. Interestingly, we found contrasting gender preferences: SVM better detected females (AUC-ROC = 0.869 vs 0.796 for males), while XGBoost favored males. Decision curve analysis (DCA) demonstrated that a greater AUC-ROC was not necessarily preferred. For the 0.5 threshold, the net benefit of SVM (0.050) was higher than that of other models. SHAP analysis confirmed sex, ejection fraction, and NT-proBNP as the main predictors, but demonstrated inconsistent feature relations between datasets. We argue that (1) gender-specific threshold optimization, (2)ensemble methods to counteract bias, and (3)robust external validation are necessary and provide a possible path for clinically deployable, fairness-aware HF prediction tools.

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.