1
Return

Predicting influenza in the post-COVID era: assessing LSTM; GRU; and transformer robustness to covariate shift

delete2026-08-11
delete0
delete
OA
AI
A
AN Atiqa Naeem Alam Din
W
WA Woldegebriel Assefa Woldegerima *
J
JW Jianhong Wu
DOI:10.3389/frai.2026.1886896delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Forecasting influenza has become increasingly challenging due to post-COVID disruptions in seasonality and strain circulation. This work compares the performance of Long Short Term Memory Networks (LSTM); Gated Recurrent Unit (GRU); and transformer models in forecasting influenza spread using multivariate epidemiological and environmental data; with a focus on robustness under post-COVID non-stationarity. We compare LSTM; GRU; and transformer architectures within a multivariate deep learning framework using influenza and temperature data from Ontario (2014–2025); with data split into training; validation; and testing periods. Although recurrent models outperform transformers on limited; noisy data; all architectures exhibit marked performance collapse under post-COVID non-stationarity. The GRU and LSTM track pre-COVID seasonal peaks more closely; yet both substantially under-estimate the post-COVID resurgence; indicating that none of the models generalize across the regime shift. These findings position our study as a diagnostic of how architectural inductive biases break down under covariate shift. Furthermore; this manuscript assesses how the COVID-19 pandemic affected the accuracy and performance of machine learning algorithms and notes the integration of transfer learning and attention mechanisms to improve model performance.
Keywords:
influenza
transformer
attention mechanism
regime shift
back propagation
GRU (Gated Recurrent Unit)
LSTM (Long Short Term Memory Networks)

Journal

F
Frontiers in Artificial Intelligence
IF:
4.7
Papers:
2.2K
Citations:
4.4K

Organization

L
D
disease-informed modeling
Scholars:
3
Papers: 1
Citations: 0
Cited Papers

Cited Papers

Citing Papers

Citing Papers