用合成数据增强训练,提升新冠预测模型准确率。
Augmented data and neural networks for robust epidemic forecasting: application to COVID-19 in Italy
- 用传染病模型生成带不确定性的合成数据,扩充真实数据集
- NAR模型短期预测更准,PINNs擅长捕捉长期趋势
- 适合需要高精度短期预测或研究动态规律的研究者
本文提出一种数据增强策略,旨在提升神经网络训练效果,从而提高预测准确性。方法基于合适的流行病学分层模型生成合成数据,并引入不确定性信息;利用真实数据校准模型后,结合深度学习技术生成额外合成数据用于训练。实验表明,使用增强数据训练的神经网络预测性能显著提升。重点对比了两类模型:物理信息神经网络(PINNs)和非线性自回归(NAR)模型。NAR在短期预测中表现优异,能直接从数据学习动态,避免引入物理约束带来的计算开销;而PINNs虽定量预测较弱,但能较好捕捉系统的定性长期行为,更适合分析整体动态趋势。针对意大利伦巴第大区第二阶段新冠疫情的数值模拟验证了该方法的有效性。
原文摘要 · Abstract (English)
In this work, we propose a data augmentation strategy aimed at improving the training phase of neural networks and, consequently, the accuracy of their predictions. Our approach relies on generating synthetic data through a suitable compartmental model combined with the incorporation of uncertainty. The available data are then used to calibrate the model, which is further integrated with deep learning techniques to produce additional synthetic data for training. The results show that neural networks trained on these augmented datasets exhibit significantly improved predictive performance. We focus in particular on two different neural network architectures: Physics-Informed Neural Networks (PINNs) and Nonlinear Autoregressive (NAR) models. The NAR approach proves especially effective for short-term forecasting, providing accurate quantitative estimates by directly learning the dynamics from data and avoiding the additional computational cost of embedding physical constraints into the training. In contrast, PINNs yield less accurate quantitative predictions but capture the qualitative long-term behavior of the system, making them more suitable for exploring broader dynamical trends. Numerical simulations of the second phase of the COVID-19 pandemic in the Lombardy region (Italy) validate the effectiveness of the proposed approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。