A genetic algorithm based federated learning model for stroke risk prediction

Abstract

A Stroke is a sudden loss of blood circulation in certain parts of the brain that results in a loss of neurological function. To save a patient from stroke, an immediate diagnosis and treatment plan must be implemented. Artificial intelligence-based machine learning algorithms play a major role in the prediction. To predict a person likely to have a stroke, stroke healthcare data records must be accessed, which is very sensitive. Data shared for machine learning training pose security risks and have concerns about privacy. To overcome this issue, Genetic Algorithm and Federated Learning (GA-FL) –based hybridization approach is proposed to predict the risk of stroke in a person. Federated Learning was developed by Google, which can provide security to the data during the training process because every client participating in this training process needs to exchange only the training parameters without sharing the data. In addition to the security features, a genetic algorithm was used to optimize the parameters required to train a model using the perceptron neural network model. The experimental results show that our proposed research model (GA-FL) provides security and predicts the risk of stroke more accurately than any other existing algorithm.