Abstract:Accurate prediction of epidemics is pivotal for making well-informed decisions for the control of infectious diseases, but addressing heterogeneity in the system poses a challenge. In this study, we propose a novel modeling framework integrating the spatio-temporal heterogeneity of susceptible individuals into homogeneous models, by introducing a continuous recruitment process for the susceptibles. A neural network approximates the recruitment rate to develop a Universal Differential Equations (UDE) model. Simultaneously, we pre-set a specific form for the recruitment rate and develop a mechanistic model. Data from a COVID Omicron variant outbreak in Shanghai are used to train the UDE model using deep learning methods and to calibrate the mechanistic model using MCMC methods. Subsequently, we project the attack rate and peak of new infections for the first Omicron wave in China after the adjustment of the dynamic zero-COVID policy. Our projections indicate an attack rate and a peak of new infections of 80.06% and 3.17% of the population, respectively, compared with the homogeneous model's projections of 99.97% and 32.78%, thus providing an 18.6% improvement in the prediction accuracy based on the actual data. Our simulations demonstrate that heterogeneity in the susceptibles decreases herd immunity for ~37.36% of the population and prolongs the outbreak period from ~30 days to ~70 days, also aligning with the real case. We consider that this study lays the groundwork for the development of a new class of models and new insights for modelling heterogeneity. This study developed a novel modeling framework that explicitly integrates spatial heterogeneity into traditionally homogeneous compartmental models for precise and efficient prediction of epidemics of infectious diseases. We employed the deep learning method to learn the recruitment rate of susceptible individuals from data given its inherent complexity and limited information. Moreover, we predefined the mechanistic form of the recruitment rate to facilitate interpretation and scenario analysis, where we fitted the recruitment rate of mechanistic model to those obtained from deep learning-based approach. Compared to the homogeneous model, our mechanistic model predicts significantly lower peak and final sizes of the epidemic, as well as reduced levels of herd immunity. Using real data as a benchmark, our model's predictive accuracy has markedly improved. The framework of our mechanistic model employs a straightforward approach to integrate heterogeneity into compartment models, thereby facilitating the extension of the modelling framework to encompass various aspects of infectious disease transmission with spatio-temporal heterogeneity among the susceptible population.

A unified framework of epidemic spreading prediction by empirical mode decomposition based ensemble learning techniques

Forecasting the Long-Term Trend of COVID-19 Epidemic Using a Dynamic Model

Real-Time Forecasting of Hand-Foot-and-Mouth Disease Outbreaks using the Integrating Compartment Model and Assimilation Filtering

An adaptive weight ensemble approach to forecast influenza activity in an irregular seasonality context

Epidemic Control Based on Fused Body Sensed and Social Network Information

Epidemic prediction of dengue fever based on vector compartment model and Markov chain Monte Carlo method

Inter- and Intra-Series Embeddings Fusion Network for Epidemiological Forecasting

A swarm-optimizer-assisted simulation and prediction model for emerging infectious diseases based on SEIR

Flexible Modeling of Epidemics with an Empirical Bayes Framework

Ensemble Prediction Method Based on Decomposition–Reconstitution–Integration for COVID-19 Outbreak Prediction

Bayesian inference for the onset time and epidemiological characteristics of emerging infectious diseases

Interdisciplinary modelling and forecasting of dengue

Prediction of an epidemic spread based on the adaptive genetic algorithm

Evaluation of models for multi-step forecasting of hand, foot and mouth disease using multi-input multi-output: A case study of Chengdu, China

Ensemble Approaches for Robust and Generalizable Short-Term Forecasts of Dengue Fever. A retrospective and prospective evaluation study in over 180 locations around the world

Data assimilation method for improving the global spatiotemporal predictions of epidemic dynamics yielded by an ensemble Kalman filter and Metropolis–Hastings sampling

Ensemble learning methods of inference for spatially stratified infectious disease systems

Forecasting Epidemics Through Nonparametric Estimation of Time-Dependent Transmission Rates Using the SEIR Model

Ensemble forecasting for disease outbreak detection

SEIARN: Intelligent Early Warning Model of Epidemic Spread Based on LSTM Trajectory Prediction

Managing spatio-temporal heterogeneity of susceptibles by embedding it into an homogeneous model: A mechanistic and deep learning study