Improving Pediatric Emergency Department Triage with Modality Dropout in Late Fusion Multimodal EHR Models
Organizations: Department of Computer Science, Stanford University · Department of Bioengineering, Stanford University
Abstract
Emergency department triage relies heavily on both quantitative vital signs and qualitative clinical notes, yet multimodal machine learning models predicting triage acuity often suffer from modality collapse by over-relying on structured tabular data. This limitation severely hinders demographic generalizability, particularly for pediatric patients where developmental variations in vital signs make unstructured clinical narratives uniquely crucial. To address this gap, we propose a late-fusion multimodal architecture that processes tabular vitals via XGBoost and unstructured clinical text via Bio_ClinicalBERT, combined through a Logistic Regression meta-classifier to predict the 5-level Emergency Severity Index. To explicitly target the external validity problem, we train our model exclusively on adult encounters from the MIMIC-IV and NHAMCS datasets and evaluate its zero-shot generalization on a traditionally overlooked pediatric cohort. Furthermore, we employ symmetric modality dropout during training to prevent the ensemble from overfitting to adult-specific clinical correlations. Our results demonstrate that the multimodal framework significantly outperforms single-modality baselines. Most notably, applying a 30-40% symmetric modality dropout rate yielded steep performance improvements in the unseen pediatric cohort, elevating the Quadratic Weighted Kappa to 0.351. These findings highlight modality dropout as a critical regularization technique for mitigating modality collapse and enhancing cross-demographic generalization in clinical AI.
Figures & tables
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Performances on Adult Cohort | ||||||
|---|---|---|---|---|---|---|
| Model | Training Error | Test Error | QWK | Accuracy | Balanced Acc | Macro F1 |
| XGBoost Class | 0.888 | 0.912 | 0.424 | 0.600 | 0.360 | 0.370 |
| XGBoost Regress | 0.612 | 0.621 | 0.413 | 0.600 | 0.340 | 0.350 |
| TF-IDF | 1.74 | 1.74 | 0.434 | 0.498 | 0.491 | 0.364 |
| BioClinicalBERT | 0.762 | 0.778 | 0.538 | 0.654 | 0.410 | 0.433 |
| Multimodal | 0.721 | 0.724 | 0.633 | 0.696 | 0.471 | 0.496 |
| Performances on Pediatric Cohort | ||||
|---|---|---|---|---|
| Model | QWK | Accuracy | Balanced Acc | Macro F1 |
| XGBoost Class | 0.302 | 0.560 | 0.340 | 0.380 |
| XGBoost Regress | 0.297 | 0.510 | 0.260 | 0.250 |
| TF-IDF | 0.134 | 0.325 | 0.272 | 0.226 |
| BioClinicalBERT | 0.287 | 0.481 | 0.285 | 0.284 |
| Multimodal | 0.331 | 0.559 | 0.309 | 0.317 |
Clinical Triage
Clinical triage uses patient symptoms, records, and measurements to prioritize urgency, referral, and care when time and resources are limited.
See all 48 papers