TY - JOUR
T1 - Time series modeling and forecasting of epidemic spreading processes using deep transfer learning
AU - Xue, Dong
AU - Wang, Ming
AU - Liu, Fangzhou
AU - Buss, Martin
N1 - Publisher Copyright:
© 2024 Elsevier Ltd
PY - 2024/8
Y1 - 2024/8
N2 - Traditional data-driven methods for modeling and predicting epidemic spreading typically operate in an independent and identically distributed setting. However, epidemic spreading on complex networks exhibits significant heterogeneity across different phases, regions, and viruses, indicating that epidemic time series may not be independent and identically distributed due to temporal and spatial variations. In this article, a novel deep transfer learning method integrating convolutional neural networks (CNNs) and bi-directional long short-term memory (BiLSTM) networks is proposed to model and forecast epidemics with heterogeneous data. The proposed method combines a CNN-based layer for local feature extraction, a BiLSTM-based layer for temporal analysis, and a fully connected layer for prediction, and employs transfer learning to enhance the generalization ability of the CNN-BiLSTM model. To improve prediction performance, hyperparameter tuning is conducted using particle swarm optimization during model training. Finally, we adopt the proposed approach to characterize the spatio-temporal spreading dynamics of COVID-19 and infer the pathological heterogeneity among epidemics of severe acute respiratory syndrome (SARS), influenza A (H1N1), and COVID-19. The comprehensive results demonstrate the effectiveness of the proposed approach in exploring the spatiotemporal variations in the spread of epidemics and characterizing the epidemiological features of different viruses. Moreover, the proposed method can significantly reduce modeling and predicting errors in epidemic spread to some extent.
AB - Traditional data-driven methods for modeling and predicting epidemic spreading typically operate in an independent and identically distributed setting. However, epidemic spreading on complex networks exhibits significant heterogeneity across different phases, regions, and viruses, indicating that epidemic time series may not be independent and identically distributed due to temporal and spatial variations. In this article, a novel deep transfer learning method integrating convolutional neural networks (CNNs) and bi-directional long short-term memory (BiLSTM) networks is proposed to model and forecast epidemics with heterogeneous data. The proposed method combines a CNN-based layer for local feature extraction, a BiLSTM-based layer for temporal analysis, and a fully connected layer for prediction, and employs transfer learning to enhance the generalization ability of the CNN-BiLSTM model. To improve prediction performance, hyperparameter tuning is conducted using particle swarm optimization during model training. Finally, we adopt the proposed approach to characterize the spatio-temporal spreading dynamics of COVID-19 and infer the pathological heterogeneity among epidemics of severe acute respiratory syndrome (SARS), influenza A (H1N1), and COVID-19. The comprehensive results demonstrate the effectiveness of the proposed approach in exploring the spatiotemporal variations in the spread of epidemics and characterizing the epidemiological features of different viruses. Moreover, the proposed method can significantly reduce modeling and predicting errors in epidemic spread to some extent.
KW - CNN-BiLSTM model
KW - Deep transfer learning
KW - Epidemic spreading
KW - Time-series data
UR - http://www.scopus.com/inward/record.url?scp=85196730150&partnerID=8YFLogxK
U2 - 10.1016/j.chaos.2024.115092
DO - 10.1016/j.chaos.2024.115092
M3 - Article
AN - SCOPUS:85196730150
SN - 0960-0779
VL - 185
JO - Chaos, Solitons and Fractals
JF - Chaos, Solitons and Fractals
M1 - 115092
ER -