Skip to content

How (Not) to Hybridize Neural and Mechanistic Models for Epidemiological Forecasting

Feb 2026 · arXiv.org · Vol abs/2602.06323 · 1 citation · 65 references
Computer Science

TL;DR

This work decomposes infections into trend, seasonal, and residual components and uses these signals to drive continuous-time latent dynamics while jointly forecasting and inferring time-varying transmission, recovery, and immunity-loss rates.

Abstract

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.

View source

Similar papers

Open access Jul 2026

Physics-Informed Neural Networks (PINNs) for Parameter Estimation in the SIRS-D Epidemiological Model

Infectious diseases exhibit complex and rapidly evolving transmission dynamics, requiring modeling approaches that can accurately capture these mechanisms. The SIRS-D compartmental model provides a suitable framework, as it incorporates temporary immunity and disease-induced mortality within the epidemic process. Accurate parameter estimation is essential for quantifying the transmission rate, recovery rate, waning immunity rate, and mortality rate, which collectively govern the system behavior. Among existing estimation methods, Physics-Informed Neural Networks (PINNs) offer significant advantages by integrating observational data with the underlying structure of differential equations, thereby preserving physical consistency while maintaining robustness under imperfect data conditions. In this study, PINNs are employed to estimate the parameters of the SIRS-D model using synthetic data generated through the fourth-order Runge–Kutta (RK4) method to ensure stable and consistent numerical solutions. To better represent real-world measurement conditions, 5% noise is added to the synthetic data, introducing realistic variability into the training process. The results demonstrate that PINNs successfully reconstruct the trajectories of S(t), I(t), R(t), and D(t) with low prediction errors. The model achieves MAE values of 0.0065 (S), 0.0067 (I), 0.0208 (R), and 0.0043 (D), with corresponding RMSE values of 0.0090, 0.0074, 0.0253, and 0.0058. Moreover, the estimated parameters closely match the true values, yielding ????????=0.5031, ????=0.0996, ????=0.0095, and ????=0.0149, demonstrating strong parameter identification capability. These findings confirm that PINNs constitute a reliable and accurate framework for analyzing infectious disease dynamics and offer promising potential for extension to more complex epidemiological models and real-world datasets.

Fitri Cahyani, Abdurakhman Abdurakhman, Chyntia Meininda Anjanni · 0 citations
Open access Aug 2026

Hierarchical Bayesian autoregressive smooth transition time series models

Hierarchical Bayesian smooth transition autoregressive models provide accurate forecasts by accommodating nonlinear regime-switching dynamics while delivering robust uncertainty quantification, which makes them well suited for infectious disease surveillance and public health decision-making in resource-limited, high-uncertainty settings.

G. Singini, Samuel Manda · 0 citations
Aug 2026

SUC–PINNs: A physics-informed neural networks approach to inverse problems in epidemic models with partial observability

Experiments with synthetic data and COVID-19 surveillance data show that SUC–PINN recovers plausible hidden infection trajectories, yields stable parameter estimates, and provides accurate short-term forecasts, support SUC–PINN as a practical computational approach for inverse modeling and prediction in partially observed epidemic dynamics.

U. M. Rifanti, N. Susyanto, Ratinan Boonklurb · 0 citations
Aug 2026

MechGNN-Epi: Mechanistically Constrained Spatiotemporal Graph Learning for Regional Epidemic Forecasting

This work proposes MechGNN-Epi, a hybrid framework that couples a spatiotemporal graph encoder with a differentiable SIR update that yields epidemiologically constrained trajectories and produces region- and time-indexed parameter proxies that can be inspected as diagnostic signals, while not being guaranteed as causally identifiable mechanistic parameters.

Debashis Chatterjee, Sagnik Acharyya, Subrata Rana · 0 citations
Jul 2026

Fractional physics-informed neural network approach to inverse problems in epidemiological modeling

This study proposes a Fractional-Order Physics-Informed Neural Network (FPINN) framework for solving inverse parameter estimation problems in both fractional SIR and augmented SEIR epidemiological models and demonstrates that the proposed method accurately reconstructs epidemic trajectories and captures the influence of memory effects on disease evolution.

S. Naveen, V. Parthiban · 0 citations
Review Open access Aug 2026

Neural Time Series Forecasting: A Problem-Driven Survey from Data Challenges to Model Choice

The literature is organized around five recurring difficulties: nonlinear and nonstationary behavior, contamination and structural breaks, uncertainty, long contexts and cross-variable dependence, and limited target-domain data.

Chuting Wen · 0 citations

Related blog posts

MIT News · Artificial Intelligence Aug 27, 2026

Looking beyond natural sequences

A new machine-learning framework aims to improve the success rate of computational protein design while moving away from results that reproduce sequences found in nature.