AMGST: Adaptive multi-graph convolution and spatiotemporal attention network for traffic forecasting

Pei Shi, Qixiang Lu, Jiahui Chen, Xiaoliu Lv, Lu Zhang, Liang Kuang and Jiadong Sun

PLOS ONE, 2026, vol. 21, issue 6, 1-22

Abstract: Traffic forecasting is crucial for optimizing traffic management and control strategies. As a powerful approach for analyzing and mining graph-structured data, graph convolution has shown great potential in traffic prediction. However, it still struggles to fully capture global spatial correlations and long-term dynamic temporal dependencies inherent in spatiotemporal traffic patterns. Moreover, the quality of the graph structure directly affects the extraction of these correlations. To address these challenges, we propose AMGST, an Adaptive Multi-Graph Convolution and Spatiotemporal Multi-Head Self-Attention Network for traffic forecasting. AMGST integrates an Adaptive Spatiotemporal Embedding (ASTE) generator, a multi-graph diffusion convolution module, and a spatiotemporal attention mechanism. First, dynamic spatiotemporal representations are generated using the ASTE module. Then, the multi-graph diffusion convolution leverages both a maximum mutual information coefficient matrix and an adaptive matrix to extract fine-grained spatial features. A global spatial attention mechanism is applied to capture dynamic spatial correlations, while a temporal attention module models nonlinear temporal dependencies. Experimental results on four public traffic datasets, including both speed and flow measurements, demonstrate that AMGST consistently surpasses the baselines, confirming its effectiveness in providing accurate traffic forecasts.

Date: 2026
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0342235 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 42235&type=printable (application/pdf)

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:plo:pone00:0342235

DOI: 10.1371/journal.pone.0342235

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().