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 Deep Learning Enhanced Compartmental Model and its Application in Modeling Omicron Dynamics and Development in China

A Deep Learning-Enhanced Compartmental Model and Its Application in Modeling Omicron in China

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

Deep Epidemiological Modeling by Black-box Knowledge Distillation: An Accurate Deep Learning Model for COVID-19

Physics-Informed Neural Networks Integrating Compartmental Model for Analyzing COVID-19 Transmission Dynamics.

Epi-DNNs: Epidemiological Priors Informed Deep Neural Networks for Modeling COVID-19 Dynamics.

Transmission dynamics informed neural network with application to COVID-19 infections

Deep Dynamic Epidemiological Modelling for COVID-19 Forecasting in Multi-level Districts

A Comparative Study on Deep Learning Models for COVID-19 Forecast

RNN Enhanced Compartmental Model for Infectious Disease Prediction

Cooperation or Confrontation? —— Prediction of COVID-19 Pandemic Situation Via Deep Learning

Enhancing Online Epidemic Supervising System by Compartmental and GRU Fusion Model

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

Euler iteration augmented physics-informed neural networks for time-varying parameter estimation of the epidemic compartmental model

DDE: Deep Dynamic Epidemiological Modeling for Infectious Illness Development Forecasting in Multi-level Geographic Entities

A data-driven epidemic model with human mobility and vaccination protection for COVID-19 prediction

Predictive Modeling for Epidemic Outbreaks: A New Approach and COVID-19 Case Study

Time Series Modeling and Forecasting of Epidemic Spreading Processes Using Deep Transfer Learning

DDE: Deep Dynamic Epidemiological Modelling for Infectious Illness Development Forecasting in Multi-level Geographic Entities

NeuralCODE: Neural Compartmental Ordinary Differential Equations Model with AutoML for Interpretable Epidemic Forecasting

Long-term trend prediction of pandemic combining the compartmental and deep learning models