Machine learning predictions of COVID-19 second wave end-times in Indian states

Abstract

The estimate of the remaining time of an ongoing wave of epidemic spreading is a critical issue. Due to the variations of a wide range of parameters in an epidemic, for simple models such as Susceptible-Infected-Removed (SIR) model, it is difficult to estimate such a time scale. On the other hand, multidimensional data with a large set attributes are precisely what one can use in statistical learning algorithms to make predictions. Here we show, how the predictability of the SIR model changes with various parameters using a supervised learning algorithm. We then estimate the condition in which the model gives the least error in predicting the duration of the first wave of the COVID-19 pandemic in different states in India. Finally, we use the SIR model with the above mentioned optimal conditions to generate a training data set and use it in the supervised learning algorithm to estimate the end-time of the ongoing second wave of the pandemic in different states in India.

arXiv comment8 pages, 6 figures, 1 table

References in corpus (7)

• Inferring change points in the COVID-19 spreading reveals the effectiveness of interventions
• Machine Learning Predicts Laboratory Earthquakes
• Real-time forecasts and risk assessment of novel coronavirus (COVID-19) cases: A data-driven analysis
• Machine learning plastic deformation of crystals
• Square lattice site percolation at increasing ranges of neighbor interactions
• SEIRD model to study the asymptomatic growth during COVID-19 pandemic in India
• What is the potential for a second peak in the evolution of SARS-CoV-2 in emerging and developing economies? Insights from a SIRASD model considering the informal economy