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.
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· The eurasia proceedings of s...· 0 citations
Public health policy and disease surveillance systems require accurate forecasting of infectious disease dynamics to support timely interventions and resource allocation. However, classical linear time-series models often fail to capture abrupt regime shifts, nonlinear transmission patterns, and heterogeneous reporting commonly observed in surveillance data.
This study investigates the practical advantages of hierarchical Bayesian smooth transition autoregressive (BH-STAR) models, including logistic (LSTAR) and exponential (ESTAR) specifications. Performance is evaluated through extensive simulation studies under controlled nonlinear data-generating mechanisms, followed by an empirical application to COVID-19 surveillance data from 53 African countries collected between March 2020 and December 2022.
Simulation studies revealed a key directional asymmetry in model misspecification: fitting a logistic transition function to data generated under exponential dynamics resulted in moderate, stable parameter bias, whereas fitting an exponential transition function to logistic dynamics induced severe, compounding bias. Despite this parameter confounding under model misspecification, predictive accuracy remained stable across both data-generating processes. In the empirical application, the BH-STAR models consistently outperformed linear alternatives in out-of-sample forecasting. The hierarchical logistic STAR (HLSTAR) model achieved the highest overall predictive accuracy, reducing validation errors to an MAE of 0.58 and a MdAPE of 12.8%, corresponding to country-level forecast error reductions of 30–50% and overall error reductions exceeding 50% relative to the standard autoregressive benchmark.
Hierarchical Bayesian smooth transition autoregressive models provide accurate forecasts by accommodating nonlinear regime-switching dynamics while delivering robust uncertainty quantification. These features make them well suited for infectious disease surveillance and public health decision-making in resource-limited, high-uncertainty settings.
G. Singini, Samuel Manda· Frontiers in Applied Mathema...· 0 citations
Partially observed epidemic systems are difficult to analyze because confirmed cases provide only an indirect view of transmission and hidden infections are usually unobserved. We consider the susceptible–unidentified infected–confirmed (SUC) epidemic model and develop SUC–PINN, a physics-informed neural network framework for estimating hidden epidemic states, transmission and confirmation rates, and the initial unidentified population from confirmed-case time series alone. The method combines confirmed-case observations with SUC residuals and initial-condition constraints, so that the learned trajectories remain tied to the governing dynamics. Experiments with synthetic data and COVID-19 surveillance data from Indonesia, Thailand, India, and the Philippines show that SUC–PINN recovers plausible hidden infection trajectories, yields stable parameter estimates, and provides accurate short-term forecasts. These results 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· Advances in Complex Systems· 0 citations
Mechanistic compartmental models such as SIR provide interpretability and enforce conservation of population, but their predictive performance can degrade in heterogeneous settings where transmission varies over time and across regions. Conversely, spatiotemporal graph neural networks (GNNs) can learn flexible spatial interactions from data, yet unconstrained predictors may yield epidemiologically inconsistent dynamics.
We propose MechGNN-Epi, a hybrid framework that couples a spatiotemporal graph encoder with a differentiable SIR update. Rather than directly predicting future infections, the model learns region- and time-specific transmission and recovery parameters from graph-based latent representations and then propagates compartments through a mechanistic solver. This design 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. We evaluate the pipeline on U.S. state/territory COVID-19 surveillance data using strict chronological splits and report variability across multiple random seeds, multi-horizon rollouts, and ablation/sensitivity studies against graph and model variants.
Surveys of time series forecasting usually proceed by model family, from recurrent networks to Transformers and foundation models. That chronology is useful, but it gives limited guidance when the practical question is why a forecast fails. This article instead organizes the literature around five recurring difficulties: nonlinear and nonstationary behavior, contamination and structural breaks, uncertainty, long contexts and cross-variable dependence, and limited target-domain data. Studies published between 2014–2025 are compared through the assumptions they make, the settings in which they work, and the failure modes they leave unresolved. The review covers recurrent and probabilistic models, decomposition methods, robust and uncertainty-aware forecasting, Transformer variants, theory-guided methods, and foundation models. Quantitative results are used only when the underlying protocol is sufficiently clear. Across these families, the evidence supports a restrained workflow: begin with a strong simple baseline, identify the dominant source of error, and add complexity only when it addresses that source.
Chuting Wen· Applied and Computational En...· 0 citations
Related blog posts
MIT News · Artificial Intelligence· news.mit.eduAug 18, 2026
A new method for surgically removing training examples from a model reveals that as datasets grow, the link between what a model learns and what it produces dissolves.