2021
DOI: 10.1109/access.2021.3091622
|View full text |Cite
|
Sign up to set email alerts
|

ICU Survival Prediction Incorporating Test-Time Augmentation to Improve the Accuracy of Ensemble-Based Models

Abstract: This work presents a novel method for applying test-time augmentation (TTA) to tabular data. We used TTA along with an ensemble of 42 models to achieve higher performance on the MIT Global Open Source Severity of Illness Score dataset consisting of 131,051 ICU visits and outcomes. This method achieved an AUC of 0.915 on the private test set (19,669 admissions) and won first place at Stanford University's WiDS Datathon 2020 challenge on Kaggle, while the Acute Physiology and Chronic Health Evaluation (APACHE) I… Show more

Help me understand this report

Search citation statements

Order By: Relevance

Paper Sections

Select...
1
1
1
1

Citation Types

1
14
0

Year Published

2022
2022
2024
2024

Publication Types

Select...
9
1

Relationship

1
9

Authors

Journals

citations
Cited by 25 publications
(20 citation statements)
references
References 21 publications
1
14
0
Order By: Relevance
“…Feature extraction, engineering, selection ( Ofer et al, 2021 ), and ML model selection, parameter tuning, and training were performed using the SparkBeyond autoML framework (See patent/US20170017900A1). Previous work has shown the benefit autoML models, in order to comprehensively and automatically find possible predictive signals in complex data, including in biology and healthcare ( Cohen et al, 2021 ). The system automatically extracts and ranks a wide range of compositional features from training data.…”
Section: Methodsmentioning
confidence: 99%
“…Feature extraction, engineering, selection ( Ofer et al, 2021 ), and ML model selection, parameter tuning, and training were performed using the SparkBeyond autoML framework (See patent/US20170017900A1). Previous work has shown the benefit autoML models, in order to comprehensively and automatically find possible predictive signals in complex data, including in biology and healthcare ( Cohen et al, 2021 ). The system automatically extracts and ranks a wide range of compositional features from training data.…”
Section: Methodsmentioning
confidence: 99%
“…In fact, the patients admitted to two types of ICUs account for nearly 70% of the recorded total observations. Following recent work on measuring feature importance for predicting mortality [27] , [28] , we select the related variables and summarize their descriptions in Table 4 . All these variables are recognized as important predictors of mortality in patients.…”
Section: Experiments On the Icu Mortality Data In Us Hospitalsmentioning
confidence: 99%
“…Test data augmentation can help increase the robustness of a trained model [ 73 , 74 , 75 ]. Test data augmentation may be utilized to enhance deep network prediction performance and open up new intriguing possibilities for medical image interpretation [ 76 , 77 , 78 ]. The most often utilized methods for augmenting data include rotating, mirroring, flipping, zooming, and cropping.…”
Section: Datasetsmentioning
confidence: 99%