Light attention-based neural networks for traffic flow prediction

Yong Li, Jiajun Wang and Liujiang Kang

Physica A: Statistical Mechanics and its Applications, 2025, vol. 673, issue C

Abstract: Spatial–temporal traffic patterns in transportation significantly influence the design of prediction models, which require both high accuracy and computational efficiency. This paper introduces the Light Attention-based Spatial-Temporal Neural Networks (Light-ASTNN), a lightweight traffic prediction model designed for higher prediction accuracy. The model integrates network topology information from a transportation network into a spatial attention to enhance the attention mechanism’s capacity. The effectiveness of the proposed model is validated through comparable experiments with a previous model, using 5 real-world traffic graph network-based datasets. The experimental results show that the proposed model can achieve a better performance in both the accuracy and computational efficiency, despite the fewer parameters. Furthermore, the experiments further highlight the critical role of network topology information in computing spatial correlations using the attention mechanism.

Keywords: Spatial–temporal traffic patterns; Traffic prediction; Spatial attention; Network topology (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0378437125003176
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:673:y:2025:i:c:s0378437125003176

DOI: 10.1016/j.physa.2025.130665

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 ().