arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.10052cs.LGcs.AI
收藏

集成LSTM和GNN用于COVID-19预测

Integrating LSTMs and GNNs for COVID-19 Forecasting

Massachusetts Institute of Technology(麻省理工学院)

Nathan Sesti, Juan Jose Garau-Luis, Edward Crawley, Bruce Cameron

摘要:

COVID-19的传播恰逢图神经网络(GNN)的兴起,导致几项研究提出使用它们来更好地预测大流行的演变。许多这样的模型还包括长短期记忆(LSTM)网络,这是一种用于时间序列预测的常用工具。在这项工作中,我们通过在LSTM的门内实现GNN并利用空间信息来进一步研究这两种方法的集成。此外,我们引入了一个跳过连接,证明是至关重要的,共同捕捉的空间和时间模式的数据。我们根据37个欧洲国家过去472天的数据验证了我们的每日COVID-19新病例预测模型,与基于平均绝对标度误差(MASE)的最先进的图形时间序列模型相比,表现出更优越的性能。这一领域的研究有重要的应用政策制定,我们分析其潜在的流行病资源控制。

英文摘要:

The spread of COVID-19 has coincided with the rise of Graph Neural Networks (GNNs), leading to several studies proposing their use to better forecast the evolution of the pandemic. Many such models also include Long Short Term Memory (LSTM) networks, a common tool for time series forecasting. In this work, we further investigate the integration of these two methods by implementing GNNs within the gates of an LSTM and exploiting spatial information. In addition, we introduce a skip connection which proves critical to jointly capture the spatial and temporal patterns in the data. We validate our daily COVID-19 new cases forecast model on data of 37 European nations for the last 472 days and show superior performance compared to state-of-the-art graph time series models based on mean absolute scaled error (MASE). This area of research has important applications to policy-making and we analyze its potential for pandemic resource control.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑