论文标题
传输图神经网络用于大流行预测
Transfer Graph Neural Networks for Pandemic Forecasting
论文作者
论文摘要
Covid-19的最近爆发影响了世界各地数百万个人,并对全球医疗保健构成了重大挑战。从大流行的初期开始,很明显它具有高度传染性,人类流动性对其传播产生了重大贡献。在本文中,我们研究了人口运动对COVID-19的传播的影响,并利用了图表学习领域的最新进展,以捕获潜在的动态。具体而言,我们创建一个图形,其中节点对应于一个国家的区域,而边缘权重表示人类从一个地区到另一个地区的移动性。然后,我们采用图形神经网络来预测未来病例的数量,编码控制扩散到我们的学习模型的基本扩散模式。此外,为了说明培训数据的数量有限,我们利用了整个国家的大流行异步爆发,并使用基于模型的元学习方法将知识从一个国家的模型转移到另一个国家的模型。我们将提出的方法与3个欧洲国家的简单基线和更传统的预测技术进行了比较。实验结果证明了我们方法的优势,突出了GNN在流行病学预测中的实用性。如果利用了过去/平行爆发的数据,转移学习提供了最佳模型,强调了其潜力提高预测的准确性的潜力。
The recent outbreak of COVID-19 has affected millions of individuals around the world and has posed a significant challenge to global healthcare. From the early days of the pandemic, it became clear that it is highly contagious and that human mobility contributes significantly to its spread. In this paper, we study the impact of population movement on the spread of COVID-19, and we capitalize on recent advances in the field of representation learning on graphs to capture the underlying dynamics. Specifically, we create a graph where nodes correspond to a country's regions and the edge weights denote human mobility from one region to another. Then, we employ graph neural networks to predict the number of future cases, encoding the underlying diffusion patterns that govern the spread into our learning model. Furthermore, to account for the limited amount of training data, we capitalize on the pandemic's asynchronous outbreaks across countries and use a model-agnostic meta-learning based method to transfer knowledge from one country's model to another's. We compare the proposed approach against simple baselines and more traditional forecasting techniques in 3 European countries. Experimental results demonstrate the superiority of our method, highlighting the usefulness of GNNs in epidemiological prediction. Transfer learning provides the best model, highlighting its potential to improve the accuracy of the predictions in case of secondary waves, if data from past/parallel outbreaks is utilized.