Local-Global Dynamic Information Fusion Graph Learning for Traffic Flow Prediction

Yuan Xu, Fan Qin, Wei Ke, Yan Lin He, Ming Qing Zhang, Qun Xiong Zhu, Yang Zhang

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

In the field of intelligent transportation systems, accurately predicting traffic flow is a challenging endeavor for enabling efficient services in urban road networks. While current research has made significant advancements in capturing nonlinear features and spatiotemporal dependencies at a fixed geographic node level, it often overlooks the dynamic interactions of traffic flow across different regions. In this work, we present a novel local-global dynamic information fusion graph learning model (LGDF-GL) for traffic flow prediction. Different from other graph learning model, we first design a dynamic adjacency matrix by region-adaptive parameter learning, and fuse it with a static adjacency matrix based on node geographic information by dynamic information fusion module, and the fused result serves as the diagonal elements of the local adjacency matrix. Subsequently, we input it into the graph convolutional network (GCN) and gated recurrent unit (GRU) aggregation model to extract local feature information containing dynamic spatiotemporal features from traffic flow data. The local feature information is pruned and concatenated to obtain global feature representations. Then we achieve global feature representation prediction through multiple fully connected layers. Finally, we conducted experiments on two real-world traffic flow datasets to evaluate the performance of our proposed method. The results clearly demonstrate that the LGDF-GL method achieves superior prediction ability compared to several baseline models.

Original languageEnglish
Title of host publicationProceedings of 2024 IEEE 13th Data Driven Control and Learning Systems Conference, DDCLS 2024
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages208-213
Number of pages6
ISBN (Electronic)9798350361674
DOIs
Publication statusPublished - 2024
Event13th IEEE Data Driven Control and Learning Systems Conference, DDCLS 2024 - Kaifeng, China
Duration: 17 May 202419 May 2024

Publication series

NameProceedings of 2024 IEEE 13th Data Driven Control and Learning Systems Conference, DDCLS 2024

Conference

Conference13th IEEE Data Driven Control and Learning Systems Conference, DDCLS 2024
Country/TerritoryChina
CityKaifeng
Period17/05/2419/05/24

Keywords

  • Gated recurrent unit
  • Graph convolutional network
  • Local-global dynamic information fusion
  • Traffic flow prediction

Fingerprint

Dive into the research topics of 'Local-Global Dynamic Information Fusion Graph Learning for Traffic Flow Prediction'. Together they form a unique fingerprint.

Cite this