TraverseNet: Unifying Space and Time in Message Passing for Traffic Forecasting

被引:19
作者
Wu, Zonghan [1 ]
Zheng, Da [2 ]
Pan, Shirui [3 ,4 ]
Gan, Quan [2 ]
Long, Guodong
Karypis, George [2 ]
机构
[1] Univ Technol Sydney, FEIT, Ctr Artificial Intelligence, Ultimo, NSW 2007, Australia
[2] Amazon, Seattle, WA 98109 USA
[3] Monash Univ, Dept Data Sci & AI, Clayton, Vic 3800, Australia
[4] Griffith Univ, Sch Informat & Commun Technol, Southport, Qld 4222, Australia
基金
澳大利亚研究理事会;
关键词
Convolution; Message passing; Correlation; Kernel; Graph neural networks; Forecasting; Convolutional neural networks; Deep learning; graph autoencoder; graph convolutional networks; graph neural networks (GNNs); graph representation learning; network embedding; GRAPH; NETWORKS;
D O I
10.1109/TNNLS.2022.3186103
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This article aims to unify spatial dependency and temporal dependency in a non-Euclidean space while capturing the inner spatial-temporal dependencies for traffic data. For spatial-temporal attribute entities with topological structure, the space-time is consecutive and unified while each node's current status is influenced by its neighbors' past states over variant periods of each neighbor. Most spatial-temporal neural networks for traffic forecasting study spatial dependency and temporal correlation separately in processing, gravely impaired the spatial-temporal integrity, and ignore the fact that the neighbors' temporal dependency period for a node can be delayed and dynamic. To model this actual condition, we propose TraverseNet, a novel spatial-temporal graph neural network, viewing space and time as an inseparable whole, to mine spatial-temporal graphs while exploiting the evolving spatial-temporal dependencies for each node via message traverse mechanisms. Experiments with ablation and parameter studies have validated the effectiveness of the proposed TraverseNet, and the detailed implementation can be found from https://github.com/nnzhan/TraverseNet.
引用
收藏
页码:2003 / 2013
页数:11
相关论文
共 41 条
[1]   Graph Neural Networks With Convolutional ARMA Filters [J].
Bianchi, Filippo Maria ;
Grattarola, Daniele ;
Livi, Lorenzo ;
Alippi, Cesare .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (07) :3496-3507
[2]  
Cho K., 2014, ARXIV14061078, P1724, DOI [10.3115/v1/D14-1179, DOI 10.3115/V1/D14-1179]
[3]   Graph Representation Learning via Hard and Channel-Wise Attention Networks [J].
Gao, Hongyang ;
Ji, Shuiwang .
KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, :741-749
[4]  
Gilmer J, 2017, PR MACH LEARN RES, V70
[5]  
Guo SN, 2019, AAAI CONF ARTIF INTE, P922
[6]  
Hadou S., 2022, PROC INT C LEARN REP
[7]  
Hamilton WL, 2017, ADV NEUR IN, V30
[8]  
Isufi E., 2021, PROC IEEE DATA SCI L, P1, DOI DOI 10.1109/TPAMI.2021.3111054
[9]  
Jin Ming, 2022, ARXIV220208408
[10]  
Kipf T. N., 2016, P 5 INT C LEARNING R