| Literature DB >> 33415612 |
Qingchun Guo1,2, Zhenfang He3,4.
Abstract
The outbreak of coronavirus disease 2019 (COVID-19) has seriously affected the environment, ecology, economy, society, and human health. With the global epidemic dynamics becoming more and more serious, the prediction and analysis of the confirmed cases and deaths of COVID-19 has become an important task. We develop an artificial neural network (ANN) for modeling of the confirmed cases and deaths of COVID-19. The confirmed cases and deaths data are collected from January 20 to November 11, 2020 by the World Health Organization (WHO). By introducing root mean square error (RMSE), correlation coefficient (R), and mean absolute error (MAE), statistical indicators of the prediction model are verified and evaluated. The size of training and test confirmed cases and death base employed in the model is optimized. The best simulating performance with RMSE, R, and MAE is realized using the 7 past days' cases as input variables in the training and test dataset. And the estimated R are 0.9948 and 0.9683, respectively. Compared with different algorithms, experimental simulation shows that trainbr algorithm has better performance than other algorithms in reproducing the amount of the confirmed cases and deaths. This study shows that the ANN model is suitable for predicting the confirmed cases and deaths of COVID-19 in the future. Using the ANN model, we also predict the confirmed cases and deaths of COVID-19 from June 5, 2020 to November 11, 2020. During the predicting period, the R, RMSE, and MAE for new infected confirmed cases of COVID-19 are 0.9848, 17,554, and 12,229, respectively; the R, RMSE, and MAE for new confirmed deaths of COVID-19 are 0.8593, 631.8, and 463.7, respectively. The predicted confirmed cases and deaths of COVID-19 are very close to the actual confirmed cases and deaths. The results show that continuous and strict control measures should be taken to prevent the further spread of the epidemic.Entities:
Keywords: Artificial intelligence; COVID-19; Deaths; Epidemic; Infected cases; SARS-CoV-2
Year: 2021 PMID: 33415612 PMCID: PMC7789896 DOI: 10.1007/s11356-020-11930-6
Source DB: PubMed Journal: Environ Sci Pollut Res Int ISSN: 0944-1344 Impact factor: 4.223
Fig. 1The architecture of ANN predicting the COVID-19 epidemic
Fig. 2The cycles of new infected confirmed cases of COVID-19
Figure 3Optimization of network topologies for predicting the COVID-19 epidemic. a RMSE for different nodes in the input layer. b RMSE for different nodes in the hidden layer
Variables selected using the proposed ANN model
| Variables | RMSE | MAE | ||||
|---|---|---|---|---|---|---|
| Training | Testing | Training | Testing | Training | Testing | |
| 1 | 0.9853 | 0.1546 | 6585.6 | 14,394 | 3897.3 | 11,018 |
| 2 | 0.9868 | 0.2229 | 6253.5 | 13,643 | 3837.5 | 11,132 |
| 3 | 0.9867 | 0.2273 | 6228.8 | 12,866 | 3862 | 10,449 |
| 4 | 0.9872 | 0.1873 | 6178.8 | 13,552 | 4158.7 | 11,458 |
| 5 | 0.9875 | 0.2081 | 6061.4 | 13,390 | 3730.7 | 11,236 |
| 6 | 0.9891 | 0.2443 | 5730.7 | 14,773 | 3532.5 | 12,958 |
| 7 | 0.9948 | 0.9683 | 3859.4 | 3102.9 | 2303.7 | 2090.6 |
| 8 | 0.9887 | 0.6762 | 5728.6 | 15,227 | 3668.8 | 13,052 |
| 9 | 0.9892 | 0.6719 | 5701.8 | 14,251 | 3560.8 | 12,021 |
| 10 | 0.989 | 0.677 | 5748.6 | 15,019 | 3668.3 | 12,665 |
| 11 | 0.9895 | 0.6714 | 5591.4 | 14,184 | 3536.5 | 11,863 |
| 12 | 0.9899 | 0.6673 | 5452.8 | 13,825 | 3363.7 | 11,643 |
| 13 | 0.9901 | 0.6736 | 5410.4 | 14,007 | 3277.4 | 11,738 |
| 14 | 0.9916 | 0.8446 | 4918.8 | 9752 | 3143.8 | 7519 |
| 15 | 0.9899 | 0.7232 | 5416.1 | 14949 | 3441 | 12,610 |
| 16 | 0.9898 | 0.6652 | 5507.2 | 15,309 | 3570.3 | 12,884 |
Comparison between different nodes in hidden layer
| Nodes | RMSE | MAE | ||||
|---|---|---|---|---|---|---|
| Training | Testing | Training | Testing | Training | Testing | |
| 1 | 0.9867 | 0.353 | 6242.7 | 12,579 | 4061.2 | 10,446 |
| 2 | 0.9903 | 0.8034 | 5246.3 | 7171.9 | 3373.7 | 6282.5 |
| 3 | 0.9913 | 0.9343 | 4986.2 | 4610.7 | 3038.5 | 3523.5 |
| 4 | 0.9925 | 0.934 | 4626.7 | 4557.7 | 2696.8 | 3612.5 |
| 5 | 0.9891 | 0.566 | 5788.6 | 15,006 | 3616 | 12,771 |
| 6 | 0.9948 | 0.9683 | 3859.4 | 3102.9 | 2303.7 | 2090.6 |
| 7 | 0.989 | 0.5573 | 5794.8 | 15,851 | 3697.7 | 13,501 |
| 8 | 0.9887 | 0.6104 | 5835.5 | 13,106 | 3703.4 | 10,976 |
| 9 | 0.9891 | 0.587 | 5796.1 | 14,413 | 3863.9 | 12,237 |
| 10 | 0.9894 | 0.6362 | 5743 | 15,529 | 3790.8 | 13,108 |
| 11 | 0.9891 | 0.6005 | 5780.5 | 13,390 | 3797 | 11,274 |
| 12 | 0.9891 | 0.5958 | 5790.7 | 13,479 | 3860.9 | 11,362 |
| 13 | 0.9891 | 0.6062 | 5750.2 | 13,537 | 3703.2 | 11,401 |
| 14 | 0.989 | 0.5786 | 5768.4 | 12,898 | 3669.9 | 10,832 |
| 15 | 0.9889 | 0.5794 | 5796.9 | 12,891 | 3662.6 | 10,820 |
| 16 | 0.9895 | 0.7329 | 5663.9 | 12,602 | 3554.3 | 10,433 |
Figure 4Optimization of training algorithms for predicting the COVID-19 epidemic
Simulation performance and optimization of training algorithms for the ANN model.
| Training functions | RMSE | MAE | ||||
|---|---|---|---|---|---|---|
| Training | Testing | Training | Testing | Training | Testing | |
| trainbr | 0.9948 | 0.9683 | 3859.4 | 3102.9 | 2303.7 | 2090.6 |
| trainlm | 0.9926 | 0.9152 | 3859.4 | 5359.8 | 2588.4 | 4018.7 |
| traingdx | 0.9859 | 0.5529 | 6474.4 | 16,696 | 4512.7 | 14,399.2 |
| traingd | 0.9774 | 0.3153 | 12497 | 31,499 | 10,950 | 29,203.5 |
| traingdm | 0.9748 | 0.2415 | 9698.8 | 28,951 | 7763 | 26,445.3 |
| trainrp | 0.9877 | 0.5393 | 5929.9 | 10,751 | 3747.6 | 8208.7 |
| traincgp | 0.9903 | 0.842 | 5255.8 | 6721.8 | 3387.3 | 6227.2 |
| traincgf | 0.9898 | 0.7956 | 5408.6 | 8154.2 | 3632.3 | 6477.2 |
| traincgb | 0.9883 | 0.4287 | 6150.5 | 14,420 | 4644.4 | 12,517.3 |
| trainscg | 0.9897 | 0.7045 | 5512 | 10,617 | 3546.2 | 8860.9 |
| trainbfg | 0.9902 | 0.8085 | 5287.6 | 6974.6 | 3530.5 | 6120.7 |
| trainoss | 0.9929 | 0.7654 | 4817.3 | 9533.7 | 3143.9 | 8154.2 |
Comparison between various transfer functions
| Transfer function | RMSE | MAE | ||||
|---|---|---|---|---|---|---|
| Training | Testing | Training | Testing | Training | Testing | |
| tansig-purelin | 0.9886 | 0.6121 | 5923.6 | 16,494 | 4011.5 | 2570.3 |
| tansig-logsig | 0.8519 | 0.2859 | 49,990 | 98,425 | 36,582 | 10,950 |
| tansig-tansig | 0.8987 | 0.3475 | 49,992 | 98,415 | 36,640 | 10,949 |
| logsig-purelin | 0.9893 | 0.5656 | 5666.5 | 14,566 | 3636.8 | 2388.9 |
| logsig-tansig | 0.794 | 0.3671 | 50,015 | 98,425 | 37,098 | 10,950 |
| logsig-logsig | 0.8577 | 0.4409 | 50,096 | 98,401 | 38,285 | 10,948 |
| purelin-tansig | 0.9185 | 0.2577 | 49,983 | 98,397 | 36,478 | 10,947 |
| purelin-logsig | 0.936 | 0.2941 | 50,026 | 98,397 | 37,482 | 10,947 |
| purelin-purelin | 0.9848 | 0.3944 | 6572.7 | 11,027 | 4491.6 | 2963.9 |
| tansig-poslin | 0.9948 | 0.9683 | 3859.4 | 3102.9 | 2303.7 | 2090.6 |
| logsig-poslin | 0.9885 | 0.5262 | 5873.7 | 14,344 | 4013.8 | 2365.3 |
COVID-19 prediction of the amount of new infected confirmed cases, total infected confirmed cases, new infected deaths, total infected deaths in 2020
| COVID-19 | RMSE | MAE | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Training | Testing | Predicting | Training | Testing | Predicting | Training | Testing | Predicting | |
| New cases | 0.9948 | 0.9683 | 0.9848 | 3859.4 | 3102.9 | 17,554.0 | 2303.7 | 2090.6 | 12,229.0 |
| Total cases | 1 | 0.9998 | 0.9999 | 16,586.0 | 14,694.0 | 453,720.0 | 13,903.0 | 1223.3 | 324,000.0 |
| New deaths | 0.979 | 0.6028 | 0.8593 | 567.2 | 828.8 | 631.8 | 320.0 | 58.7 | 463.7 |
| Total deaths | 0.9998 | 0.9989 | 0.9999 | 4125.2 | 1028.4 | 4861.6 | 2806.0 | 94.6 | 4213.4 |
Fig. 5Prediction of the amount of new infected confirmed cases of COVID-19 in 2020
Fig. 6Prediction of the amount of total infected confirmed cases of COVID-19 in 2020
Fig. 7Prediction of the amount of new infected deaths of COVID-19 in 2020
Fig. 8Prediction of the amount of total infected deaths of COVID-19 in 2020
COVID-19 prediction of the amount of new infected confirmed cases in 10 countries in 2020
| Country | RMSE | MAE | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Training | Testing | Predicting | Training | Testing | Predicting | Training | Testing | Predicting | |
| USA | 0.956 | 0.9165 | 0.9696 | 3987.4 | 1938.5 | 5139.1 | 2331.3 | 1567.4 | 4074.6 |
| India | 0.9848 | 0.9285 | 0.9933 | 269.6 | 359.7 | 2969.9 | 137.5 | 291.5 | 2106.0 |
| Brazil | 0.9752 | 0.9964 | 0.8505 | 1037.2 | 985.6 | 7187.2 | 511.5 | 830.2 | 5264.9 |
| Russian Federation | 0.976 | 0.9991 | 0.9978 | 312.8 | 24.0 | 311.9 | 86.2 | 18.9 | 205.3 |
| France | 0.8047 | 0.9709 | 0.9971 | 968.0 | 405.0 | 1153.0 | 574.3 | 333.0 | 767.8 |
| Spain | 0.9438 | 0.9766 | 0.9914 | 839.0 | 75.0 | 864.5 | 514.6 | 61.2 | 578.6 |
| UK | 0.9702 | 0.9894 | 0.9884 | 537.7 | 137.6 | 1243.2 | 316.1 | 123.7 | 640.7 |
| Argentina | 0.8717 | 0.9853 | 0.9812 | 51.3 | 95.6 | 877.0 | 27.6 | 71.6 | 625.6 |
| Colombia | 0.9436 | 0.9819 | 0.9366 | 74.1 | 118.6 | 1015.6 | 41.3 | 109.2 | 744.8 |
| Italy | 0.9664 | 0.9485 | 0.9983 | 503.4 | 108.5 | 533.7 | 337.5 | 98.7 | 304.3 |
Fig. 9Prediction of the amount of new infected confirmed cases of COVID-19 in 10 countries in 2020