Knowledge fusion enhanced graph neural network for traffic flow prediction

Shun Wang, Yong Zhang, Yongli Hu and Baocai Yin

Physica A: Statistical Mechanics and its Applications, 2023, vol. 623, issue C

Abstract: Traffic flow prediction is a very important and challenging task in intelligent transportation systems. There has been a lot of related research work on this issue, especially the application of graph convolutional networks has achieved quite good results. However, the existing methods usually only consider the temporal and spatial dependence in traffic data, and cannot fully explore the implicit semantic relationship from traffic knowledge. To solve this problem, we model the transportation system as topological graphs containing different types of knowledge such as network structure, regional functionality, and traffic flow patterns. We propose a Knowledge Fusion Enhanced Graph Neural Network (KFGNN) module based on multiple graph convolutional networks. Specifically, topological graphs are represented by relation matrices obtained by calculating traffic semantic similarity, and are used as the input of the Graph Convolutional Network(GCN) layer to capture the semantic dependence. The KFGNN module finally fuses these features to obtain a complex semantic representation of the traffic flow. Finally, knowledge fusion enhanced models (KE-TGCN, KE-STGCN and KE-GWN) are proposed to verify the effectiveness and versatility of this module. Experimental results on real-world datasets show that knowledge-enhanced models have higher prediction performance compared with classic GCN-based models.

Keywords: Graph neural network; Traffic flow prediction; Knowledge fusion. (search for similar items in EconPapers)
Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (4)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0378437123003977
Full text for ScienceDirect subscribers only. Journal offers the option of making the article available online on Science direct for a fee of $3,000

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:phsmap:v:623:y:2023:i:c:s0378437123003977

DOI: 10.1016/j.physa.2023.128842

Access Statistics for this article

Physica A: Statistical Mechanics and its Applications is currently edited by K. A. Dawson, J. O. Indekeu, H.E. Stanley and C. Tsallis

More articles in Physica A: Statistical Mechanics and its Applications from Elsevier
Bibliographic data for series maintained by Catherine Liu ().