arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.13367cs.LGcs.AIq-bio.BMstat.ML
收藏

基于堆叠包围机器学习算法的心力衰竭患者生存预测

Survival Prediction of Heart Failure Patients using Stacked Ensemble Machine Learning Algorithm

Islamic University of Technology(伊斯兰科技大学);Khulna University of Engineering & Technology(库尔纳工程技术大学);University of Toronto(多伦多大学)

S. M Mehedi Zaman, Wasay Mahmood Qureshi, Md. Mohsin Sarker Raihan, Ocean Monjur, Abdullah Bin Shams

摘要:

心血管疾病,特别是心力衰竭是我们这个时代的主要健康危害问题之一,也是全球死亡的主要原因。使用机器学习(ML)模型的数据采掘技术的进步为有前途的预测方法铺平了道路。数据采掘是将医疗机构创建的大量原始数据转换为有意义的信息的过程,这些信息可以帮助做出预测和关键决策。本研究的主要目的是收集心力衰竭患者的各种随访数据,分析这些数据,并利用几种ML模型来预测心血管患者的生存可能性。由于数据集中类的不平衡性,采用了合成少数类过采样技术(SMOTE)。在我们的研究中使用了两种无监督模型(K-Means和模糊C-均值聚类)和三种有监督分类器(随机森林、XGBoost和决策树)。经过彻底的调查,我们的结果表明,监督ML算法的性能优于无监督模型。此外,我们设计并提出了一个有监督的堆叠集成学习模型,可以达到99.98%的准确率,精确率,召回率和F1得分。我们的研究表明,只有从患者中收集的某些属性才能使用监督ML算法成功预测心力衰竭后的生存可能性。

英文摘要:

Cardiovascular disease, especially heart failure is one of the major health hazard issues of our time and is a leading cause of death worldwide. Advancement in data mining techniques using machine learning (ML) models is paving promising prediction approaches. Data mining is the process of converting massive volumes of raw data created by the healthcare institutions into meaningful information that can aid in making predictions and crucial decisions. Collecting various follow-up data from patients who have had heart failures, analyzing those data, and utilizing several ML models to predict the survival possibility of cardiovascular patients is the key aim of this study. Due to the imbalance of the classes in the dataset, Synthetic Minority Oversampling Technique (SMOTE) has been implemented. Two unsupervised models (K-Means and Fuzzy C-Means clustering) and three supervised classifiers (Random Forest, XGBoost and Decision Tree) have been used in our study. After thorough investigation, our results demonstrate a superior performance of the supervised ML algorithms over unsupervised models. Moreover, we designed and propose a supervised stacked ensemble learning model that can achieve an accuracy, precision, recall and F1 score of 99.98%. Our study shows that only certain attributes collected from the patients are imperative to successfully predict the surviving possibility post heart failure, using supervised ML algorithms.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑