Spatial Origin-Destination Flow Imputation Using Graph Convolutional Networks

Xin Yao, Yong Gao, Di Zhu, Ed Manley, Jiaoe Wang, Yu Liu

Research output: Contribution to journalArticlepeer-review

93 Scopus citations

Abstract

Due to the limitation of data collection techniques and privacy issues, the problem of missing spatial origin-destination flows frequently occurs. Data imputation provides great support for the acquisition of complete flow data, which enables us to better understand regional connections and mobility patterns. However, existing models or approaches neglect the network structure of spatial flows, thus resulting in inappropriate estimates and a low performance. The development of graph neural networks offers a powerful tool to deal with graph-structured data. In this article, we proposed a spatial interaction graph convolutional network model, which combines graph convolution and a mapping function to predict flow data from the perspective of network learning. This model utilizes geographical unit embedding in local spatial networks to improve prediction accuracy. A negative sampling technique is adopted to reduce misestimation. Experiments on Beijing taxi trip data verified the usefulness of our model in spatial flow prediction. We also demonstrated that a biased training sample had a negative impact on the model's performance. More attributes of geographical units, a more proper negative sampling rate and a larger training set can increase the prediction accuracy of flow data.

Original languageEnglish (US)
Pages (from-to)7474-7484
Number of pages11
JournalIEEE Transactions on Intelligent Transportation Systems
Volume22
Issue number12
DOIs
StatePublished - Dec 1 2021
Externally publishedYes

Bibliographical note

Publisher Copyright:
© 2000-2011 IEEE.

Keywords

  • Origin-destination flow
  • data imputation
  • graph convolution
  • graph embedding
  • spatial interaction network

Fingerprint

Dive into the research topics of 'Spatial Origin-Destination Flow Imputation Using Graph Convolutional Networks'. Together they form a unique fingerprint.

Cite this