Centralized CNN–GRU Model by Federated Learning for COVID-19 Prediction in India

Abstract

In 2019, the corona virus was found in Wuhan, China. The corona virus has traveled several countries in the world from the beginning of 2020. The early estimation of COVID-19 cases is one of the efficient approaches to control the pandemic. Many researchers had proposed the deep learning model for the efficient estimation of COVID-19 cases for different provinces in the world. The research work had not focused on the discussion of robustness in the model. In this study, centralized federated-convolutional neural network–gated recurrent unit (Fed-CNN–GRU) model is proposed for the estimation of active cases per day in different provinces of India. In India, the uneven transmission of COVID-19 virus was seen in 36 provinces due to the different geographical areas and population densities. So, the methodology of this study had focused on the development of single deep learning algorithm, which is robust and reliable to estimate the active cases of COVID-19 in different provinces of India. The concept of transfer and federated learning is involved to enhance the estimation of active cases of COVID-19 by the CNN–GRU model. The study had considered the active cases per day dataset for 36 provinces in India from 12 March, 2020 to 17 January, 2022. Based on the study, it is proven that the centralized CNN–GRU model by federated learning had captured the transmission dynamics of COVID-19 in different provinces with an enhanced result.