Abstract
Epidemic forecasting and policy evaluation rely on mathematical models to predict infectious disease trends and assess the impact of public health policies. Traditional models typically require extensive epidemiological data and may struggle in data-limited settings. Transformer-based, foundation AI models have demonstrated strong predictive capabilities in various time series applications. We investigated whether they can be the basis of a new epidemic modeling framework. We evaluated five foundation models - TabPFN-TS, TimeGPT, TimesFM, Lag-Llama, and Chronos - across diverse pathogens, diseases and locations, including influenza-like illness, RSV, chickenpox, dengue, COVID-19 and neonatal bronchiolitis. Models were tested for long-term forecasting (multi-season predictions), short-term forecasting (four-week-ahead predictions), and epidemic peak timing estimation. We also assessed their ability to generate counterfactual scenarios in policy evaluation, using COVID-19 restriction measures in Italy, RSV immunization in France, and synthetic epidemic data as validation. Foundation models demonstrated strong predictive accuracy, possibly outperforming traditional statistical and mechanistic models in data-limited contexts. They generated multi-season forecasts and short-term forecasts with good accuracy and stable uncertainty. They gave reliable peak timing estimates months before the actual peak. In policy evaluation, TabPFN-TS accurately estimated intervention effects, matching estimates from an independent epidemiological study. Our findings suggest that foundation models can complement existing approaches in epidemic modeling. Their ability to generate accurate forecasts and counterfactual analyses with minimal data input highlights their potential for real-time public health decision-making, particularly in emergent and resource-constrained settings. Further research should explore domain-specific adaptations to optimize performance for infectious disease modeling.
Competing Interest Statement
The authors have declared no competing interest.
Funding Statement
This study was partially supported by Horizon Europe grant SIESTA (101131957) to E.V., the CIPHOD project (ANR-23-CPJ1-0212-01) to F. B., the European Union's Horizon Europe research and innovation programme Cofund SOUND.AI under the Marie Sklodowska-Curie Grant Agreement No 101081674 to S.K.
Author Declarations
I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.
Yes
The details of the IRB/oversight body that provided approval or exemption for the research described are given below:
No individual data were used here, not even in an anonymized form. Only population-level case count were used. Data on ILI, Chickenpox, Influenza, RSV in France are available from Reseau Sentinelles (https://www.sentiweb.fr/france/fr/?page=database). Data on COVID-19 in Italy are available from https://doi.org/10.1038/s41597-023-02276-y. Bronchiolitis data are from https://doi.org/10.1016/S2352-4642(24)00171-8.
I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.
Yes
I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).
Yes
I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.
Yes
Data Availability
Data on ILI, Chickenpox, Influenza, RSV in France are available from Reseau Sentinelles (https://www.sentiweb.fr/france/fr/?page=database). Data on COVID-19 in Italy are available from https://doi.org/10.1038/s41597-023-02276-y. Bronchiolitis data are from https://doi.org/10.1016/S2352-4642(24)00171-8.