Predictionof Brain Stroke Using Federated Learning
Journal Title: International Journal of Innovations in Science and Technology - Year 2024, Vol 6, Issue 4
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 KNearest 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 enhancing stroke 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.
Authors and Affiliations
Rabia Tehseen, Uzma Omer, Rubab Javaid, Maham Mehr, Madiha Yousaf, AyeshaZaheer
Enhancing Cardiovascular Disease Risk Prediction Using Resampling and Machine Learning
Cardiovascular Disease (CVD) remains a critical health concern around the globe, requiring precise risk prediction approaches for timely intervention. The primary motive of this study is to enhance CVD risk prediction...
Using Spatial Covariance of Geometric and Shape Based Features for Recognition of Basic and Compound Emotions
Introduction / Importance of Study: Compound emotion recognition has been an emerging area of research for the last decade due to its vast applications in surveillance systems, suspicious person detection, detection of...
Enhancing Management Strategies: Machine Learningand Creative Performance Insights in Employee Attrition Analysisand Prediction
Employee attrition and excessive turnover are major difficulties in today's competitive employment market, affecting many industries. To overcome these difficulties, firms are increasingly relying on artificial intelli...
Investigate the Operating Temperature Effect onFast Pyrolysis Products ofFood Waste withHydrogen
Energy crises and environmental pollution are the main issues of concern all over the world and the disposal of wastes by converting into gaseous products can reduce this to a level. Investigating how operating tempera...
EvaluatingFasterR-CNNandYOLOv8forTrafficObject Detection andClass-Based Counting
Real-time traffic object detection is a critical component necessary for achieving a fully autonomous traffic system. Traffic object detection, along with background classification, is a significant area of research ai...