A Multivariate Deep Learning Model with Coupled Human Intervention Factors for COVID-19 Forecasting

ABSTRACT

Artificial intelligence (AI) technology plays a crucial role in infectious disease outbreak prediction and control. Many human interventions can influence the spread of epidemics, including government responses, quarantine, and economic support. However, most previous AI-based models have failed to consider human interventions when predicting the trend of infectious diseases. This study selected four human intervention factors that may affect COVID-19 transmission, examined their relationship to epidemic cases, and developed a multivariate long short-term memory network model (M-LSTM) incorporating human intervention factors. Firstly, we analyzed the correlations and lagged effects between four human factors and epidemic cases in three representative countries, and found that these four factors typically delayed the epidemic case data by approximately 15 days. On this basis, a multivariate epidemic prediction model (M-LSTM) was developed. The model prediction results show that coupling human intervention factors generally improves model performance, but adding certain intervention factors also results in lower performance. Overall, a multivariate deep learning model with coupled variable correlation and lag outperformed other comparative models, and thus validated its effectiveness in predicting infectious diseases.