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.

Time-dependent heterogeneity leads to transient suppression of the COVID-19 epidemic, not herd immunity

Heterogeneity and Superspreading Effect on Herd Immunity

Monitoring Real-Time Transmission Heterogeneity from Incidence Data

Modelling the Effect of Human Heterogeneity on Infectious Disease Transmission Dynamics

Super-Spreaders Out, Super-Spreading In: The Effects of Infectiousness Heterogeneity and Lockdowns on Herd Immunity

Population heterogeneity in vaccine coverage impacts epidemic thresholds and bifurcation dynamics

Endemic infectious states below the epidemic threshold and beyond herd immunity

Individual dynamics and local heterogeneity provide a microscopic view of the epidemic spreading

The Complex Interplay Between Risk Tolerance and the Spread of Infectious Diseases

Adaptive human behavior and delays in information availability autonomously modulate epidemic waves

Spatial Heterogeneity Can Lead to Substantial Local Variations in COVID-19 Timing and Severity

A Hybrid Epidemic Model to Explore Stochasticity in COVID-19 Dynamics

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

COVID-19: What If Immunity Wanes?

Spatial Heterogeneity and Infection Patterns on Epidemic Transmission Disclosed by a Combined Contact-Dependent Dynamics and Compartmental Model.

Dynamic heterogeneity in COVID-19: Insights from a mathematical model

Modeling Incorporating the Severity-Reducing Long-term Immunity: Higher Viral Transmission Paradoxically Reduces Severe COVID-19 During Endemic Transition

A Polarized Temporal Network Model to Study the Spread of Recurrent Epidemic Diseases in a Partially Vaccinated Population

Dynamics of epidemic diseases without guaranteed immunity

Data-driven modeling reveals a universal dynamic underlying the COVID-19 pandemic under social distancing

Diverse local epidemics reveal the distinct effects of population density, demographics, climate, depletion of susceptibles, and intervention in the first wave of COVID-19 in the United States