Department of Computer Science and Technology, Hangzhou Dianzi University, Hangzhou, China; Zhejiang Provincial Key Laboratory of Industrial Internet in Discrete Industries, Hangzhou, China.
Hangzhou Dianzi University ITMO Joint Institute, Hangzhou Dianzi University, Hangzhou, China.
Neural Netw. 2024 Apr;172:106093. doi: 10.1016/j.neunet.2023.106093. Epub 2024 Jan 3.
Traffic Prediction based on graph structures is a challenging task given that road networks are typically complex structures and the data to be analyzed contains variable temporal features. Further, the quality of the spatial feature extraction is highly dependent on the weight settings of the graph structures. In the transportation field, the weights of these graph structures are currently calculated based on factors like the distance between roads. However, these methods do not take into account the characteristics of the road itself or the correlations between different traffic flows. Existing approaches usually pay more attention to local spatial dependencies extraction while global spatial dependencies are ignored. Another major problem is how to extract sufficient information at limited depth of graph structures. To address these challenges, we propose a Random Graph Diffusion Attention Network (RGDAN) for traffic prediction. RGDAN comprises a graph diffusion attention module and a temporal attention module. The graph diffusion attention module can adjust its weights by learning from data like a CNN to capture more realistic spatial dependencies. The temporal attention module captures the temporal correlations. Experiments on three large-scale public datasets demonstrate that RGDAN produces predictions with 2%-5% more precision than state-of-the-art methods.
基于图结构的交通预测是一项具有挑战性的任务,因为道路网络通常是复杂的结构,并且要分析的数据包含可变的时间特征。此外,空间特征提取的质量高度依赖于图结构的权重设置。在交通领域,这些图结构的权重目前是根据道路之间的距离等因素来计算的。然而,这些方法没有考虑道路本身的特点或不同交通流之间的相关性。现有的方法通常更关注局部空间依赖的提取,而忽略全局空间依赖。另一个主要问题是如何在有限的图结构深度中提取足够的信息。为了解决这些挑战,我们提出了一种用于交通预测的随机图扩散注意网络(RGDAN)。RGDAN 包括一个图扩散注意模块和一个时间注意模块。图扩散注意模块可以通过像 CNN 一样从数据中学习来调整其权重,以捕获更真实的空间依赖。时间注意模块捕捉时间相关性。在三个大规模的公共数据集上的实验表明,RGDAN 的预测精度比最先进的方法高出 2%-5%。