Using a Simple Open-Source Automated Machine Learning Algorithm to Forecast COVID-19 Spread: A Modelling Study

Abstract

Machine learning algorithms have been used to develop prediction models in various infectious and non-infectious settings including interpretation of images in predicting the outcome of diseases. We demonstrate the application of one such simple automated machine learning algorithm to a dataset obtained about COVID-19 spread in South Korea to better understand the disease dynamics. Data from 20th January 2020 (when the first case of COVID-19 was detected in South Korea) to 4th March 2020 was accessed from Korea's centre for disease control (KCDC). A future time-series of specified length (taken as 7 days in our study) starting from 5th March 2020 to 11th March 2020 was generated and fed to the model to generate predictions with upper and lower trend bounds of 95% confidence intervals. The model was assessed for its ability to reliably forecast using mean absolute percentage error (MAPE) as the metric. As on 4th March 2020, 145,541 patients were tested for COVID-19 (in 45 days) in South Korea of which 5166 patients tested positive. The predicted values approximated well with the actual numbers. The difference between predicted and observed values ranged from 4.08% to 12.77% . On average, our predictions differed from actual values by 7.42% (MAPE) over the same period. Open source and automated machine learning tools like Prophet can be applied and are effective in the context of COVID-19 for forecasting spread in naïve communities. It may help countries to efficiently allocate healthcare resources to contain this pandemic.