Dataset Information

Comparison of ARIMA model and XGBoost model for prediction of human brucellosis in mainland China: a time-series study.

ABSTRACT: Objectives: Human brucellosis is a public health problem endangering health and property in China. Predicting the trend and the seasonality of human brucellosis is of great significance for its prevention. In this study, a comparison between the autoregressive integrated moving average (ARIMA) model and the eXtreme Gradient Boosting (XGBoost) model was conducted to determine which was more suitable for predicting the occurrence of brucellosis in mainland China.

Design: Time-series study.

Setting: Mainland China.

Methods: Data on human brucellosis in mainland China were provided by the National Health and Family Planning Commission of China. The data were divided into a training set and a test set. The training set was composed of the monthly incidence of human brucellosis in mainland China from January 2008 to June 2018, and the test set was composed of the monthly incidence from July 2018 to June 2019. The mean absolute error (MAE), root mean square error (RMSE) and mean absolute percentage error (MAPE) were used to evaluate the effects of model fitting and prediction.

Results: The number of human brucellosis patients in mainland China increased from 30?002 in 2008 to 40?328 in 2018. There was an increasing trend and obvious seasonal distribution in the original time series. For the training set, the MAE, RSME and MAPE of the ARIMA(0,1,1)×(0,1,1)₁₂ model were 338.867, 450.223 and 10.323, respectively, and the MAE, RSME and MAPE of the XGBoost model were 189.332, 262.458 and 4.475, respectively. For the test set, the MAE, RSME and MAPE of the ARIMA(0,1,1)×(0,1,1)₁₂ model were 529.406, 586.059 and 17.676, respectively, and the MAE, RSME and MAPE of the XGBoost model were 249.307, 280.645 and 7.643, respectively.

Conclusions: The performance of the XGBoost model was better than that of the ARIMA model. The XGBoost model is more suitable for prediction cases of human brucellosis in mainland China.

SUBMITTER: Alim M

PROVIDER: S-EPMC7722837 | biostudies-literature | 2020 Dec

REPOSITORIES: biostudies-literature

ACCESS DATA

Publications

Comparison of ARIMA model and XGBoost model for prediction of human brucellosis in mainland China: a time-series study.

Alim Mirxat M Ye Guo-Hua GH Guan Peng P Huang De-Sheng DS Zhou Bao-Sen BS Wu Wei W

BMJ open 20201207 12

<h4>Objectives</h4>Human brucellosis is a public health problem endangering health and property in China. Predicting the trend and the seasonality of human brucellosis is of great significance for its prevention. In this study, a comparison between the autoregressive integrated moving average (ARIMA) model and the eXtreme Gradient Boosting (XGBoost) model was conducted to determine which was more suitable for predicting the occurrence of brucellosis in mainland China.<h4>Design</h4>Time-series s ...[more]

PMID: 33293308

Similar Datasets

Project description:BackgroundHemorrhagic fever with renal syndrome (HFRS) is still attracting public attention because of its outbreak in various cities in China. Predicting future outbreaks or epidemics disease based on past incidence data can help health departments take targeted measures to prevent diseases in advance. In this study, we propose a multistep prediction strategy based on extreme gradient boosting (XGBoost) for HFRS as an extension of the one-step prediction model. Moreover, the fitting and prediction accuracy of the XGBoost model will be compared with the autoregressive integrated moving average (ARIMA) model by different evaluation indicators.MethodsWe collected HFRS incidence data from 2004 to 2018 of mainland China. The data from 2004 to 2017 were divided into training sets to establish the seasonal ARIMA model and XGBoost model, while the 2018 data were used to test the prediction performance. In the multistep XGBoost forecasting model, one-hot encoding was used to handle seasonal features. Furthermore, a series of evaluation indices were performed to evaluate the accuracy of the multistep forecast XGBoost model.ResultsThere were 200,237 HFRS cases in China from 2004 to 2018. A long-term downward trend and bimodal seasonality were identified in the original time series. According to the minimum corrected akaike information criterion (CAIC) value, the optimal ARIMA (3, 1, 0) × (1, 1, 0)12 model is selected. The index ME, RMSE, MAE, MPE, MAPE, and MASE indices of the XGBoost model were higher than those of the ARIMA model in the fitting part, whereas the RMSE of the XGBoost model was lower. The prediction performance evaluation indicators (MAE, MPE, MAPE, RMSE and MASE) of the one-step prediction and multistep prediction XGBoost model were all notably lower than those of the ARIMA model.ConclusionsThe multistep XGBoost prediction model showed a much better prediction accuracy and model stability than the multistep ARIMA prediction model. The XGBoost model performed better in predicting complicated and nonlinear data like HFRS. Additionally, Multistep prediction models are more practical than one-step prediction models in forecasting infectious diseases.

Project description:BackgroundGonorrhea has long been a serious public health problem in mainland China that requires attention, modeling to describe and predict its prevalence patterns can help the government to develop more scientific interventions.MethodsTime series (TS) data of the gonorrhea incidence in China from January 2004 to August 2022 were collected, with the incidence data from September 2021 to August 2022 as the validation. The seasonal autoregressive integrated moving average (SARIMA) model, long short-term memory network (LSTM) model, and hybrid SARIMA-LSTM model were used to simulate the data respectively, the model performance were evaluated by calculating the mean absolute percentage error (MAPE), root mean square error (RMSE), and mean absolute error (MAE) of the training and validation sets of the models.ResultsThe Seasonal components after data decomposition showed an approximate bimodal distribution with a period of 12 months. The three models identified were SARIMA(1,1,1) (2,1,2)12, LSTM with 150 hidden units, and SARIMA-LSTM with 150 hidden units, the SARIMA-LSTM model fitted best in the training and validation sets, for the smallest MAPE, RMSE, and MPE.ConclusionsThe overall incidence trend of gonorrhea in mainland China has been on the decline since 2004, with some periods exhibiting an upward trend. The incidence of gonorrhea displays a seasonal distribution, typically peaking in July and December each year. The SARIMA model, LSTM model, and SARIMA-LSTM model can all fit the monthly incidence time series data of gonorrhea in mainland China. However, in terms of predictive performance, the SARIMA-LSTM model outperforms the SARIMA and LSTM models, with the LSTM model surpassing the SARIMA model. This suggests that the SARIMA-LSTM model can serve as a preferred tool for time series analysis, providing evidence for the government to predict trends in gonorrhea incidence. The model's predictions indicate that the incidence of gonorrhea in mainland China will remain at a high level in 2024, necessitating that policymakers implement public health measures in advance to prevent the spread of the disease.

Project description:BackgroundEstablishing epidemiological models and conducting predictions seems to be useful for the prevention and control of human brucellosis. Autoregressive integrated moving average (ARIMA) models can capture the long-term trends and the periodic variations in time series. However, these models cannot handle the nonlinear trends correctly. Recurrent neural networks can address problems that involve nonlinear time series data. In this study, we intended to build prediction models for human brucellosis in mainland China with Elman and Jordan neural networks. The fitting and forecasting accuracy of the neural networks were compared with a traditional seasonal ARIMA model.MethodsThe reported human brucellosis cases were obtained from the website of the National Health and Family Planning Commission of China. The human brucellosis cases from January 2004 to December 2017 were assembled as monthly counts. The training set observed from January 2004 to December 2016 was used to build the seasonal ARIMA model, Elman and Jordan neural networks. The test set from January 2017 to December 2017 was used to test the forecast results. The root mean squared error (RMSE), mean absolute error (MAE) and mean absolute percentage error (MAPE) were used to assess the fitting and forecasting accuracy of the three models.ResultsThere were 52,868 cases of human brucellosis in Mainland China from January 2004 to December 2017. We observed a long-term upward trend and seasonal variance in the original time series. In the training set, the RMSE and MAE of Elman and Jordan neural networks were lower than those in the ARIMA model, whereas the MAPE of Elman and Jordan neural networks was slightly higher than that in the ARIMA model. In the test set, the RMSE, MAE and MAPE of Elman and Jordan neural networks were far lower than those in the ARIMA model.ConclusionsThe Elman and Jordan recurrent neural networks achieved much higher forecasting accuracy. These models are more suitable for forecasting nonlinear time series data, such as human brucellosis than the traditional ARIMA model.

Project description:BACKGROUND:Despite the considerable efforts made to address the issue of brucellosis worldwide, its prevalence in dairy products continues to be difficult to estimate and represents a key public health issue around the world today. The aim of the present study was to better understand the epidemiology of this disease in mainland China. We set out to investigate the yearly spatial distribution and possible hotspots of the disease. METHODS:Human brucellosis data from mainland China between 2007 and 2016 were collected from the China Information System for Disease Control and Prevention. A geographic information system ArcGIS10.3 (ESRI, Redlands) was used to identify potential changes in the spatial and temporal distribution of human brucellosis in mainland China during the study period. These distributions were evaluated using three-dimensional trend analysis and spatial autocorrelation analyse. A gravity-center was used to analyse the migration track of human brucellosis. RESULTS:A total of 399,578 cases of human brucellosis were reported during the 10-year study period. The monthly incidence of brucellosis in China demonstrates clear seasonality. Spring and summer are the peak seasons, while May is the peak month for brucellosis. Three-dimensional trend analysis suggests that brucellosis is on the rise from south to north, and that the epidemic situation in northern China is more severe. Between 2007 and 2016, the overall migration distance of the brucellosis incidence gravity-center was 906.43?km, and the direction was southwest. However, the overall gravity center of brucellosis was still in the northern part of China. In the global autocorrelation analysis, brucellosis in China demonstrated a non-random distribution between 2013 and 2014, with spatial autocorrelation (Z?>?1.96, P?<?0.05) and a clustering trend, while no clustering trend was found from 2007 to 2012 or from 2015 to 2016. In the local autocorrelation analysis, a Low-Low cluster phenomenon was found in the south of China in 2013 and 2014. CONCLUSION:Human brucellosis remains a widespread challenge, particularly in northern China. The hotspots highlight potential high-risk areas which may require special plans and resources for monitoring and controlling the disease.

Dataset Information

Comparison of ARIMA model and XGBoost model for prediction of human brucellosis in mainland China: a time-series study.

Publications

Comparison of ARIMA model and XGBoost model for prediction of human brucellosis in mainland China: a time-series study.

Similar Datasets

OmicsDI is part of the ELIXIR infrastructure

Tweets