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

Abstract

Abstract 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.

Keywords

• EEMD
• LSTM
• Hybrid model
• HFMD prediction