arrow
Return

Backbone-based Dynamic Spatio-Temporal Graph Neural Network for epidemic forecasting

delete2024-07-01
delete1
PRE
AI
Y
Yuexing Han
G
Gouhei Tanaka
王兵 (Bing Wang)
DOI:10.1016/j.knosys.2024.111952delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Accurate epidemic forecasting is a critical task in controlling epidemic spread. Many deep learning -based models focus only on static or dynamic graphs when dealing with spatial information, ignoring their relationship. Additionally, these models often rely on recurrent structures, which can lead to error accumulation and computational time consumption. To address the aforementioned problems, we propose a novel model called Backbone -based Dynamic Spatio-Temporal Graph Neural Network (BDSTGNN). Intuitively, the continuous and smooth changes in graph structure make adjacent graph structures share a basic pattern. To capture this property, we use adaptive methods to generate static backbone graphs containing the primary information, and use temporal models to generate dynamic temporal graphs, and then fuse them to generate a backbone -based dynamic graph. To overcome potential limitations associated with recurrent structures, we introduce a linear model DLinear to handle temporal dependencies, and combine it with dynamic graph convolution for epidemic forecasting. Extensive experiments on two datasets demonstrate that BDSTGNN outperforms baseline models, and ablation comparison further verifies the effectiveness of model components. Furthermore, we analyze and measure the significance of backbone and temporal graphs by using information metrics from different aspects. Finally, we verify the superior efficiency of the BDSTGNN.
Keywords:
Epidemic forecasting
Spatio-temporal neural networks
Dynamic graph
Time series decomposition

Journal

K
Knowledge-Based Systems
IF:
7.6
Papers:
1.2W
Citations:
4.5W

Organization

U
University of Tokyo
Scholars:
7.1W
Papers: 6.5W
Citations: 2.2K
S
shanghai university
Scholars:
3.9W
Papers: 2.7W
Citations: 52