Influencing Factors Analysis and Prediction Model Development of Stroke: The Machine Learning Approach

Abstract

Prediction is an important way to analyse stroke risk management. This study explored the critical influencing factors of stroke, used the classical multilayer perception (MLP) and radial basis function (RBF) machine learning (ML) algorithms to develop the model for stroke prediction. The two models were trained with Bagging and Boosting ensemble learning algorithms. The performances of the prediction models were also compared with other classical ML algorithms. The result showed that (1) total cholesterol (TC) and other nine factors were selected as principal factors for the stroke prediction; (2) the MLP model outperformed RBF model in terms of accuracy, generalization and inter-rater reliability; (3) ensemble algorithm was superior to single algorithms for high-dimension dataset in this study. It may come to the conclusion that this study improved the stroke prediction methods and contributed much to the prevention of stroke.