余剑峤
James Jianqiao Yu
首页 论文 服务 ENG

讲师(助理教授)

计算机科学系

约克大学

英国约克 YO10 5GH CSE/139

jqyu(at)ieee.org Google Scholar
Spatial-Temporal Traffic Data Imputation via Graph Attention Convolutional Network

作者
Yongchao Ye, Shiyao Zhang, and James J.Q. Yu*

发表
Proc. International Conference on Artificial Neural Networks, Bratislava, Slovakia, September 2021

摘要
High-quality traffic data is crucial for intelligent transportation system and its data-driven applications. However, data missing is common in collecting real-world traffic datasets due to various factors. Thus, imputing missing values by extracting traffic characteristics becomes an essential task. By using conventional convolutional neural network layers or focusing on standalone road sections, existing imputation methods cannot model the non-Euclidean spatial correlations of complex traffic networks. To address this challenge, we propose a graph attention convolutional network (GACN), a novel model for traffic data imputation. Specifically, the model follows an encoder-decoder structure and incorporates graph attention mechanism to learn spatial correlation of the traffic data collected by adjacent sensors on traffic graph. Temporal convolutional layers are stacked to extract relations in time-series after graph attention layers. Through comprehensive case studies on the dataset from the Caltrans performance measurement system (PeMS), we demonstrate that the proposed GACN consistently outperforms other baselines and has steady performance in extreme missing rate scenarios.