A multivariate multi-step LSTM forecasting model for tuberculosis incidence with model explanation in Liaoning Province, China

A hybrid model for hand-foot-mouth disease prediction based on ARIMA-EEMD-LSTM

BACKGROUND

Hand, foot, and mouth disease (HFMD) is a common infectious disease that poses a serious threat to children all over the world. However, the current prediction models for HFMD still require improvement in accuracy. In this study, we proposed a hybrid model based on autoregressive integrated moving average (ARIMA), ensemble empirical mode decomposition (EEMD) and long short-term memory (LSTM) to predict the trend of HFMD.

METHODS

The data used in this study was sourced from the National Clinical Research Center for Child Health and Disorders, Chongqing, China. The daily reported incidence of HFMD from 1 January 2015 to 27 July 2023 was collected to develop an ARIMA-EEMD-LSTM hybrid model. ARIMA, LSTM, ARIMA-LSTM and EEMD-LSTM models were developed to compare with the proposed hybrid model. Root mean square error (RMSE), mean absolute error (MAE) and coefficient of determination (R2) were adopted to evaluate the performances of the prediction models.

RESULTS

Overall, ARIMA-EEMD-LSTM model achieved the most accurate prediction for HFMD, with RMSE, MAPE and R2 of 4.37, 2.94 and 0.996, respectively. Performing EEMD on the residual sequence yields 11 intrinsic mode functions. EEMD-LSTM model is the second best, with RMSE, MAPE and R2 of 6.20, 3.98 and 0.996.

CONCLUSION

Results showed the advantage of ARIMA-EEMD-LSTM model over the ARIMA model, the LSTM model, the ARIMA-LSTM model and the EEMD-LSTM model. For the prevention and control of epidemics, the proposed hybrid model may provide a more powerful help. Compared with other three models, the two integrated with EEMD method showed significant improvement in predictive capability, offering novel insights for modeling of disease time series.