How (Not) to Hybridize Neural and Mechanistic Models for Epidemiological Forecasting
Not on the map. Why?
Abstract (arXiv)
Epidemiological forecasting from surveillance data is a hard problem and hybridizing mechanistic compartmental models with neural models is a natural direction. The mechanistic structure helps keep trajectories epidemiologically plausible, while neural components can capture non-stationary, data-adaptive effects. In practice, however, many seemingly straightforward couplings fail under partial observability and continually shifting transmission dynamics driven by behavior, waning immunity, seasonality, and interventions. We catalog these failure modes and show that robust performance requires making non-stationarity explicit: we extract multi-scale structure from the observed infection series and use it as an interpretable control signal for a controlled neural ODE coupled to an epidemiological model. Concretely, we decompose infections into trend, seasonal, and residual components and use these signals to drive continuous-time latent dynamics while jointly forecasting and inferring time-varying transmission, recovery, and immunity-loss rates. Across early outbreak and multi-wave regimes, our approach attains the lowest RMSE on five datasets (21-63% reduction over the strongest default-configured baseline), achieves the best peak detection accuracy, and infers time-varying epidemiological rates within ground-truth ranges, without relying on auxiliary covariates.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 17 of its 64 references that are in the map. First posted on arXiv 6 Feb 2026.
Builds on
17 of 64 references.
References · 17
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2018An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modelingcited by 1,447
- 2022Are Transformers Effective for Time Series Forecasting?cited by 1,400
- 2019Enhancing the Locality and Breaking the Memory Bottleneck of Transformer on Time Series Forecastingcited by 606
- 2020Neural Controlled Differential Equations for Irregular Time Seriescited by 501
- 2020Universal Differential Equations for Scientific Machine Learningcited by 345
- 2017Learning Koopman Invariant Subspaces for Dynamic Mode Decompositioncited by 156
- 2020Forecasting Sequential Data using Consistent Koopman Autoencoderscited by 114
- 2024TimeMixer++: A General Time Series Pattern Machine for Universal Predictive Analysiscited by 113
- 2024KAN-ODEs: Kolmogorov-Arnold Network Ordinary Differential Equations for Learning Dynamical Systems and Hidden Physicscited by 58
- 2021Fully Learnable Deep Wavelet Transform for Unsupervised Monitoring of High-Frequency Time Seriescited by 29
5 more
- 2025TimeKAN: KAN-based Frequency Decomposition Learning Architecture for Long-term Time Series Forecastingcited by 27
- 2022EINNs: Epidemiologically-informed Neural Networkscited by 21
- 2022First De-Trend then Attend: Rethinking Attention for Time-Series Forecastingcited by 19
- 2018Epidemiological data challenges: planning for a more robust future through data standardscited by 2
- 2018Forecasting Based on Surveillance Datacited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.
- 2026Toward World Models for Epidemiologycited by 3
- 2026Graph World Models for Constrained Epidemic Policy Planningcited by 0
- 2026Graph World Models for Constrained Epidemic Policy Planning28 Sep 2026
- 2026Toward World Models for Epidemiology10 Apr 2026