Abstract
Stroke is a dangerous health issue that happens when bleeding valves in the brain get damaged. and blood supply to the brain is cut off. As a result of these factors, numerous body parts may cease to function. Stroke is currently a significant risk factor for mortality, as per World Health Organization (WHO). It may be preferable to mitigate the severity of stroke by detecting it early. In recent years, data science has been critical to the growth of research in the medical field. Various machine learning techniques are built employing a patient’s physical and physiological reporting data to forecast the risk of stroke. In this article, we use five machine learning approaches to find the best effective model that can predict the risk of stroke, including Decision Tree (DT), XGBoost, Light Gradient Boosting Machine (LGBM), Random Forest (RF), and K-nearest Neighbors learning. Kaggle was used to collect the dataset. Random forest produces acceptable findings with an accuracy of roughly 96%, which might be included in actual clinical information. The machine learning approach can aid in the prediction of many diseases like a stroke in the early stages.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Govindarajan, P., Soundarapandian, R.K., Gandomi, A.H., Patan, R., Jayaraman, P., Manikandan, R.: Classification of stroke disease using machine learning algorithms. Neural Comput. Appl. 32(3), 817–828 (2020)
Reza, S.M., Rahman, M.M., Mamun, S.A.: A new approach for road networks-a vehicle xml device collaboration with big data. In: Proceedings of the International Conference on Electrical Engineering and Information and Communication Technology, pp. 1–5. Mirpur, Dhaka (2014)
Amini, L., Azarpazhouh, R., Farzadfar, M.T., et al.: Prediction and control of stroke by data mining. Int. J. Prev. Med. 4(2), S245–S249 (2013)
Cheng, C.A., Lin, Y.C., Chiu, H.W.: Prediction of the prognosis of ischemic stroke patients after intravenous thrombolysis using artificial neural networks. Stud. Health Technol. Inf. 202, 115–118 (2014)
Sung, S.-F., et al.: Developing a stroke severity index based on administrative data was feasible using data mining techniques. J. Clin. Epidemiol. 68(11), 1292–1300 (2015)
Monteiro, M., et al.: Using machine learning to improve the prediction of functional outcome in ischemic stroke patients. IEEE/ACM Trans. Comput. Biol. Bioinf. 15(6), 1953–1959 (2018)
Cheon, S., Kim, J., Lim, J.: -e use of deep learning to predict stroke patient mortality. Int. J. Environ. Res. Public Health 16(11), 1876 (2019)
Singh M.S., Choudhary, P.: Stroke prediction using artificial intelligence. In: Proceedings of the 2017 8th Annual Industrial Automation And Electromechanical Engineering Conference (IEMECON), pp. 158–161. Bangkok, Thailand (2017)
Chin, C.-L., et al.: An automated early ischemic stroke detection system using CNN deep learning algorithm. In: Proceedings of the 2017 IEEE 8th International Conference on Awareness Science and Technology (iCAST), pp. 368–372. Taichung, Taiwan (2017)
Adam, S.Y., Yousif, A., Bashir, M.B.: Classification of ischemic stroke using machine learning algorithms. Int. J. Comput. Appl. 149(10), 26–31 (2016)
Kansadub, T., -ammaboosadee, S., Kiattisin, S., Jalayondeja, C.: Stroke risk prediction model based on demographic data. In: Proceedings of the 2015 8th Biomedical Engineering International Conference (BMEiCON), pp. 1–3. Pattaya, Thailand (2015)
Badriyah, T., Sakinah, N., Syarif, I., Syarif, D.R.: Machine learning algorithm for stroke disease classification. In: 2020 International Conference on Electrical, Communication, and Computer Engineering (ICECCE), pp. 1–5. IEEE (2020)
Stroke Dataset. https://www.kaggle.com/fedesoriano/stroke-prediction-dataset. Accessed 30 Dec 2021
Ali, M.M., Paul, B.K., Ahmed, K., Bui, F.M., Quinn, J.M., Moni, M.A.: Heart disease prediction using supervised machine learning algorithms: performance analysis and comparison. Comput. Biol. Med. 136, 104672 (2021)
Ali, M.M., et al.: Machine learning-based statistical analysis for early stage detection of cervical cancer. Comput. Biol. Med. 139, 104985 (2021)
Ramraj, S., Uzir, N., Sunil, R., Banerjee, S.: Experimenting XGBoost algorithm for prediction and classification of different datasets. Int. J. Control Theory Appl. 9, 651–662 (2016)
Suguna, N., Thanushkodi, K.: An improved k-nearest neighbor classification using genetic algorithm. Int. J. Comput. Sci. Issues 7(2), 18–21 (2010)
Ahamed, B.S., Arya, S.: LGBM classifier based technique for predicting type-2 diabetes. Eur. J. Mol. Clin. Med. 8(3), 454–467 (2021)
Tazin, T., Alam, M.N., Dola, N.N., Bari, M.S., Bourouis, S., Khan, M.M.: Stroke disease detection and prediction using robust learning approaches. J. Healthcare Eng. 2021, 1–12 (2021)
Sailasya, G., Kumari, G.L.A.: Analyzing the performance of stroke prediction using ML classification algorithms. Int. J. Adv. Comput. Sci. Appl. 12(6), 539–545 (2021)
Hemu, A.A., Mim, R.B., Ali, M.M., Nayer, M., Ahmed, K., Bui, F.M.: Identification of significant risk factors and impact for ASD prediction among children using machine learning approach. In: 2022 Second International Conference on Advances in Electrical, Computing, Communication and Sustainable Technologies (ICAECT), pp. 1–6. IEEE (2022)
Rahaman, M.A., Ali, M.M., Ahmed, K., Bui, F.M., Mahmud, S.H.: Performance analysis between YOLOv5s and YOLOv5m model to detect and count blood cells: deep learning approach. In: Proceedings of the 2nd International Conference on Computing Advancements, pp. 316–322 (2022)
Ontor, M.Z.H., Ali, M.M., Hossain, S.S., Nayer, M., Ahmed, K., Bui, F.M.: YOLO_CC: deep Learning based approach for early stage detection of cervical cancer from cervix images using YOLOv5s model. In: 2022 Second International Conference on Advances in Electrical, Computing, Communication and Sustainable Technologies (ICAECT), pp. 1–5. IEEE (2022)
Acknowledgement
The study was supported by funding from the Natural Sciences and Engineering Research Council of Canada (NSERC).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Hossain, M.S., Shovo, M.H., Ali, M.M., Nayer, M., Ahmed, K., Bui, F.M. (2022). ML_SPS: Stroke Prediction System Employing Machine Learning Approach. In: Kumar, A., Fister Jr., I., Gupta, P.K., Debayle, J., Zhang, Z.J., Usman, M. (eds) Artificial Intelligence and Data Science. ICAIDS 2021. Communications in Computer and Information Science, vol 1673. Springer, Cham. https://doi.org/10.1007/978-3-031-21385-4_19
Download citation
DOI: https://doi.org/10.1007/978-3-031-21385-4_19
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-21384-7
Online ISBN: 978-3-031-21385-4
eBook Packages: Computer ScienceComputer Science (R0)