用贝叶斯优化的XGBoost框架,减轻重症医疗中的性别预测偏见。
FairMed-XGB: A Bayesian-Optimised Multi-Metric Framework with Explainability for Demographic Equity in Critical Healthcare Data
- 融合三种公平性指标的损失函数,用贝叶斯搜索优化模型。
- 在多个数据库上减少40%~72%的性别偏差,准确率几乎不变。
- 通过SHAP解释偏差修正机制,适合关注医疗AI伦理的开发者。
部署于重症监护环境的机器学习模型存在性别偏差,削弱临床信任与治疗公平性。本文提出FairMed-XGB框架,系统检测并缓解性别预测偏差,同时保持模型性能与可解释性。该框架将统计均等差异、泰尔指数和沃斯泰因距离结合成公平性感知损失函数,通过贝叶斯搜索优化至XGBoost分类器。在基于MIMIC-IV-ED和eICU数据库的七个临床队列上评估显示:统计均等差异在MIMIC-IV-ED上降低40%~51%,在eICU上降低10%~19%;泰尔指数下降四到五个数量级至接近零;沃斯泰因距离减少20%~72%。预测准确性无明显下降(AUC-ROC降幅<0.02)。基于SHAP的可解释性分析表明,该框架减少了对性别代理特征的依赖,为临床医生提供偏差修正的可操作洞察。FairMed-XGB为高风险医疗场景中可信的公平决策提供了一种鲁棒、可解释且合乎伦理的解决方案。
原文摘要 · Abstract (English)
Machine learning models deployed in critical care settings exhibit demographic biases, particularly gender disparities, that undermine clinical trust and equitable treatment. This paper introduces FairMed-XGB, a novel framework that systematically detects and mitigates gender-based prediction bias while preserving model performance and transparency. The framework integrates a fairness-aware loss function combining Statistical Parity Difference, Theil Index, and Wasserstein Distance, jointly optimised via Bayesian Search into an XGBoost classifier. Post-mitigation evaluation on seven clinically distinct cohorts derived from the MIMIC-IV-ED and eICU databases demonstrates substantial bias reduction: Statistical Parity Difference decreases by 40 to 51 percent on MIMIC-IV-ED and 10 to 19 percent on eICU; Theil Index collapses by four to five orders of magnitude to near-zero values; Wasserstein Distance is reduced by 20 to 72 percent. These gains are achieved with negligible degradation in predictive accuracy (AUC-ROC drop <0.02). SHAP-based explainability reveals that the framework diminishes reliance on gender-proxy features, providing clinicians with actionable insights into how and where bias is corrected. FairMed-XGB offers a robust, interpretable, and ethically aligned solution for equitable clinical decision-making, paving the way for trustworthy deployment of AI in high-stakes healthcare environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。