审查(临床试验)
机器学习
人工智能
计算机科学
生存分析
加权
公制(单位)
逆概率加权
事件(粒子物理)
统计
工程类
数学
医学
估计员
放射科
物理
量子力学
运营管理
作者
B. Srujana,Deepti Verma,Sameen Naqvi
标识
DOI:10.1080/03610918.2022.2060510
摘要
Machine Learning Models are known to understand the intricacies of the data well, but native ML models cannot be used in time-to-event analysis due to censoring. In this paper, we explore the use of Machine Learning Models in the field of Survival Analysis using right censored Heart Failure Clinical Records Dataset. For this purpose, we first identify the top most important features responsible for death due to heart failure using Recursive Feature Elimination and then see how Machine Learning models can be adapted to improve the time-to-event analysis outcomes. To deal with this, Machine Learning Models are modified using the techniques Inverse Probability of Censoring Weighting (IPCW) and IPCW Bagging and are trained using the processed dataset alongside various survival analysis models. Area Under the time-dependent ROC (AUC) is used as a performance metric. The results reveal that the average AUC value for Survival Analysis Models is 0.51 while that of Machine Learning Models processed using IPCW increased to 0.80, and those processed using IPCW Bagging increased by 0.82. This reflects that Machine Learning models outperform Survival Analysis models in the case of time-to-event analysis of right censored dataset, and hence, are better indicators of risk of heart disease.
科研通智能强力驱动
Strongly Powered by AbleSci AI