arXiv:2504.00485cs.LGcs.AI2025-04被引 1

用特征选择提升9种机器学习模型,实现心脏病99%准确分类

Stroke Disease Classification Using Machine Learning with Feature Selection Techniques

  • 结合特征选择与9种算法,优化心脏病预测模型
  • XGBoost达99%准确率、100%ROC AUC,性能最优
  • 适合医疗辅助诊断与可解释性要求高的场景

心脏病仍是全球主要致死致残原因,亟需高效可靠的预测模型以实现早期发现与干预。尽管现有研究采用多种机器学习方法预测心脏病,但准确率仍不理想。为此,本文应用九种机器学习算法(XGBoost、逻辑回归、决策树、随机森林、K近邻、支持向量机、高斯朴素贝叶斯、自适应提升、线性回归),基于多项生理指标预测心脏病。通过特征选择技术筛选关键预测因子,提升模型性能与可解释性。模型训练中采用网格搜索超参数调优和交叉验证,有效降低过拟合风险。此外,提出一种融合特征选择的新型投票系统以进一步提升分类效果。使用准确率、精确率、召回率、F1分数及受试者工作特征曲线下面积(ROC AUC)等指标评估模型表现。其中,XGBoost表现最佳,达到99%准确率、99%精确率、99% F1分数、98%召回率及100% ROC AUC。

原文摘要 · Abstract (English)

Heart disease remains a leading cause of mortality and morbidity worldwide, necessitating the development of accurate and reliable predictive models to facilitate early detection and intervention. While state of the art work has focused on various machine learning approaches for predicting heart disease, but they could not able to achieve remarkable accuracy. In response to this need, we applied nine machine learning algorithms XGBoost, logistic regression, decision tree, random forest, k-nearest neighbors (KNN), support vector machine (SVM), gaussian naïve bayes (NB gaussian), adaptive boosting, and linear regression to predict heart disease based on a range of physiological indicators. Our approach involved feature selection techniques to identify the most relevant predictors, aimed at refining the models to enhance both performance and interpretability. The models were trained, incorporating processes such as grid search hyperparameter tuning, and cross-validation to minimize overfitting. Additionally, we have developed a novel voting system with feature selection techniques to advance heart disease classification. Furthermore, we have evaluated the models using key performance metrics including accuracy, precision, recall, F1-score, and the area under the receiver operating characteristic curve (ROC AUC). Among the models, XGBoost demonstrated exceptional performance, achieving 99% accuracy, precision, F1-Score, 98% recall, and 100% ROC AUC. This study offers a promising approach to early heart disease diagnosis and preventive healthcare.

心脏病预测机器学习特征选择XGBoost

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。