Ensemble SHAP Aggregation and Attribution Variability in Clinical Machine Learning: A COVID-19 Mortality Study
Dai, N.; Briceno, R. K.; Castaneda, A. N.; Tresierra, M. A.; Esteban, M. L.; Rosas, M. E.; Hinojosa, R. C.
Show abstract
Objective: To combine performance-weighted ensemble SHAP aggregation with resampling-based assessment of feature-importance variability and patient-level attribution alignment for interpreting COVID-19 mortality predictions. Methods: We analyzed a prospective cohort of 1,857 patients hospitalized with COVID-19 at two hospitals in Peru. Ten predictive algorithms were evaluated using five-fold cross-validation, and their mean AUROC values determined their attribution-aggregation weights. Within each model and training trial, signed SHAP values were normalized by the mean total absolute attribution across evaluation patients before performance-weighted aggregation. Feature importance was summarized across 30 resampled training trials using means and normal-approximation 95% confidence intervals. Patient-level attribution alignment was assessed using cosine similarity between signed feature-attribution vectors for corresponding patients, with whole-trial patient-correspondence randomization and Benjamini--Hochberg correction. Subgroup observations were reweighted toward the evaluated population's feature distributions for secondary comparisons of existing absolute SHAP values. Results: Among the 1,857 included patients, 982 (52.9%) died during hospitalization. Random forest achieved the highest mean AUROC (0.925 +/- 0.010), followed by AdaBoost (0.919 +/- 0.017) and logistic regression (0.918 +/- 0.007), with variability reported as the SEM. Attribution profiles were more similar across repeated training trials of the same algorithm (mean Pearson r=0.783; SEM, 0.004) than across algorithms within the same trial (mean Pearson r=0.369; SEM, 0.006). The largest normalized attribution magnitudes were observed for dexamethasone use at home without oxygen support (6.186%; 95% CI, 5.795--6.577), PaO2/FiO2 ratio (3.014%; 95% CI, 2.787--3.242), shortness of breath (2.774%; 95% CI, 2.592--2.956), and FiO2 (2.560%; 95% CI, 2.284--2.836). Features differed in patient-level attribution alignment, indicating that importance magnitude and robustness provided complementary information. Among 1,763 eligible subgroup--feature comparisons, 64 had nominal one-sided p < 0.05, although none remained below 0.05 after Benjamini--Hochberg adjustment. Conclusions: Algorithms with similar predictive performance produced substantially different feature-attribution profiles. Normalized performance-weighted SHAP aggregation provided a representative relative-importance summary across algorithms, while resampling-based alignment assessment qualified the consistency of individual feature contributions. The resulting attribution patterns describe fitted-model behavior and should not be interpreted as causal clinical effects.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- A comprehensive ML-based Respiratory Monitoring System for Physiological Monitoring & Resource Planning in the ICU 93%
- Machine Learning Generalizability Across Healthcare Settings: Insights from multi-site COVID-19 screening 92%
- Predicting critical state after COVID-19 diagnosis: Model development using a large US electronic health record dataset 92%
Similar papers in this journal
- Optimising supervised machine learning algorithms predicting cigarette cravings and lapses for a smoking cessation just-in-time adaptive intervention (JITAI) 92%
- Machine learning in predicting respiratory failure in patients with COVID-19 pneumonia - challenges, strengths, and opportunities in a global health emergency 92%
- A comparison of machine learning models versus clinical evaluation for mortality prediction in patients with sepsis 92%
Similar papers in this journal
- Development and Prospective Validation of a Transparent Deep Learning Algorithm for Predicting Need for Mechanical Ventilation 92%
- Rethinking Blood Eosinophils for Assessing ICS Response in COPD: A Post-Hoc Analysis from FLAME 89%
- Effect of Ventilator Mode on Ventilator-Free Days in Critically Ill Adults: A Randomized Trial 88%
Similar papers in this journal
- Developing And Validating COVID-19 Adverse Outcome Risk Prediction Models From A Bi-National European Cohort Of 5594 Patients 93%
- Evaluation of Domain Generalization and Adaptation on Improving Model Robustness to Temporal Dataset Shift in Clinical Medicine 93%
- Predicting bloodstream infection outcome using machine learning 92%
Similar papers in this journal
- Remote Covid Assessment in Primary Care (RECAP) risk prediction tool: derivation and real-world validation studies 92%
- Performance of intensive care unit severity scoring systems across different ethnicities. 91%
- Predictive performance and clinical application of COV50, a urinary proteomic biomarker in early COVID-19 infection: a cohort study 90%
"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.