IEEE J Biomed Health Inform. 2023 May;27(5):2585-2596. doi: 10.1109/JBHI.2023.3247687. Epub 2023 May 4.
Early forecasting of influenza is an important task for public health to reduce losses due to influenza. Various deep learning-based models for multi-regional influenza forecasting have been proposed to forecast future influenza occurrences in multiple regions. While they only use historical data for forecasting, temporal and regional patterns need to be jointly considered for better accuracy. Basic deep learning models such as recurrent neural networks and graph neural networks have limited ability to model both patterns together. A more recent approach uses an attention mechanism or its variant, self-attention. Although these mechanisms can model regional interrelationships, in state-of-the-art models, they consider accumulated regional interrelationships based on attention values that are calculated only once for all of the input data. This limitation makes it difficult to effectively model the regional interrelationships that change dynamically during that period. Therefore, in this article, we propose a recurrent self-attention network (RESEAT) for various multi-regional forecasting tasks such as influenza and electrical load forecasting. The model can learn regional interrelationships over the entire period of the input data using self-attention, and it recurrently connects the attention weights using message passing. We demonstrate through extensive experiments that the proposed model outperforms other state-of-the-art forecasting models in terms of the forecasting accuracy for influenza and COVID-19. We also describe how to visualize regional interrelationships and analyze the sensitivity of hyperparameters to forecasting accuracy.
流感的早期预测是公共卫生的一项重要任务,可以减少流感造成的损失。已经提出了各种基于深度学习的多区域流感预测模型,以预测多个地区未来的流感发生情况。虽然这些模型仅使用历史数据进行预测,但为了提高准确性,还需要联合考虑时间和区域模式。基本的深度学习模型(如递归神经网络和图神经网络)在同时建模这两种模式方面的能力有限。一种更近期的方法使用注意力机制或其变体,即自注意力。虽然这些机制可以对区域间的相互关系进行建模,但在最先进的模型中,它们基于仅为所有输入数据计算一次的注意力值来考虑累积的区域间相互关系。这种限制使得难以有效地对该期间动态变化的区域间关系进行建模。因此,在本文中,我们提出了一种用于各种多区域预测任务(如流感和电力负荷预测)的递归自注意力网络(RESEAT)。该模型可以使用自注意力来学习输入数据整个周期的区域间关系,并通过消息传递递归地连接注意力权重。我们通过广泛的实验证明,与其他最先进的预测模型相比,所提出的模型在流感和 COVID-19 的预测准确性方面表现更好。我们还描述了如何可视化区域间关系,并分析超参数对预测准确性的敏感性。