An Empirical Study of Machine Learning Robustness and Scalability for Imbalanced Tabular Clinical Data in Emergency and Critical Care
Not on the map. Why?
Abstract (arXiv)
Every year, millions of patients pass through emergency departments and intensive care units, where clinicians must make high-stakes decisions under time pressure and uncertainty. Machine learning could support prediction of deterioration, triage, and rare critical outcomes, but clinical data are often severely imbalanced, biasing models toward majority classes and reducing predictive performance. Developing robust and efficient models for imbalanced clinical tabular data therefore remains an important challenge. We evaluated six model families on imbalanced tabular data from the MIMIC-IV-ED and eICU databases: Decision Tree, Random Forest, XGBoost, TabNet, TabICL, and TabPFN v2.6. Trainable models were optimized using Bayesian hyperparameter tuning, while foundation models were evaluated in their pretrained inference regime without task-specific reweighting. Models were assessed using Macro F1-score, robustness to increasing imbalance, and computational scalability across seven clinical prediction tasks. Results differed across datasets. On MIMIC-IV-ED, TabPFN v2.6 and TabICL achieved the strongest average Macro F1 ranks, with XGBoost remaining competitive. On eICU, XGBoost consistently performed best, followed by other tree-based methods, while foundation models achieved intermediate performance. Across both datasets, TabNet showed the largest degradation under increasing imbalance and the highest computational cost. Training-time analysis showed that tree-based methods scaled most favorably with dataset size, while foundation models offered low per-task adaptation cost. These findings suggest that no single model family dominates across all clinical settings. However, tabular foundation models are narrowing the performance gap with strong classical baselines while offering a distinct efficiency-performance trade-off that may benefit resource-constrained clinical environments.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 14 of its 56 references that are in the map. First posted on arXiv 25 Dec 2025.
Builds on
14 of 56 references.
References · 14
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
- 2019Class-Balanced Loss Based on Effective Number of Samplescited by 1,239
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 643
- 2019TabNet: Attentive Interpretable Tabular Learningcited by 640
- 2021Tabular Data: Deep Learning is Not All You Needcited by 530
- 2017A systematic study of the class imbalance problem in convolutional neural networkscited by 510
- 2018Scalable and accurate deep learning for electronic health recordscited by 275
- 2025TabICL: A Tabular Foundation Model for In-Context Learning on Large Datacited by 267
- 2018Empirical Risk Minimization under Fairness Constraintscited by 226
- 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 115
- 2019Explainable AI for Trees: From Local Explanations to Global Understandingcited by 64
2 more
Built on by
Not cited yet.