将新生儿死亡率预测的机器学习技术进行比较:来自建模竞赛的见解
Brynne A Sullivan1, Alvaro G Moreira2, Ryan M McAdams3,4
1University of Virginia, Department of Pediatrics, Division of Neonatology, Charlottesville, VA, USA. brynne@virginia.edu.
Pediatric research
|December 16, 2024
概括
预测新生儿重症监护室 (NICU) 死亡率是复杂的. 简单的逻辑回归优于复杂的机器学习模型,强调谨慎的方法选择而不是复杂性,以更好地分层风险.
科学领域:
- 新生儿医学 新生儿医学
- 医疗保健中的机器学习
- 临床信息学 临床信息学
背景情况:
- 由于患者数据复杂,预测新生儿重症监护室 (NICU) 的死亡率具有挑战性.
- 机器学习 (ML) 为改善NICU设置中的风险分层提供了一个机会.
研究的目的:
- 为了比较各种ML模型对NICU死亡率的预测性能.
- 在以团队为基础的建模竞赛格式中评估模型性能.
主要方法:
- 五个由新生儿科医生领导的团队将ML技术 (逻辑回归,CatBoost,神经网络,随机森林,XGBoost) 应用于超过6,000例NICU入院的数据集.
- 模型使用人口统计,临床,心率和氧和数据预测死亡风险.
- 模型性能是使用接收机操作员特征曲线 (AUC) 下的面积来评估的.
主要成果:
- 后勤回归在测试数据上实现了最高的AUC,超过了更复杂的模型.
- 观众偏好喜欢复杂的模型 (CNN) 来感知现实世界的适用性.
- 团队使用了各种策略来选择功能,超参数调整和模型评估.
结论:
- 模型的复杂性并不能保证在NICU死亡率预测中的优异预测性能.
- 选择建模方法时,应优先考虑数据特征,可解释性和团队专业知识.
- 这项研究主张对ML方法进行深思熟虑的选择,而不是仅仅追求复杂性.
相关概念视频
Mechanistic Models: Compartment Models in Algorithms for Numerical Problem Solving
40
Mechanistic models play a crucial role in algorithms for numerical problem-solving, particularly in nonlinear mixed effects modeling (NMEM). These models aim to minimize specific objective functions by evaluating various parameter estimates, leading to the development of systematic algorithms. In some cases, linearization techniques approximate the model using linear equations.
In individual population analyses, different algorithms are employed, such as Cauchy's method, which uses a...
In individual population analyses, different algorithms are employed, such as Cauchy's method, which uses a...
40
Regression Toward the Mean
6.3K
Regression toward the mean (“RTM”) is a phenomenon in which extremely high or low values—for example, and individual’s blood pressure at a particular moment—appear closer to a group’s average upon remeasuring. Although this statistical peculiarity is the result of random error and chance, it has been problematic across various medical, scientific, financial and psychological applications. In particular, RTM, if not taken into account, can interfere when...
6.3K
Comparing the Survival Analysis of Two or More Groups
149
Survival analysis is a cornerstone of medical research, used to evaluate the time until an event of interest occurs, such as death, disease recurrence, or recovery. Unlike standard statistical methods, survival analysis is particularly adept at handling censored data—instances where the event has not occurred for some participants by the end of the study or remains unobserved. To address these unique challenges, specialized techniques like the Kaplan-Meier estimator, log-rank test, and...
149
Kaplan-Meier Approach
94
The Kaplan-Meier estimator is a non-parametric method used to estimate the survival function from time-to-event data. In medical research, it is frequently employed to measure the proportion of patients surviving for a certain period after treatment. This estimator is fundamental in analyzing time-to-event data, making it indispensable in clinical trials, epidemiological studies, and reliability engineering. By estimating survival probabilities, researchers can evaluate treatment effectiveness,...
94


