Neural Network-Enhanced Disease Spread Dynamics Over Time and Space

Abstract

This study presents a neural network-enhanced approach to modeling disease spread dynamics over time and space. Neural networks are used to estimate time-varying parameters, with two calibration methods explored: Approximate Bayesian Computation (ABC) with Trust Region Reflective (TRF) optimization, and backpropagation with the Adam optimizer. Simulations show that the second method is faster for larger networks, while the first offers a greater diversity of acceptable solutions. The model is extended spatially by introducing a pathogen compartment, which diffuses through environmental transmission and interpersonal contact. We examine scenarios of exact reporting, overreporting, and underreporting, highlighting their effects on public behavior and infection peaks. Our results demonstrate that neural network-enhanced models more accurately capture dynamic changes in disease spread and illustrate the potential of scientific machine learning (SciML) to improve the predictive power of epidemiological models beyond traditional approaches.