Learning Ecological and Epidemic Processes using Neural ODEs, Kolmogorov-Arnold Network ODEs and SINDy

Read full text
Authors: Maria Vasilyeva, Zheng Wei, Kelum Gajamannage, Hyangim Ji, Artem Krasnikov, Alexey L. Sadovski
Cited by: 1

Abstract

We consider epidemic and ecological models to investigate their coupled dynamics. Starting with the classical Susceptible-Infected-Recovered (SIR) model for basic epidemic behavior and the predator-prey (Lotka-Volterra, LV) system for ecological interactions, we then combine these frameworks into a coupled Lotka-Volterra-Susceptible-Infected-Susceptible (LVSIS) model. The resulting system consists of four differential equations describing the evolution of susceptible and infected prey and predator populations, incorporating ecological interactions, disease transmission, and spatial dispersal. To learn the underlying dynamics directly from data, we employ several data-driven modeling frameworks: Neural Ordinary Differential Equations (Neural ODEs), Kolmogorov-Arnold Network Ordinary Differential Equations (KANODEs), and Sparse Identification of Nonlinear Dynamics (SINDy). Numerical experiments based on synthetic data are conducted to investigate the learning ability of these models in capturing the epidemic and ecological behavior. We further extend our approach to spatio-temporal models, aiming to uncover hidden local couplings.

Fields of Study & Subjects

Model Reduction and Neural Networks COVID-19 epidemiological studies Mathematical and Theoretical Epidemiology and Ecology Models

Publication details

Journal / source
arXiv (Cornell University)
Publication date
14 January 2026
Document type
Preprint