Prediction of COVID-19 confirmed cases combining deep learning methods and Bayesian optimization

• Published in: Chaos, solitons and fractals Vol. 142; p. 110511
• Main Authors: Abbasimehr, Hossein, Paki, Reza
• Format: Journal Article
• Language: English
• Published: England Elsevier Ltd 01.01.2021
• Subjects: Bayesian optimization; CNN; COVID-19; Deep learning; LSTM; Multi-head attention; COVID-19; Deep learning; CNN; LSTM; Bayesian optimization; Multi-head attention
• ISSN: 0960-0779; 1873-2887; 0960-0779
• DOI: 10.1016/j.chaos.2020.110511

•Three methods combining deep learning and Bayesian optimization are proposed.•Bayesian optimization efficiently selects the optimized values for hyperparameters.•The design of methods is based on the multiple-output forecasting strategy.•The proposed methods outperform the benchmark model on COVID-19 time series data. COVID-19 virus has encountered people in the world with numerous problems. Given the negative impacts of COVID-19 on all aspects of people's lives, especially health and economy, accurately forecasting the number of cases infected with this virus can help governments to make accurate decisions on the interventions that must be taken. In this study, we propose three hybrid approaches for forecasting COVID-19 time series methods based on combining three deep learning models such as multi-head attention, long short-term memory (LSTM), and convolutional neural network (CNN) with the Bayesian optimization algorithm. All models are designed based on the multiple-output forecasting strategy, which allows the forecasting of the multiple time points. The Bayesian optimization method automatically selects the best hyperparameters for each model and enhances forecasting performance. Using the publicly available epidemical data acquired from Johns Hopkins University's Coronavirus Resource Center, we conducted our experiments and evaluated the proposed models against the benchmark model. The results of experiments exhibit the superiority of the deep learning models over the benchmark model both for short-term forecasting and long-horizon forecasting. In particular, the mean SMAPE of the best deep learning model is 0.25 for the short-term forecasting (10 days ahead). Also, for long-horizon forecasting, the best deep learning model obtains the mean SMAPE of 2.59.