Application of a data-driven XGBoost model for the prediction of COVID-19 in the USA: a time-series study
Zheng-Gang Fang1, Shu-Qin Yang1, Cai-Xia Lv1
1Department of Epidemiology, China Medical University, Shenyang, China.
Objective:
The COVID-19 outbreak was first reported in Wuhan, China, and has been acknowledged as a pandemic due to its rapid spread worldwide. Predicting the trend of COVID-19 is of great significance for its prevention. A comparison between the autoregressive integrated moving average (ARIMA) model and the eXtreme Gradient Boosting (XGBoost) model was conducted to determine which was more accurate for anticipating the occurrence of COVID-19 in the USA.
Design:
Time-series study.
Setting:
The USA was the setting for this study.
Main Outcome Measures:
Three accuracy metrics, mean absolute error (MAE), root mean square error (RMSE) and mean absolute percentage error (MAPE), were applied to evaluate the performance of the two models.
Results:
In our study, for the training set and the validation set, the MAE, RMSE and MAPE of the XGBoost model were less than those of the ARIMA model.
Conclusions:
The XGBoost model can help improve prediction of COVID-19 cases in the USA over the ARIMA model.
More Related Videos
Related Concept Videos
Residuals and Least-Squares Property
If the observed data point lies above the line, the residual is positive, and the line underestimates the actual data value for y. If the observed data point lies below the line, the residual is negative, and the line overestimates the actual data value for y.
The process of fitting the best-fit...
Prediction Intervals
However, the point estimate is most likely not the exact value of the population parameter, but close to it. After calculating point estimates, we construct interval estimates, called confidence intervals or prediction intervals. This prediction interval comprises a range of values unlike the point estimate and is a better predictor of the observed sample value, y.
Steps in Outbreak Investigation
Statistical Methods for Analyzing Epidemiological Data
End Point Prediction: Gran Plot
For potentiometric titration, the Gran plot is created by plotting...
Cancer Survival Analysis


