Department of Biomedical Engineering, Carnegie Mellon University, PA, 15213, Pittsburgh, United States.
Department of Mechanical Engineering, Carnegie Mellon University, PA, 15213, Pittsburgh, United States.
Comput Biol Med. 2022 May;144:105342. doi: 10.1016/j.compbiomed.2022.105342. Epub 2022 Feb 23.
After nearly two years since the first identification of SARS-CoV-2 virus, the surge in cases because of virus mutations is a cause of grave public health concern across the globe. As a result of this health crisis, predicting the transmission pattern of the virus is one of the most vital tasks for preparing and controlling the pandemic. In addition to mathematical models, machine learning tools, especially deep learning models have been developed for forecasting the trend of the number of patients affected by SARS-CoV-2 with great success. In this paper, three deep learning models, including CNN, LSTM, and the CNN-LSTM have been developed to predict the number of COVID-19 cases for Brazil, India and Russia. We also compare the performance of our models with the previously developed deep learning models and notice significant improvements in prediction performance. Although our models have been used only for forecasting cases in these three countries, the models can be easily applied to datasets of other countries. Among the models developed in this work, the LSTM model has the highest performance when forecasting and shows an improvement in the forecasting accuracy compared with some existing models. The research will enable accurate forecasting of the COVID-19 cases and support the global fight against the pandemic.
自首次鉴定 SARS-CoV-2 病毒以来,已经过去了将近两年,由于病毒突变导致的病例激增是全球严重的公共卫生关注的原因。由于这场健康危机,预测病毒的传播模式是准备和控制大流行的最重要任务之一。除了数学模型外,机器学习工具,特别是深度学习模型,已被成功开发用于预测受 SARS-CoV-2 影响的患者数量的趋势。在本文中,我们开发了三个深度学习模型,包括 CNN、LSTM 和 CNN-LSTM,用于预测巴西、印度和俄罗斯的 COVID-19 病例数。我们还将我们的模型与以前开发的深度学习模型的性能进行了比较,并注意到预测性能有了显著提高。虽然我们的模型仅用于预测这三个国家的病例,但这些模型可以很容易地应用于其他国家的数据集。在这项工作中开发的模型中,LSTM 模型在预测方面表现最好,并且与一些现有模型相比,在预测准确性方面有所提高。这项研究将能够对 COVID-19 病例进行准确预测,并支持全球抗击大流行。