Mortality Prediction in COVID-19 Using Time Series and Machine Learning Techniques

Abstract

Predicting mortality in COVID-19 is one of the most significant and difficult tasks at hand. This study compares time series and machine learning methods, including support vector machines (SVMs) and neural networks (NNs), to forecast the mortality rate in seven countries: the United States, India, Brazil, Russia, France, China, and Bangladesh. Data were gathered between December 31, 2019, when COVID-19 began, and March 31, 2021. The study used 457 observations with 4 variables: daily confirmed cases, daily deaths, daily mortality rate, and date. To predict the death rate in the seven countries that were chosen, the data were analyzed using time series analysis and machine learning techniques. Models were compared to obtain more accurate mortality predictions. The autoregressive integrated moving average (ARIMA) model with the lowest AIC value for each nation is found through time series analysis. By increasing the hidden layer and applying machine learning techniques, the NN model for each country is chosen, and the optimal model is determined by determining the model with the lowest error value. Additionally, SVM analyzes every country and calculates its R² and root-mean-square error (RMSE). The lowest RMSE value is used to compare all of the time series and machine learning models. According to the comparison table, SVM provides a more accurate model to predict the mortality rate of the seven countries, with the lowest RMSE value. During the study period, mortality rates increased in Brazil and Russia and decreased in the United States, India, France, China, and Bangladesh, according to the comparison value of RMSE in this study. Furthermore, this paper shows that SVM outperforms all other models in terms of performance. According to the author’s analysis of the data, SVM is a machine learning technique that can be used to accurately predict mortality in a pandemic scenario.