LLM2D
通过特征选择技术的一种新型投票系统增强中风疾病分类的机器学习模型
Enhancing stroke disease classification through machine learning models via a novel voting system by feature selection techniques
作者: Mahade Hasan, Farhana Yasmin, Md. Mehedi Hassan, Xue Yu, Soniya Yeasmin, Herat Joshi, Sheikh Mohammed Shariful Islam
发布日期: 4/2/2025
arXiv ID: oai:arXiv.org:2504.00485v1

摘要

arXiv:2504.00485v1 Announce Type: cross 摘要:心脏病仍然是全球主要的死亡和发病率原因,需要开发准确可靠的预测模型以促进早期检测和干预。虽然最先进的研究工作集中在各种机器学习方法来预测心脏病,但它们无法实现显著的准确性。为应对这一需求,我们应用了九种机器学习算法:XGBoost、逻辑回归、决策树、随机森林、K近邻(KNN)、支持向量机(SVM)、高斯朴素贝叶斯(Gaussian NB)、自适应提升以及线性回归,基于一系列生理指标预测心脏病。我们的方法涉及特征选择技术,以识别最相关的预测因子,旨在完善模型以增强性能和可解释性。这些模型经过训练,并采用了如网格搜索超参数调整和交叉验证等过程,以降低过拟合。此外,我们还开发了一种新的投票系统,并通过特征选择技术促进心脏病分类。此外,我们使用准确率、精确率、召回率、F1分数以及受试者操作特征曲线下的面积(ROC AUC)等关键性能指标来评估这些模型。在这些模型中,XGBoost表现出色,分别实现了99%的准确率、精确率、F1分数,98%的召回率以及100%的ROC AUC。本研究提供了一种有前景的心脏病早期诊断和预防性医疗保健方法。