Goto

Collaborating Authors

 crosslag


CrossLag: Predicting Major Dengue Outbreaks with a Domain Knowledge Informed Transformer

Prabu, Ashwin, Tran, Nhat Thanh, Zhou, Guofa, Xin, Jack

arXiv.org Artificial Intelligence

ABSTRACT A variety of models have been developed to forecast dengue cases to date. However, it remains a challenge to predict major dengue outbreaks that need timely public warnings the most. In this paper, we introduce CrossLag, an environmentally informed attention that allows for the incorporation of lagging endogenous signals behind the significant events in the exogenous data into the architecture of the transformer at low parameter counts. We use TimeXer, a recent general-purpose transformer distinguishing exogenous-endogenous inputs, as the baseline for this study. Our proposed model outperforms TimeXer by a considerable margin in detecting and predicting major outbreaks in Singapore dengue data over a 24-week prediction window.