Discriminating HFrEF vs HFpEF from chest radiographs: Mitigating demographic performance gaps via augmentation and multimodal fusion
Deep learning models applied to chest radiography have shown promise for cardiac phenotyping, yet concerns remain regarding demographic performance disparities and deployment robustness. In this proof-of-concept study, we present a fairness-centered evaluation of chest X-ray-based differentiation between heart failure with reduced ejection fraction (HFrEF) and heart failure with preserved ejection fraction (HFpEF) using publicly available data. We constructed a cohort from MIMIC-CXR linked to MIMIC-IV ICD-10 phenotypes, restricted to cases with radiographic pulmonary edema, and evaluated image-only modeling, data augmentation, and a lightweight multimodal fusion strategy incorporating demographic and comorbidity information. A DenseNet121 image-only model achieved modest discrimination (AUC = 0.61-0.64) and exhibited measurable performance gaps across race, age, and sex subgroups. Standard data augmentation improved overall performance and reduced several subgroup disparities. Multimodal fusion further enhanced discrimination (AUC = 0.76) and reduced the largest observed demographic AUC gap by up to 83% (relative reduction). Threshold-level subgroup metrics, including sensitivity and specificity, together with calibration analyses, demonstrated more balanced error profiles across demographic groups at clinically relevant operating points. These findings highlight that simple, reproducible interventions can substantially improve both performance and equity in chest X-ray-based heart failure phenotyping on public datasets.
Authors
- Chao-Ju Chen
- Liam G. McCoy (ORCID: https://orcid.org/0000-0002-4468-2256)
- H. Shu
- Ryan Wang (ORCID: https://orcid.org/0000-0003-2459-2461)
- Po‐Chih Kuo (ORCID: https://orcid.org/0000-0003-4020-3147)
- Leo Anthony Celi
Institutions
- Beth Israel Deaconess Medical Center (US)
- Harvard University (US)
- University of Alberta (CA)
- National Tsing Hua University (TW)
- Massachusetts Institute of Technology (US)
Publication Details
- Journal
- PLOS Digital Health
- Published
- 2026-06-12
- DOI
- https://doi.org/10.1371/journal.pdig.0001467
- Primary Topic
- COVID-19 diagnosis using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Science Foundation
- Ministry of Education
- Ministry of Science and Technology, Taiwan