An Adaptive Spatio-Temporal Traffic Flow Prediction Using Self-Attention and Multi-Graph Networks.

Basma Alsehaimi, Ohoud Alzamzami, Nahed Alowidi, Manar Ali
Author Information
  1. Basma Alsehaimi: Department of Computer Science, King AbdulAziz University, Jeddah 21589, Saudi Arabia.
  2. Ohoud Alzamzami: Department of Computer Science, King AbdulAziz University, Jeddah 21589, Saudi Arabia. ORCID
  3. Nahed Alowidi: Department of Computer Science, King AbdulAziz University, Jeddah 21589, Saudi Arabia. ORCID
  4. Manar Ali: Department of Computer Science, King AbdulAziz University, Jeddah 21589, Saudi Arabia. ORCID

Abstract

Traffic flow prediction is a pivotal element in Intelligent Transportation Systems (ITSs) that provides significant opportunities for real-world applications. Capturing complex and dynamic spatio-temporal patterns within traffic data remains a significant challenge for traffic flow prediction. Different approaches to effectively modeling complex spatio-temporal correlations within traffic data have been proposed. These approaches often rely on a single model to capture temporal dependencies, which neglects the varying influences of different time periods on traffic flow. Additionally, these models frequently utilize either static or dynamic graphs to represent spatial dependencies, which limits their ability to address complex and overlapping spatial relationships. Moreover, some approaches struggle to fully capture spatio-temporal variations, leading to the exclusion of critical information and ultimately resulting in suboptimal prediction performance. Thus, this paper introduces the Adaptive Spatio-Temporal Attention-Based Multi-Model (ASTAM), an architecture designed to capture spatio-temporal dependencies within traffic data. The ASTAM employs multi-temporal gated convolution with multi-scale temporal input segments to model complex non-linear temporal correlations. It utilizes static and dynamic parallel multi-graphs to facilitate the modeling of complex spatial dependencies. Furthermore, this model incorporates a spatio-temporal self-attention mechanism to adaptively capture the dynamic and long-term spatio-temporal variations in traffic flow. Experiments conducted on four real-world datasets reveal that the proposed architecture outperformed 13 baseline approaches, achieving average reductions of 5.0% in MAE, 13.28% in RMSE, and 6.46% in MAPE across four datasets.

Keywords

References

  1. IEEE Trans Neural Netw. 2009 Jan;20(1):61-80 [PMID: 19068426]
  2. Sensors (Basel). 2024 Apr 07;24(7): [PMID: 38610556]
  3. Sci Rep. 2024 Nov 2;14(1):26473 [PMID: 39488556]

Word Cloud

Created with Highcharts 10.0.0trafficspatio-temporalflowcomplexpredictiondynamicapproachescapturetemporaldependencieswithindatamodelspatialTrafficsignificantreal-worldmodelingcorrelationsproposedstaticvariationsAdaptiveSpatio-TemporalASTAMarchitectureconvolutionmechanismfourdatasets13attentiongraphnetworkpivotalelementIntelligentTransportationSystemsITSsprovidesopportunitiesapplicationsCapturingpatternsremainschallengeDifferenteffectivelyoftenrelysingleneglectsvaryinginfluencesdifferenttimeperiodsAdditionallymodelsfrequentlyutilizeeithergraphsrepresentlimitsabilityaddressoverlappingrelationshipsMoreoverstrugglefullyleadingexclusioncriticalinformationultimatelyresultingsuboptimalperformanceThuspaperintroducesAttention-BasedMulti-Modeldesignedemploysmulti-temporalgatedmulti-scaleinputsegmentsnon-linearutilizesparallelmulti-graphsfacilitateFurthermoreincorporatesself-attentionadaptivelylong-termExperimentsconductedrevealoutperformedbaselineachievingaveragereductions50%MAE28%RMSE646%MAPEacrossFlowPredictionUsingSelf-AttentionMulti-GraphNetworksnetworksconvolutional

Similar Articles

Cited By

No available data.