跳到主要导航 跳到搜索 跳到主要内容

Analysis of Machine Learning Models for Stroke Prediction with Emphasis on Hyperparameter Tuning Techniques

  • Sakib Hasan
  • , Alamgir Islam
  • , Tanjin Islam
  • , Hongbin Ma*
  • *此作品的通讯作者
  • Beijing Institute of Technology
  • Jiangsu University of Science and Technology

科研成果: 书/报告/会议事项章节会议稿件同行评审

摘要

Stroke remains a significant global cause of death and disability, necessitating early and accurate prediction models for prompt intervention. This study contrasts the performance of Support Vector Machine (SVM) and Random Forest (RF) models to enhance stroke prediction approaches. Emphasizing the critical role of hyper parameter adjustment in improving model efficiency, two tuning methods—Grid Search Cross-Validation (GS-CV) and Randomized Search Cross-Validation (RS-CV)—are investigated. Data prepossessing utilizes a data set from the Medical Clinic of Bangladesh, comprising 5,110 patient records. Imbalanced data is addressed through the Synthetic Minority Over-sampling Technique (SMOTE). Despite being good at predicting accuracy, SVM with RS-CV tuning is more accurate, achieving a 96% accuracy than RF with GS-CV tuning that achieves 92% accuracy. Such outcomes highlight the significance of choosing proper hyperparameter tuning techniques and ML models for stroke prediction. They also imply an outlet for use in healthcare contexts concerning early identification and prophylactic steps. This comparison study adds to the current debate about machine learning in medical prediction, focusing on the methodological aspects critical to constructing reliable and effective predictive systems.

源语言英语
主期刊名Computational Intelligence and Industrial Applications - 11th International Symposium, ISCIIA 2024, Proceedings
编辑Bin Xin, Hongbin Ma, Jinhua She, Weihua Cao
出版商Springer Science and Business Media Deutschland GmbH
1-9
页数9
ISBN(印刷版)9789819647552
DOI
出版状态已出版 - 2025
已对外发布
活动11th International Symposium on Computational Intelligence and Industrial Applications, ISCIIA 2024 - Beijing, 中国
期限: 1 11月 20245 11月 2024

出版系列

姓名Communications in Computer and Information Science
2466 CCIS
ISSN(印刷版)1865-0929
ISSN(电子版)1865-0937

会议

会议11th International Symposium on Computational Intelligence and Industrial Applications, ISCIIA 2024
国家/地区中国
Beijing
时期1/11/245/11/24

指纹

探究 'Analysis of Machine Learning Models for Stroke Prediction with Emphasis on Hyperparameter Tuning Techniques' 的科研主题。它们共同构成独一无二的指纹。

引用此