Next Article in Journal
BGP Neighbor Trust Establishment Mechanism Based on the Bargaining Game
Previous Article in Journal
Repositioning of Romanian Seaside Tourism as an Effect of Climate Change
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Analysis and Prediction of COVID-19 Using SIR, SEIQR, and Machine Learning Models: Australia, Italy, and UK Cases

1
Faculty of Engineering, Universiti Putra Malaysia, Serdang 43400, Malaysia
2
Data Science Institute, University of Technology Sydney, Ultimo 2007, Australia
3
Computational Mechanics Laboratory, School of Pedagogical and Technological Education, 15122 Athens, Greece
*
Author to whom correspondence should be addressed.
Submission received: 14 January 2021 / Revised: 13 February 2021 / Accepted: 15 February 2021 / Published: 3 March 2021

Abstract

:
The novel coronavirus disease, also known as COVID-19, is a disease outbreak that was first identified in Wuhan, a Central Chinese city. In this report, a short analysis focusing on Australia, Italy, and UK is conducted. The analysis includes confirmed and recovered cases and deaths, the growth rate in Australia compared with that in Italy and UK, and the trend of the disease in different Australian regions. Mathematical approaches based on susceptible, infected, and recovered (SIR) cases and susceptible, exposed, infected, quarantined, and recovered (SEIQR) cases models are proposed to predict epidemiology in the above-mentioned countries. Since the performance of the classic forms of SIR and SEIQR depends on parameter settings, some optimization algorithms, namely Broyden–Fletcher–Goldfarb–Shanno (BFGS), conjugate gradients (CG), limited memory bound constrained BFGS (L-BFGS-B), and Nelder–Mead, are proposed to optimize the parameters and the predictive capabilities of the SIR and SEIQR models. The results of the optimized SIR and SEIQR models were compared with those of two well-known machine learning algorithms, i.e., the Prophet algorithm and logistic function. The results demonstrate the different behaviors of these algorithms in different countries as well as the better performance of the improved SIR and SEIQR models. Moreover, the Prophet algorithm was found to provide better prediction performance than the logistic function, as well as better prediction performance for Italy and UK cases than for Australian cases. Therefore, it seems that the Prophet algorithm is suitable for data with an increasing trend in the context of a pandemic. Optimization of SIR and SEIQR model parameters yielded a significant improvement in the prediction accuracy of the models. Despite the availability of several algorithms for trend predictions in this pandemic, there is no single algorithm that would be optimal for all cases.

1. Introduction

In December 2019, the Chinese government informed the rest of the world that a virus was rapidly spreading throughout China. A few months later, it had detrimentally spread to several other countries. The United States Centers for Disease Control and Prevention (CDC) identified a seafood market in Wuhan as the center of the outbreak of the novel coronavirus disease (COVID-19), which is caused by severe acute respiratory syndrome coronavirus 2. The World Health Organization (WHO) reported a case in Thailand on 13 January 2020, the first case to be identified outside China. On 16 January 2020, Japan confirmed its first case of the novel coronavirus, followed by South Korea on 20 January. As of today, most countries around the world have been affected.
Numerous studies have been conducted to predict the spread of the virus in order to seek the best prevention measures. For instance, a simulation model based on mobility data was proposed [1], and a particle swarm optimization (PSO) algorithm was used to estimate the parameters (susceptible, infected, recovered) in the susceptible, infected, and recovered (SIR) cases model [1,2]. The results indicate that the latter method is precise enough, with a low margin of error compared with analytical methods. Another study [3] calibrated the SIR model to South Africa after considering different scenarios for the reproduction number (R0) for reporting infections and short-term healthcare resource estimations. Meanwhile, daily temperature and relative humidity were both reported to influence the occurrence of COVID-19 in Hubei province and some other provinces [4].
The authors of [5] proposed a heuristic algorithm to model and evaluate the risk of the COVID-19 pandemic in six different countries/states, namely New York, California, the whole of the USA, Iran, Sweden, and the UK.
Another work [6] developed two COVID-19 prediction models based on genetic programming and found that both models were highly reliable for prediction of COVID-19 cases in India. The researchers in [7] reviewed the most recent COVID-19 forecasting models and identified the most predominant factors to be hospitalization [8,9,10,11], intensive care units [12,13,14], vaccination [15,16], and age groups [17,18,19].
The first case of COVID-19 in Australia was reported in January 2020. In this paper, we also report on a short analysis focusing on Australia, which continued as a short-term simulation.
The manuscript is organized in several sections. Section 2 presents the research methodology. Section 3 and Section 4 introduce the SIR and susceptible, exposed, infected, quarantined, and recovered (SEIQR) models. Section 5 describes the prediction algorithms (the logistic function and Prophet algorithm). Section 6 provides the results, followed by a discussion and concluding remarks in Section 7.

2. Research Methodology

The study was carried out in several phases. First, data were collected from the World Health Organization (WHO) and John Hopkins University, which obtain data from different organizations. After this, the data were analyzed and preprocessed in order to avoid any duplicate or missing values. Numerical tests were performed using Python and R and executed on an Intel® Core i7-4510U, 2.0 GHz, 8 GB, DDR3 Memory computer (Supplementary File). The flowchart of the research methodology is provided in Figure 1.

3. SIR Model

This section introduces the classic form of the SIR model [20,21], which is used to describe the transmission of COVID-19 in Australia, Italy, and UK. The flowchart of the SIR model is presented in Figure 2.
The SIR model shows how a disease spreads through a population. The equations of the SIR model are as shown below [22]:
d s d t = β I S
d I d t = β I S γ I
d R d t = γ I
where:
  • S is the number of susceptible individuals at time t;
  • I is the number of infected individuals at time t;
  • R is the number of recovered individuals at time t;
  • β and γ are the transmission rate and rate of recovery (removal), respectively.

4. SEIQR Model

The SEIQR model, an extended version of SIR [23], models the interaction of people under different conditions: susceptible (S), exposed (E), infected (I), quarantined (Q), and recovered (R). The parameters S, I, and R are the same as those in the SIR model, and E presents the fraction of individuals that have been infected but do not show any signs. The SEIQR model diagram is illustrated in Figure 3.
The equations of the SEIQR model are defined as follows:
d S ( t ) d t = β S ( t ) I ( t ) N α S ( t )
d E ( t ) d t = β S ( t ) I ( t ) N γ E ( t )
d I ( t ) d t = γ E ( t ) δ I ( t )
d Q ( t ) d t = δ I ( t ) λ ( t ) Q ( t ) κ ( t ) Q ( t )
d R ( t ) d t = λ ( t ) Q ( t )
d D ( t ) d t = κ ( t ) Q ( t )
d P ( t ) d t = α S ( t )
where α represents the protection rate; β is the infection rate and illustrates the inverse of the average latent time; γ shows the rate of recovery (removal); δ represents the inverse of the average quarantine time; λ 0   a n d   λ 1 are coefficients used in the time-dependent cure rate; κ 0   a n d   κ 1 are coefficients used in the time-dependent mortality rate [23]; and { S ( t ) , P ( t ) , E ( t ) , I ( t ) , Q ( t ) , R ( t ) , D ( t ) } refer to the susceptible, insusceptible, exposed (infected but not yet infectious, in a latent period), infectious (with infectious capacity and not yet quarantined), quarantined (confirmed and infected), recovered, and closed cases [23].

5. Prediction

The machine learning techniques described in this section were used for COVID-19 case predictions in Australia, Italy, and UK. Machine learning is a branch of computer science in which data teach algorithms, and the learning process is performed as supervised, unsupervised, and/or semi-supervised learning forms [24,25,26,27]. In this section, some approaches that were employed to predict cases (confirmed and deaths) of COVID-19 are discussed.

5.1. Logistic Function

A logistic function could be defined as follows:
f ( x ) = L 1 + e k ( x x 0 )
where e = Euler’s number, x 0 = Sigmoid’s midpoint, L is the curve’s maximum value, and K is the logistic growth of the curve.

5.2. Times Series Forecasting with the Prophet Algorithm

The Prophet algorithm is an open-source tool developed by Facebook’s Data Science team, which is aimed at business forecasting [28]. This algorithm works well with time-series data that have seasonal effects, and it is robust in dealing with missing data [29]. In the Prophet algorithm, the forecast is determined as follows [29]:
y T + h T ^ = y ¯ = ( y 1 + y 2 + y T ) / T
where y 1 , y 2 , , y T are denoted as historical data; and y T + h T ^ is a short-hand to forecast y T + h T based on available data.

6. Results

6.1. Analysis

6.1.1. New Cases

In this subsection, the confirmed growth rates in Australia, Italy, and UK were calculated for every day from 24 April to 23 May 2020. Figure 4 depicts the growth rate of confirmed cases in these countries. As can be seen, the growth rate for Australia remained below 0.5 during times of outbreak and was just above 0.0 at the end of May, while the rates for Italy and UK were generally high. The growth rate for UK was almost above 2.0 in April, and then dramatically declined in May. The rate for Italy fluctuated between 0.5 and 1.5 in April and May.
Figure 5 presents the growth rate of death cases for the above-mentioned countries, according to daily data from 24 April to 23 May 2020. The growth rate for death cases in Australia fluctuated between 0.0 and 7.0 in April and May and reached 7.0 at the end of April. During the same period, the rate remained almost below 2.0 in Italy, and in UK, the rate was just below 4.0 at the end of April and just above 0.0 at the end of May.

6.1.2. Overall Growth Rate

This section presents the numbers of active cases in these three countries, which were calculated using the following equation:
Active_cases = confirmed_cases − deaths_cases − recovered_cases
From Equation (13), the overall growth rate could be calculated according to Equation (14), in which i refers to the present day:
Overall growth rate [i] = ((active cases [i] − active case [i-1])/active case [i − 1]) × 100
Figure 6 illustrates the overall growth rate for confirmed cases in the studied countries. Negative numbers indicate that people recover faster than the rate at which they get sick, which is good news. The rate for Australia in the time period remained almost below zero and then changed from −15 at the end of April to just below −5 at the end of May. For Italy, the rate fluctuated between just above −7.5 and just above 0.0, while the rate for UK remained almost consistently positive in the time horizon (0.0–3.0). Figure 7 illustrates that the number of death cases in Australia is significantly lower than the other two countries.
Figure 8a–h shows confirmed cases versus death cases in each Australian state. By 23 May 2020, New South Wales and Northern Territory possessed the most confirmed and the least death cases in Australia, respectively. In Figure 8a–h, the number of confirmed and death cases in New South Wales significantly differed from other states in Australia and increased dramatically, while the Northern Territory experienced some fluctuation during the study time period.
With the aim of forecasting, the logistic function defined in Equation (11) was applied to the collected data (time horizon: start of outbreak in the countries). According to the results in Figure 9, Figure 10, Figure 11, Figure 12, Figure 13 and Figure 14, the logistic function is fitted to the trend of increasing cases to evaluate the performance of metric R2 scores used for confirmed and death cases. Results are presented in Table 1. The root mean square error (RMSE) was used as another metric to analyze cases, and the results in Table 2 show that the best RMSE value belongs to the Australian cases (confirmed and death).
Figure 15, Figure 16 and Figure 17 present the results of the classic SIR model. As previously mentioned, the controlling β parameter indicates the level of disease transmission, and γ is the recovery (removal) period indicating how many people could recover in a certain period. First, all parameters were initially added to the SIR model, which was then applied to real data. As can be seen in Figure 15, Figure 16 and Figure 17 and Table 3 (RMSE values), the classic SIR form was not suitable for predicting the COVID-19 pandemic in these three countries. In order to fit the SIR model to Australia, Italy, and UK, an optimizer was needed to find the unknown parameters ( β and γ ) from equation R 0 ( R 0 = β γ ) since these parameters could be estimated. Prior to the outbreak in these countries, it is essential to address whether the number of susceptible cases is equal to the population due to the absence of antibodies and vaccines for the disease. At first, R 0 = 2.7 was fixed (reported by Australian Government: Department of Health) as the median number, β = 0.378 , and γ = 0.14 .
Real data were applied to estimate the values of β and γ . An optimizer was used to find the best estimation of β and γ . The optimization algorithms used were the Broyden–Fletcher–Goldfarb–Shanno (BFGS) [30], limited memory bound constrained BFGS ( L-BFGS-B) [31], conjugate gradients (CG) [30], and Nelder–Mead algorithms [32]. The parameter settings are provided in Table 3. The flowchart of the improved SIR and SEIQR versions and parameter settings for the above-mentioned algorithms are addressed in Figure 18 and Table 4, respectively.
Table 5 provides the optimized values obtained by different algorithms (SIR model). The best values for the parameters were found using the Nelder–Mead algorithm (for the SIR model) and L-BFGS-B algorithm (for the SEIQR model). This method is illustrated in Figure 18. As previously mentioned, before the start of the outbreak, the number of susceptible cases was equal to the populations of these countries, since neither antibodies nor a developed vaccine were available. According to Wikipedia, the populations of Australia, Italy, and UK were 2506, 6006, and 6706, respectively. Table 6 illustrates the RMSE values obtained by the algorithms (for SIR and SEIQR models), showing improvements in significantly reducing the values. Figure 19a–c presents the confirmed cases provided by the optimized SEIQR model with the above-mentioned descriptions in the three countries (see Figure 18).
Figure 20, Figure 21 and Figure 22 reveal the forecasted values obtained using the Prophet algorithm, where the algorithm is fitted for the cases of Italy and UK but has errors for Australia. Table 7, Table 8 and Table 9 present the results of the predicted cumulative confirmed cases using the Prophet algorithm in the three countries, where y represents the true values of confirmed cases, ds is time, y ^ is the forecasted values, and y l o w e r ^ and y u p p e r ^ are the lower and upper bounds for the forecasted values, respectively. It should be noted that the forecasted values were determined between the cutoff and cutoff + horizon. Table 7, Table 8 and Table 9 are also cross-validation matrices that are used to find the error values between y and y ^ , from which the RMSE values can be obtained (Figure 23a–c).

7. Discussion and Conclusions

COVID-19 is a family of coronaviruses that has affected the lives of billions of people worldwide. The first section of this paper presented a short analysis of COVID-19, focusing on its effect in Australia, Italy, and UK. Specifically, the analysis gives a comparison of the confirmed cases and death rates between Australia, Italy, and UK and among the different states of Australia. The analysis reveals that Australia is in a generally good position compared with the other two countries. However, the situation in different regions of Australia is rather complicated. For example, New South Wales has the most confirmed cases and death cases, while Northern Territory shows the least confirmed and death cases (it is worth mentioning that New South Wales has a larger population).
Mathematical approaches based on SIR and SEIQR were proposed to predict the epidemiology in Australia, Italy, and UK. Since the classic forms of SIR and SEIQR are deterministic, an improved version based on parameter optimization is suggested to improve the prediction. The results were compared with the logistic function and Prophet algorithm, and are summarized as follows:
  • The comparison between the classic SIR model and real data showed a significant gap. However, initializing the parameters of the SIR model significantly improved the prediction.
  • The classic SIR model worked best for UK but was not suitable for Australia based on RMSE values.
  • The logistic function was a good model for UK with an R2 score of 0.97, while the scores for Australia and Italy were 0.67 and 0.95, respectively.
  • The best RMSE value belonged to the Australian cases (confirmed and deaths).
  • Parameter optimization for the SIR and SEIQR models significantly improved their prediction accuracy.
  • The improved version of SEIQR exhibited better performance than the SIR model (regarding RMSE values and figures).
  • The optimized SEIQR model has better prediction for UK and Italy compared with Australia.
  • The best values for the parameters were determined using the Nelder–Mead algorithm for the SIR model and the L-BFGS-B algorithm for the SEIQR model.
  • The Prophet algorithm worked better for Italy and UK cases than for Australian cases.
  • The logistic function had a better performance for cases in all three countries compared with the Prophet algorithm.
  • The improved versions of the SIR and SEIQR models exhibited a better performance than the logistic function, Prophet algorithm, and classic SIR model.
Some studies and research on related viruses have predicted that COVID-19 is dependent on environmental characteristics and will decline with higher temperature, humidity, ultraviolet (UV) light [33], and with spatial colony-growth heterogeneity [34]. Since UV light has been strongly associated with lower COVID-19 growth, projections suggest that, without intervention, COVID-19 will decrease temporarily during summer, rebound by autumn, and peak the subsequent winter. Regarding the above-mentioned discussion, the growth rate appears to be a country-specific characteristic.
The evolution of COVID-19 throughout the world is difficult to predict. Until a reliable vaccine becomes available for all, which may only happen by the end of 2021, governments will have to strike the tough balance between health and other issues, such as economic and social. Although social distancing costs an economic and psychological price, recent experience in several countries indicates that lifting a majority of restrictions increases the potential for multiple local outbreaks (second and third waves). In the absence of an effective and reliable vaccine, preparedness for this “wave” phenomenon is absolutely required.
One limitation of this study is that the authors did not account for human behavior or control measures in the models. By modeling the maximum growth rate and using a threshold number of cases, we could restrict the analyses to the period during which the disease expanded quickly: between the beginning of community transmission and the implementation of major control measures. This aspect alone is a suggested direction for further study.
Another limitation of this paper is the untimely analysis, which was an analysis based on data retrieved up until May 2020. If we look at the current data, the trend of COVID-19 cases is completely different from the predictions made in this paper. However, the results of this paper show that the growth rate is a country-specific characteristic and depends on time. Some factors relevant to these country specifications are the country’s size, population heterogeneity factor, etc. A similar effect of heterogeneous subpopulations is that, for instance, they are known for spreading bacterial colonies. Furthermore, there exist much more advanced models for the COVID-19 pandemic, whereby the inclusion of hospitalization rates, use of intensive care units (ICUs), and age groups is influential (vaccinations will be included in this list in the near future). All the above-mentioned issues are encouraged for future studies.
In addition, all the forecasting in this paper was addressed without considering the scenario of social distancing and quarantine, which is valuable as a future research direction. While this paper analyzes the improved SIR and SEIQR models, it would be interesting to test other epidemiology models. Moreover, it would be worthwhile to combine mathematical models with other observations, such as policy interventions, human behavior, and constraints, which may yield better prediction performance.

Supplementary Materials

Author Contributions

Conceptualization, I.R. and A.H.G.; methodology, I.R.; software, I.R.; validation, I.R., A.H.G., and F.C.; formal analysis, I.R.; investigation, I.R.; resources, I.R.; data curation, I.R.; writing—original draft preparation, I.R.; writing—review and editing, A.H.G., F.C., and P.G.A.; visualization, I.R.; supervision, A.H.G.; project administration, A.H.G.; funding acquisition, A.H.G. All authors have read and agreed to the published version of the manuscript.

Funding

This research received no external funding.

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

The code datasets are freely available for research purposes in a Supplementary File.

Conflicts of Interest

The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.

References

  1. Aràndiga, F.; Baeza, A.; Cordero-Carrión, I.; Donat, R.; Martí, M.C.; Mulet, P.; Yanez, D.F. A Spatial-Temporal Model for the Evolution of the COVID-19 Pandemic in Spain Including Mobility. Mathematics 2020, 8, 1677. [Google Scholar] [CrossRef]
  2. Putra, S.; Mu’tamar, Z.K. Estimation of Parameters in the SIR Epidemic Model Using Particle Swarm Optimization. Am. J. Math. Comput. Model. 2019, 4, 83–93. [Google Scholar] [CrossRef]
  3. Mbuvha, R.R.; Marwala, T. On Data-Driven Management of the COVID-19 Outbreak in South Africa. medRxiv 2020. [Google Scholar] [CrossRef]
  4. Qi, H.; Xiao, S.; Shi, R.; Ward, M.P.; Chen, Y.; Tu, W.; Su, Q.; Wang, W.; Wang, X.; Zhang, Z. COVID-19 transmission in Mainland China is associated with temperature and humidity: A time-series analysis. Sci. Total Environ. 2020, 728, 138778. [Google Scholar] [CrossRef]
  5. Asteris, P.G.; Douvika, M.G.; Karamani, C.A.; Skentou, A.D.; Chlichlia, K.; Cavaleri, L.; Daras, T.; Armaghani, D.J.; Zaoutis, T.E. A Novel Heuristic Algorithm for the Modeling and Risk Assessment of the COVID-19 Pandemic Phenomenon. Comput. Model. Eng. Sci. 2020, 125, 815–828. [Google Scholar] [CrossRef]
  6. Salgotra, R.; Gandomi, M.; Gandomi, A.H. Time Series Analysis and Forecast of the COVID-19 Pandemic in India using Genetic Programming. Chaos Solitons Fractals 2020, 138, 109945. [Google Scholar] [CrossRef] [PubMed]
  7. Rahimi, I.; Chen, F.; Gandomi, A.H. A review on COVID-19 forecasting models. Neural Comput. Appl. 2021, 1–11. [Google Scholar] [CrossRef]
  8. Reno, C.; Lenzi, J.; Navarra, A.; Barelli, E.; Gori, D.; Lanza, A.; Valentini, R.; Tang, B.; Fantini, M.P. Forecasting COVID-19-Associated Hospitalizations under Different Levels of Social Distancing in Lombardy and Emilia-Romagna, Northern Italy: Results from an Extended SEIR Compartmental Model. J. Clin. Med. 2020, 9, 1492. [Google Scholar] [CrossRef] [PubMed]
  9. Santosh, K.C. COVID-19 Prediction Models and Unexploited Data. J. Med. Syst. 2020, 44, 1–4. [Google Scholar] [CrossRef]
  10. Putra, M.; Kesavan, M.M.; Brackney, K.; Hackney, D.N.; Roosa, M.K.M. Forecasting the impact of coronavirus disease during delivery hospitalization: An aid for resource utilization. Am. J. Obstet. Gynecol. MFM 2020, 2, 100127. [Google Scholar] [CrossRef] [PubMed]
  11. Nabi, K.N. Forecasting COVID-19 pandemic: A data-driven analysis. Chaos Solitons Fractals 2020, 139, 110046. [Google Scholar] [CrossRef]
  12. Fenga, L. Forecasting the COVID-19 Diffusion in Italy and the Related Occupancy of Intensive Care Units. J. Probab. Stat. 2021, 2021, 1–9. [Google Scholar] [CrossRef]
  13. Gaglione, D.; Braca, P.; Millefiori, L.M.; Soldi, G.; Forti, N.; Marano, S.; Willett, P.K.; Pattipati, K.R. Adaptive Bayesian Learning and Forecasting of Epidemic Evolution—Data Analysis of the COVID-19 Outbreak. IEEE Access 2020, 8, 175244–175264. [Google Scholar] [CrossRef]
  14. Berta, P.; Lovaglio, P.G.; Paruolo, P.; Verzillo, S. Real Time Forecasting of Covid-19 Intensive Care Units Demand; Publications Office of the European Union: Luxembourg, 2020. [Google Scholar]
  15. Dean, N.E.; Piontti, A.P.Y.; Madewell, Z.J.; Cummings, D.A.; Hitchings, M.D.; Joshi, K.; Kahn, R.; Vespignani, A.; Halloran, M.E.; Longini, I.M. Ensemble forecast modeling for the design of COVID-19 vaccine efficacy trials. Vaccine 2020, 38, 7213–7216. [Google Scholar] [CrossRef]
  16. Kane, P.B.; Moyer, H.; MacPherson, A.; Papenburg, J.; Ward, B.J.; Broomell, S.B.; Kimmelman, J. Expert Forecasts of COVID-19 Vaccine Development Timelines. J. Gen. Intern. Med. 2020, 35, 3753–3755. [Google Scholar] [CrossRef]
  17. Keeling, M.J.; Hill, E.M.; Gorsich, E.E.; Penman, B.; Guyver-Fletcher, G.; Holmes, A.; Leng, T.; McKimm, H.; Tamborrino, M.; Dyson, L.; et al. Predictions of COVID-19 dynamics in the UK: Short-term forecasting and analysis of potential exit strategies. PLoS Comput. Biol. 2021, 17, e1008619. [Google Scholar] [CrossRef] [PubMed]
  18. Brand, S.P.C.; Aziza, R.; Kombe, I.K.; Agoti, C.N.; Hilton, J.; Rock, K.S.; Parisi, A.; Nokes, D.J.; Keeling, M.J.; Barasa, E.W. Forecasting the scale of the COVID-19 epidemic in Kenya. MedRxiv 2020. [Google Scholar] [CrossRef]
  19. Massonnaud, C.; Roux, J.; Crépey, P. COVID-19: Forecasting short term hospital needs in France. medrxiv 2020. [Google Scholar] [CrossRef] [Green Version]
  20. Kermack, W.O.; McKendrick, A.G. Contributions to the mathematical theory of epidemics—II. The problem of endemicity. Bull. Math. Biol. 1991, 53, 57–87. [Google Scholar] [CrossRef]
  21. Capasso, V.; Serio, G. A generalization of the Kermack-McKendrick deterministic epidemic model. Math. Biosci. 1978, 42, 43–61. [Google Scholar] [CrossRef]
  22. Weiss, H.H. The SIR model and the foundations of public health. Mater. Mat. 2013, 2013, 1–17. [Google Scholar]
  23. Peng, L.; Yang, W.; Zhang, D.; Zhuge, C.; Hong, L. Epidemic analysis of COVID-19 in China by dynamical modeling. arXiv 2020, arXiv:2002.06563. [Google Scholar]
  24. Mitchell, T.M. Machine learning and data mining. Commun. ACM 1999, 42, 30–36. [Google Scholar] [CrossRef]
  25. Arkes, H.R. Overconfidence in Judgmental Forecasting. In Harvey J. Greenberg; Springer International Publishing: Berlin/Heidelberg, Germany, 2001; pp. 495–515. [Google Scholar]
  26. Armstrong, J.S. Standards and Practices for Forecasting. In Harvey J. Greenberg; Springer International Publishing: Berlin/Heidelberg, Germany, 2001; pp. 679–732. [Google Scholar]
  27. Maleki, M.; Mahmoudi, M.R.; Wraith, D.; Pho, K.-H. Time series modelling to forecast the confirmed and recovered cases of COVID-19. Travel Med. Infect. Dis. 2020, 37, 101742. [Google Scholar] [CrossRef]
  28. Taylor, S.J.; Letham, B. Forecasting at Scale. Am. Stat. 2018, 72, 37–45. [Google Scholar] [CrossRef]
  29. Ndiaye, B.M.; Tendeng, L.; Seck, D. Analysis of the COVID-19 pandemic by SIR model and machine learning technics for forecasting. arXiv 2020, arXiv:2004.01574v1. [Google Scholar]
  30. Chambers, L.G.; Fletcher, R. Practical Methods of Optimization. Math. Gaz. 2001, 85, 562. [Google Scholar] [CrossRef]
  31. Byrd, R.H.; Lu, P.; Nocedal, J.; Zhu, C. A Limited Memory Algorithm for Bound Constrained Optimization. SIAM J. Sci. Comput. 1995, 16, 1190–1208. [Google Scholar] [CrossRef]
  32. Nelder, J.A.; Mead, R. A Simplex Method for Function Minimization. Comput. J. 1965, 7, 308–313. [Google Scholar] [CrossRef]
  33. Merow, C.; Urban, M.C. Seasonality and uncertainty in global COVID-19 growth rates. Proc. Natl. Acad. Sci. USA 2020, 117, 27456–27464. [Google Scholar] [CrossRef] [PubMed]
  34. Kindler, O.; Pulkkinen, O.; Cherstvy, A.G.; Metzler, R. Burst statistics in an early biofilm quorum sensing model: The role of spatial colony-growth heterogeneity. Sci. Rep. 2019, 9, 1–19. [Google Scholar] [CrossRef] [PubMed] [Green Version]
Figure 1. Flowchart of the current research process.
Figure 1. Flowchart of the current research process.
Information 12 00109 g001
Figure 2. Susceptible, infected, and recovered (SIR) model.
Figure 2. Susceptible, infected, and recovered (SIR) model.
Information 12 00109 g002
Figure 3. Susceptible, exposed, infected, quarantined, and recovered (SEIQR) model diagram [23].
Figure 3. Susceptible, exposed, infected, quarantined, and recovered (SEIQR) model diagram [23].
Information 12 00109 g003
Figure 4. Growth rate (confirmed cases in Australia, Italy, and UK).
Figure 4. Growth rate (confirmed cases in Australia, Italy, and UK).
Information 12 00109 g004
Figure 5. Growth rate (death cases in Australia, Italy, and UK).
Figure 5. Growth rate (death cases in Australia, Italy, and UK).
Information 12 00109 g005
Figure 6. Overall growth rate for confirmed cases in Australia, Italy, and UK.
Figure 6. Overall growth rate for confirmed cases in Australia, Italy, and UK.
Information 12 00109 g006
Figure 7. Number of death cases in Australia compared with Italy and UK.
Figure 7. Number of death cases in Australia compared with Italy and UK.
Information 12 00109 g007
Figure 8. Confirmed versus death cases in different Australian states: (a) New South Wales, (b) Australian Capital Territory, (c) Northern Territory, (d) Queensland, (e) South Australia, (f) Tasmania, (g) Victoria, (h) Western Australia.
Figure 8. Confirmed versus death cases in different Australian states: (a) New South Wales, (b) Australian Capital Territory, (c) Northern Territory, (d) Queensland, (e) South Australia, (f) Tasmania, (g) Victoria, (h) Western Australia.
Information 12 00109 g008aInformation 12 00109 g008bInformation 12 00109 g008cInformation 12 00109 g008d
Figure 9. Prediction of confirmed cases by logistic function (Australia).
Figure 9. Prediction of confirmed cases by logistic function (Australia).
Information 12 00109 g009
Figure 10. Prediction of death cases by logistic function (Australia).
Figure 10. Prediction of death cases by logistic function (Australia).
Information 12 00109 g010
Figure 11. Prediction of confirmed cases by logistic function (UK).
Figure 11. Prediction of confirmed cases by logistic function (UK).
Information 12 00109 g011
Figure 12. Prediction of death cases by logistic function (UK).
Figure 12. Prediction of death cases by logistic function (UK).
Information 12 00109 g012
Figure 13. Prediction of confirmed cases by logistic function (Italy).
Figure 13. Prediction of confirmed cases by logistic function (Italy).
Information 12 00109 g013
Figure 14. Prediction of death cases by logistic function (Italy).
Figure 14. Prediction of death cases by logistic function (Italy).
Information 12 00109 g014
Figure 15. Predicted cases in Australia using the susceptible, infected, recovered (SIR) model (blue: real confirmed cases; red: SIR model).
Figure 15. Predicted cases in Australia using the susceptible, infected, recovered (SIR) model (blue: real confirmed cases; red: SIR model).
Information 12 00109 g015
Figure 16. Predicted cases in Italy based on the SIR model (blue: real confirmed cases; red: SIR model).
Figure 16. Predicted cases in Italy based on the SIR model (blue: real confirmed cases; red: SIR model).
Information 12 00109 g016
Figure 17. Predicted cases in UK based on the SIR model (blue: real confirmed cases; red: SIR model).
Figure 17. Predicted cases in UK based on the SIR model (blue: real confirmed cases; red: SIR model).
Information 12 00109 g017
Figure 18. Flowchart of improved versions of SIR and SEIQR models.
Figure 18. Flowchart of improved versions of SIR and SEIQR models.
Information 12 00109 g018
Figure 19. Prediction by optimized SEIQR model for: (a) Australia, (b) Italy, and (c) UK.
Figure 19. Prediction by optimized SEIQR model for: (a) Australia, (b) Italy, and (c) UK.
Information 12 00109 g019
Figure 20. Forecasting by Prophet algorithm for the next year (confirmed cases in Australia).
Figure 20. Forecasting by Prophet algorithm for the next year (confirmed cases in Australia).
Information 12 00109 g020
Figure 21. Forecasting by Prophet algorithm for the next year (confirmed cases in Italy).
Figure 21. Forecasting by Prophet algorithm for the next year (confirmed cases in Italy).
Information 12 00109 g021
Figure 22. Forecasting by Prophet algorithm for the next year (confirmed cases in UK).
Figure 22. Forecasting by Prophet algorithm for the next year (confirmed cases in UK).
Information 12 00109 g022
Figure 23. Visualization of performance metric for Prophet algorithm (considering RMSE) for (a) UK, (b) Australia, and (c) Italy.
Figure 23. Visualization of performance metric for Prophet algorithm (considering RMSE) for (a) UK, (b) Australia, and (c) Italy.
Information 12 00109 g023aInformation 12 00109 g023b
Table 1. R2 score for different cases in the three countries.
Table 1. R2 score for different cases in the three countries.
CountryConfirmed CasesDeath Cases
Australia0.870.67
UK0.920.97
Italy0.930.95
Table 2. Root mean square error (RMSE) values for different cases in the three countries.
Table 2. Root mean square error (RMSE) values for different cases in the three countries.
CountryConfirmed CasesDeath Cases
Australia8.220.88
UK21.946.97
Italy23.248.00
Table 3. RMSE values obtained by SIR model (before optimization of parameters).
Table 3. RMSE values obtained by SIR model (before optimization of parameters).
ItalyUKAustralia
18.7515.45831.84
Table 4. Parameter settings.
Table 4. Parameter settings.
AlgorithmParameter Setting
BFGSMaxit = 100, reltol * = 10−8
Nelder–MeadMaxit = 500, reltol = 10−8, alpha = 1, beta = 0.5, gamma = 2.0
L-BFGS-BMaxit = 100, reltol = 10−8, lmm ** = 5, factr *** = 107
CGMaxit = 100, reltol = 10−8
* Reltol = Relative convergence tolerance, ** lmm = number of BFGS updates retained, *** factr = convergence factor, e = Euler’s number.
Table 5. Median values of SIR parameters determined by the Department of Health in each country.
Table 5. Median values of SIR parameters determined by the Department of Health in each country.
Country β γ R 0
AlgorithmBFGS Nelder–MeadL-BFGS-BCGBFGSNelder–MeadL-BFGS-BCGBFGSNelder–MeadL-BFGS-BCG
Australia0.0140.0140.3780.370.220.220.140.14 0.0630.0632.642.64
UK0.373.84701−30.370.370.141.94−10.140.142.640.022.642.64
Italy0.371.083555−3 0.370.370.143.9088−10.140.372.640.012.642.64
Table 6. RMSE values obtained based on the improved SIR model considering a 0.99 confidence interval.
Table 6. RMSE values obtained based on the improved SIR model considering a 0.99 confidence interval.
ModelItalyUKAustralia
SIR model1.411.011.13
SEIR model1.121.231.04
Table 7. Predicted cumulative confirmed cases in Australia (cross-validation matrix).
Table 7. Predicted cumulative confirmed cases in Australia (cross-validation matrix).
yds y ^ y l o w e r ^ y u p p e r ^ Cutoff
709521 May 202021,309.75218,998.14023,829.9554 April 2020
709922 May 202021,630.70819,245.07224,269.9044 April 2020
711423 May 202021,959.98519,424.09724,640.9394 April 2020
711424 May 202022,326.68819,766.19425,093.3534 April 2020
Table 8. Predicted cumulative confirmed cases in UK (cross-validation matrix).
Table 8. Predicted cumulative confirmed cases in UK (cross-validation matrix).
yds y ^ y l o w e r ^ y u p p e r ^ Cutoff
252,24621 May 2020143,776.53126,702.28162,413.934 April 2020
255,54422 May 2020146,462.83128,526.68165,539.804 April 2020
258,50423 May 2020148,818.88130,813.85168,216.414 April 2020
260,91624 May 2020150,344.39131,476.87170,004.004 April 2020
Table 9. Predicted cumulative confirmed cases in Italy (cross-validation matrix).
Table 9. Predicted cumulative confirmed cases in Italy (cross-validation matrix).
yds y ^ y l o w e r ^ y u p p e r ^ Cutoff
228,00621 May 2020373,982.5336,940.1415,612.74 April 2020
228,65822 May 2020379,300.7340,862.6422,338.44 April 2020
229,32723 May 2020384,792.4344,957.8429,120.34 April 2020
229,85824 May 2020390,481.8349,482.8436,663.24 April 2020
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Rahimi, I.; Gandomi, A.H.; Asteris, P.G.; Chen, F. Analysis and Prediction of COVID-19 Using SIR, SEIQR, and Machine Learning Models: Australia, Italy, and UK Cases. Information 2021, 12, 109. https://0-doi-org.brum.beds.ac.uk/10.3390/info12030109

AMA Style

Rahimi I, Gandomi AH, Asteris PG, Chen F. Analysis and Prediction of COVID-19 Using SIR, SEIQR, and Machine Learning Models: Australia, Italy, and UK Cases. Information. 2021; 12(3):109. https://0-doi-org.brum.beds.ac.uk/10.3390/info12030109

Chicago/Turabian Style

Rahimi, Iman, Amir H. Gandomi, Panagiotis G. Asteris, and Fang Chen. 2021. "Analysis and Prediction of COVID-19 Using SIR, SEIQR, and Machine Learning Models: Australia, Italy, and UK Cases" Information 12, no. 3: 109. https://0-doi-org.brum.beds.ac.uk/10.3390/info12030109

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop