Predictionof Brain Stroke Using Federated Learning

Rabia Tehseen ()
Additional contact information
Rabia Tehseen: University of Central Punjab, Lahore, Pakistan 57400

International Journal of Innovations in Science & Technology, 2024, vol. 6, issue 4, 1995-2013

Abstract: Stroke often arises from an abrupt blockage in the blood vessels supplying the brain and heart. Detecting early warning signs of stroke can significantly reduce its impact. In this study, we propose an early prediction method for stroke using various Machine Learning (ML) techniques, considering factors such as hypertension, body mass index, heart disease, average glucose levels, smoking habits, prior stroke history, and age. These attributes, rich in information, were utilized to train three distinct classifiers: Logistic Regression, Decision Tree,and K-nearest neighbors for stroke prediction. In this study, Federated Learning (FL)has been applied to combine the ML models (Logistic Regression (LR), Decision Tree (DT), and K-Nearest Neighbors(KNN)) from distributed medical data sources while preserving patient privacy. By aggregating locally trained models from multiple hospitals or devices, FL ensures the robustness of the weighted voting classifier without requiring direct data sharing, thereby enhancingstroke prediction accuracy across diverse datasets. Subsequently, the results from these base classifiers were combined using a weighted voting approach to achieve the highest accuracy. Our study demonstrated an impressive accuracy rate of 97%, with the weighted voting classifier outperforming the individual base classifiers. This model proved to be the more accurate in predicting strokes. Additionally, the Area Under the Curve (AUC) value for the weighted voting classifier was notably high, and it exhibited the lowest false positive and false negative rates compared to other classifiers. Consequently, the weighted voting classifier emerged as an almost ideal tool for predicting strokes, offering valuable support to both physicians and patients in identifying and preventing potential stroke incidents.

Keywords: Stroke; Machine Learning; CM; AUC; WV; Correlation Matrix (search for similar items in EconPapers)
Date: 2024
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journal.50sea.com/index.php/IJIST/article/view/1130/1670 (application/pdf)
https://journal.50sea.com/index.php/IJIST/article/view/1130 (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:abq:ijist1:v:6:y:2024:i:4:p:1995-2013

Access Statistics for this article

International Journal of Innovations in Science & Technology is currently edited by Prof. Dr. Syed Amer Mahmood

More articles in International Journal of Innovations in Science & Technology from 50sea
Bibliographic data for series maintained by Iqra Nazeer ().