A Dynamic Regional-Aggregation-Based Heterogeneous Graph Neural Network for Traffic Prediction

Xiangting Liu, Chengyuan Qian and Xueyang Zhao ()
Additional contact information
Xiangting Liu: School of International Education, Guangdong University of Technology, Guangzhou 511495, China
Chengyuan Qian: School of Mechanical and Energy Engineering, Tongji University, Shanghai 201804, China
Xueyang Zhao: Department of Mathematics and Physics, Harbin Institute of Petroleum, Harbin 150028, China

Mathematics, 2025, vol. 13, issue 9, 1-21

Abstract: Traffic flow prediction, crucial for intelligent transportation systems, has seen advancements with graph neural networks (GNNs), yet existing methods often fail to distinguish between the importance of different intersections. These methods usually model all intersections uniformly, overlooking significant differences in traffic flow characteristics and influence ranges between ordinary and important nodes. To tackle this, this study introduces a dynamic regional-aggregation-based heterogeneous graph neural network (DR-HGNN). This model categorizes intersections into two types—ordinary and important—to apply tailored feature aggregation strategies. Ordinary intersections aggregate features based on local neighborhood information, whereas important intersections utilize deeper neighborhood diffusion and multi-hop dependencies to capture broader traffic influences. The DR-HGNN model also employs a dynamic graph structure to reflect temporal changes in traffic flows, alongside an attention mechanism for adaptive regional feature aggregation, enhancing the identification of critical traffic nodes. Demonstrating its efficacy, the DR-HGNN achieved 19.2% and 15.4% improvements in the RMSE over 50 min predictions in the METR-LA and PEMS-BAY datasets, respectively, offering a more precise prediction method for traffic management.

Keywords: traffic flow prediction; dynamic heterogeneous graph neural network; regional aggregation; attention mechanism; spatiotemporal dependencies (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/13/9/1458/pdf (application/pdf)
https://www.mdpi.com/2227-7390/13/9/1458/ (text/html)

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:gam:jmathe:v:13:y:2025:i:9:p:1458-:d:1645589

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().