A data augmentation strategy for deep neural networks with application to epidemic modelling

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Awais, Muhammad, Ali, Abu Safyan, Dimarco, Giacomo, Ferrarese, Federica, Pareschi, Lorenzo
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916761178210304
author Awais, Muhammad
Ali, Abu Safyan
Dimarco, Giacomo
Ferrarese, Federica
Pareschi, Lorenzo
author_facet Awais, Muhammad
Ali, Abu Safyan
Dimarco, Giacomo
Ferrarese, Federica
Pareschi, Lorenzo
contents In this work, we integrate the predictive capabilities of compartmental disease dynamics models with machine learning ability to analyze complex, high-dimensional data and uncover patterns that conventional models may overlook. Specifically, we present a proof of concept demonstrating the application of data-driven methods and deep neural networks to a recently introduced Susceptible-Infected-Recovered type model with social features, including a saturated incidence rate, to improve epidemic prediction and forecasting. Our results show that a robust data augmentation strategy trough suitable data-driven models can improve the reliability of Feed-Forward Neural Networks and Nonlinear Autoregressive Networks, providing a complementary strategy to Physics-Informed Neural Networks, particularly in settings where data augmentation from mechanistic models can enhance learning. This approach enhances the ability to handle nonlinear dynamics and offers scalable, data-driven solutions for epidemic forecasting, prioritizing predictive accuracy over the constraints of physics-based models. Numerical simulations of the lockdown and post-lockdown phase of the COVID-19 epidemic in Italy and Spain validate our methodology.
format Preprint
id arxiv_https___arxiv_org_abs_2502_21033
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A data augmentation strategy for deep neural networks with application to epidemic modelling
Awais, Muhammad
Ali, Abu Safyan
Dimarco, Giacomo
Ferrarese, Federica
Pareschi, Lorenzo
Numerical Analysis
Machine Learning
Physics and Society
Populations and Evolution
In this work, we integrate the predictive capabilities of compartmental disease dynamics models with machine learning ability to analyze complex, high-dimensional data and uncover patterns that conventional models may overlook. Specifically, we present a proof of concept demonstrating the application of data-driven methods and deep neural networks to a recently introduced Susceptible-Infected-Recovered type model with social features, including a saturated incidence rate, to improve epidemic prediction and forecasting. Our results show that a robust data augmentation strategy trough suitable data-driven models can improve the reliability of Feed-Forward Neural Networks and Nonlinear Autoregressive Networks, providing a complementary strategy to Physics-Informed Neural Networks, particularly in settings where data augmentation from mechanistic models can enhance learning. This approach enhances the ability to handle nonlinear dynamics and offers scalable, data-driven solutions for epidemic forecasting, prioritizing predictive accuracy over the constraints of physics-based models. Numerical simulations of the lockdown and post-lockdown phase of the COVID-19 epidemic in Italy and Spain validate our methodology.
title A data augmentation strategy for deep neural networks with application to epidemic modelling
topic Numerical Analysis
Machine Learning
Physics and Society
Populations and Evolution
url https://arxiv.org/abs/2502.21033