A Machine Learning Approach to Detect the Brain Stroke Disease

Abstract

The brain, which comprises the cerebrum, cere-bellum, and brainstem and is covered by the skull, is a very complex and intriguing organ in the human body. Stroke is the world's second-leading cause of mortality; as a result, it requires prompt treatment to avoid brain damage. Early detection of a brain stroke can help to prevent or lessen the severity of the stroke, which can lower death rates. Using machine learning algorithms to identify risk variables is a promising method. This paper proposed a model that included a methodology to achieve an accurate brain stroke forecast. The efficient data collection, data pre-processing, and data transformation methods have been applied to provide reliable information for our proposed model to be successful. A “brain stroke dataset” was employed to build up the model. The standardization technique is used to standardize data. In the training and testing procedure, Random Forest (RF), Support Vector Machine (SVM), and Decision Tree (DT) classifiers are applied. The performance of each classifier has been estimated by adopting performance evaluation metrics such as accuracy, sensitivity (SEN), error rate, false-positive rate (FPR), false-negative rate (FNR), root mean square error, and log loss. Based on the outcome while using the RF classifier, we can determine that our proposed model provided the maximum accuracy, which was 95.30%.