Predicting Functional Outcome After Ischemic Stroke Using Logistic Regression and Machine Learning Models
This research employed binary logistic regression and machine learning techniques; Decision Tree, Random Forest, and Support Vector Machine (SVM), to predict functional outcomes following ischemic stroke. The main goal was to determine the most suitable model for the dataset through a comprehensive performance evaluation. Four models were examined for predicting post-ischemic stroke functional outcomes: Decision Tree, Random Forest, Logistic Regression, and SVM. The evaluation involved metrics such as Accuracy, Precision, F1-Score, and Recall. The Logistic Regression model achieved the highest accuracy at 90%, accurately predicting outcomes in 90% of cases. However, it had lower precision (50%), indicating an increased rate of false positive predictions. On the other hand, the SVM model displayed the highest precision (71.3%), implying fewer false positive predictions. It also attained the highest F1-Score (77.5%), indicating a strong balance between precision and Recall compared to the other models. Notably, the Logistic Regression model achieved perfect Recall (100%), correctly identifying all positive outcomes, while the Random Forest model showed significant recall performance (93.2%). Conversely, the Decision Tree model exhibited moderate accuracy (66.11%) but lower precision (66%), F1-Score (6.15%), and recall (3.2%), suggesting challenges with false positives and false negatives. Choosing the best model depends on analysis priorities. For accurate identification of positive outcomes, the Logistic Regression model's perfect recall is advantageous. For balanced performance, the SVM model's high F1-Score makes it a compelling option.
Saini, Anshul (2021). Decision Tree Algorithm - A Complete Guide. Analytics Vidhya. https://www.analyticsvidhya.com/blog/2021/08/decision-tree-algorithm/
Biau, G., & Scornet, E. (2016). A random forest guided tour. Test, 25, 197-227. https://doi.org/10.1007/s11749-016-0481-7
Choi, Y.-A., Park, S., Jun, J.-A., Ho, C. M. B., Pyo, C.-S., Lee, H., & Yu, J. (2021). Machine-learning-based elderly stroke monitoring system using electroencephalography vital signals. Applied Sciences, 11(4), 1761. https://doi.org/10.3390/app11041761
Chang, W., Liu, Y., Xiao, Y., Yuan, X., Xu, X., Zhang, S., & Zhou, S. (2019). A machine-learning-based prediction method for hypertension outcomes based on medical data. Diagnostics, 9(4), 178. https://doi.org/10.3390/diagnostics9040178
Hanna, K. L., & Rowe, F. J. (2017). Health inequalities associated with post-stroke visual impairment in the United Kingdom and Ireland: A systematic review. Neuro-Ophthalmology, 41(3), 117-136. https://doi.org/10.1080/01658107.2017.1279640
Wang, L. (2023). Logistic regression for stroke prediction: an evaluation of its accuracy and validity. Highlights in Science, Engineering and Technology, 39, 1086-1092. https://doi.org/10.54097/hset.v39i.6712
Mirzaikamrani, S. (2020). Predictive modeling and classification for Stroke using machine learning methods. http://www.diva-portal.se/smash/get/diva2:1430021/FULLTEXT01.pdf
Restrepo, L. (2004). Handbook of stroke prevention in clinical practice. Texas Heart Institute Journal, 31(4), 460.
Volinsky, C. T., & Raftery, A. E. (2000). Bayesian information criterion for censored survival models. Biometrics, 56(1), 256-262. https://doi.org/10.1111/j.0006-341x.2000.00256.x
Wu, O., Cloonan, L., Mocking, S. J. T., Bouts, M. J. R. J., Copen, W. A., Cougo-Pinto, P. T., Fitzpatrick, K., Kanakis, A., Schaefer, P. W., Rosand, J., & others. (2015). Role of acute lesion topography in initial ischemic stroke severity and long-term functional outcomes. Stroke, 46(9), 2438-2444. https://doi.org/10.1161/strokeaha.115.009643
Yang, L., Liu, Q., Zhao, Q., Zhu, X., & Wang, L. (2020). Machine learning is a valid method for predicting prehospital delay after acute ischemic stroke. Brain and Behavior, 10(10), e01794. https://doi.org/10.1002/brb3.1794
Vapnik, V. (1999). The nature of statistical learning theory. Springer Science & Business Media.
This work is licensed under a Creative Commons Attribution 4.0 International License.