Stroke Prediction Analysis using Machine Learning Classifiers and Feature Technique
Abstract
Stroke is one of the fatal brain diseases that cause death in 3 to 10 hours. However, most stroke mortality can be prevented by identifying the nature of the stroke and reacting to it promptly through smart health systems. In this paper, a machine learning model is approached for predicting the existence of stroke of a patient where the Random forest classifier outperforms the state-of-the-art models, including Logistic Regression, Decision Tree Classifier (DTC), K-NN. We conduct the experiments on datasets which has 5110 observations with 12 attributes. We also applied EDA for preprocessing and feature techniques for balancing the datasets. Finally, a cloud-based mobile app collects user data to analyze and provide the possibility of stroke for alerting the person with the accuracy of precision 96%, recall 96%, and F1-score 96%. This user-friendly system can be a lifesaver as the person gets an essential warning very easily by providing very little information from anywhere with a mobile device.
Full Text:
PDFReferences
“Stroke (Causes, Symptoms, and Complications) - Assignment Point.” https://www.assignmentpoint.com/science/medical/stroke-causes-symptoms-and (accessed May 16, 2021).
Mayo Clinic, “Stroke - Symptoms and Causes,” Mayo Clinic, Nov. 06, 2020. https://www.mayoclinic.org/diseases-conditions/stroke/symptoms-causes/syc-20350113.
B. Wedro, “Stroke Warning Signs, Symptoms, Treatment, Types & Causes,” MedicineNet, 2019. https://www.medicinenet.com/stroke_symptoms_and_treatment/article.htm
H. Rodgers, “Stroke,” Neurological Rehabilitation, pp. 427–433, 2013, doi: 10.1016/b978-0-444-52901-5.00036-8
H. Asadi, R. Dowling, B. Yan, and P. Mitchell, “Machine Learning for Outcome Prediction of Acute Ischemic Stroke Post Intra-Arterial Therapy,” PLoS ONE, vol. 9, no. 2, p. e88225, Feb. 2014, doi: 10.1371/journal.pone.0088225.
P. Govindarajan, R. K. Soundarapandian, A. H. Gandomi, R. Patan, P. Jayaraman, and R. Manikandan, “Classification of stroke disease using machine learning algorithms,” Neural Computing and Applications, vol. 32, no. 3, pp. 817–828, Jan. 2019, doi: 10.1007/s00521-019-04041-y.
T. Lumley, R. A. Kronmal, M. Cushman, T. A. Manolio, and S. Goldstein, “A stroke prediction score in the elderly,” Journal of Clinical Epidemiology, vol. 55, no. 2, pp. 129–136, Feb. 2002, doi: 10.1016/s0895-4356(01)00434-6.
P. A. Barber et al., “Prediction of stroke outcome with echoplanar perfusion- and diffusion-weighted MRI,” Neurology, vol. 51, no. 2, pp. 418–426, Aug. 1998, doi: 10.1212/wnl.51.2.418.
G. A. Hitman et al., “Stroke prediction and stroke prevention with atorvastatin in the Collaborative Atorvastatin Diabetes Study (CARDS),” Diabetic Medicine, vol. 24, no. 12, pp. 1313–1321, Dec. 2007, doi: 10.1111/j.1464-5491.2007.02268.x.
H. Asadi, R. Dowling, B. Yan, and P. Mitchell, “Machine Learning for Outcome Prediction of Acute Ischemic Stroke Post Intra-Arterial Therapy,” PLoS ONE, vol. 9, no. 2, p. e88225, Feb. 2014, doi: 10.1371/journal.pone.0088225.
A. Laghari, Z. A. Memon, S. Ullah and I. Hussain, "Cyber Physical System for Stroke Detection," in IEEE Access, vol. 6, pp. 37444-37453, 2018, doi: 10.1109/ACCESS.2018.2851540.
H. Kuang et al., "Computed Tomography Perfusion–Based Machine Learning Model Better Predicts Follow-Up Infarction in Patients With Acute Ischemic Stroke," stroke, vol. 52, no. 1, pp. 223–231, Jan. 2021, doi: 10.1161/strokeaha.120.030092.
T. Ohkubo et al., “How many times should blood pressure be measured at home for better prediction of stroke risk? Ten-year follow-up results from the Ohasama study,” Journal of Hypertension, vol. 22, no. 6, pp. 1099–1104, Jun. 2004, doi: 10.1097/00004872-200406000-00009.
B. Letham, C. Rudin, T. H. McCormick, and D. Madigan, “Interpretable classifiers using rules and Bayesian analysis: Building a better stroke prediction model,” Annals of Applied Statistics, vol. 9, no. 3, pp. 1350–1371, Sep. 2015, doi: 10.1214/15-AOAS848.
C. Hung, W. Chen, P. Lai, C. Lin and C. Lee, "Comparing deep neural network and other machine learning algorithms for stroke prediction in a large-scale population-based electronic medical claims database," 2017 39th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), 2017, pp. 3110-3113, doi: 10.1109/EMBC.2017.8037515.
A. Khosla, Y. Cao, C. C.-Y. Lin, H.-K. Chiu, J. Hu, and H. Lee, “An integrated machine learning approach to stroke prediction,” Proceedings of the 16th ACM SIGKDD international conference on Knowledge discovery and data mining - KDD ’10, 2010, doi: 10.1145/1835804.1835830.
T. Liu, W. Fan, and C. Wu, “A hybrid machine learning approach to cerebral stroke prediction based on imbalanced medical dataset,” Artificial Intelligence in Medicine, vol. 101, p. 101723, Nov. 2019, doi: 10.1016/j.artmed.2019.101723.
M. Monteiro et al., "Using Machine Learning to Improve the Prediction of Functional Outcome in Ischemic Stroke Patients," in IEEE/ACM Transactions on Computational Biology and Bioinformatics, vol. 15, no. 6, pp. 1953-1959, 1 Nov.-Dec. 2018, doi: 10.1109/TCBB.2018.2811471.
P. Prati et al., “Carotid Plaque Morphology Improves Stroke Risk Prediction: Usefulness of a New Ultrasonographic Score,” Cerebrovascular Diseases, vol. 31, no. 3, pp. 300–304, 2011, doi: 10.1159/000320852.
T. Kansadub, S. Thammaboosadee, S. Kiattisin and C. Jalayondeja, "Stroke risk prediction model based on demographic data," 2015 8th Biomedical Engineering International Conference (BMEiCON), 2015, pp. 1-3, doi: 10.1109/BMEiCON.2015.7399556.
“Stroke Prediction Dataset,” kaggle.com.https://www.kaggle.com/fedesoriano/stroke-prediction-dataset.
MM. Islam, MA. Kashem, J. Uddin, “Fish survival prediction in an aquatic environment using random forest model,” IAES International Journal of Artificial Intelligence (IJ-AI), vol. 10, no. 3, pp. 614–622, 2021, doi: 10.11591/ijai.v10.i3.pp614-622.
Md. M. Islam, J. Uddin, M. A. Kashem, F. Rabbi, and Md. W. Hasnat, “Design and Implementation of an IoT System for Predicting Aqua Fisheries Using Arduino and KNN,” Intelligent Human Computer Interaction, pp. 108–118, 2021, doi: 10.1007/978-3-030-68452-5_11.
A. Esmael, M. Elsherief, and K. Eltoukhy, “Predictive Value of the Alberta Stroke Program Early CT Score (ASPECTS) in the Outcome of the Acute Ischemic Stroke and Its Correlation with Stroke Subtypes, NIHSS, and Cognitive Impairment,” Stroke Research and Treatment, vol. 2021, pp. 1–10, Jan. 2021, doi: 10.1155/2021/5935170.
C. Y. Baek, W. N. Chang, B. Y. Park, K. B. Lee, K. Y. Kang, and M. R. Choi, “Effects of dual-task gait treadmill training on gait ability, dual-task interference, and fall efficacy in people with stroke: A Randomized Controlled Trial,” Physical Therapy, Feb. 2021, doi: 10.1093/ptj/pzab067.
DOI: http://dx.doi.org/10.24042/ijecs.v1i2.10393
Refbacks
- There are currently no refbacks.
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
International Journal of Electronics and Communications System (IJECS) is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
