UCL Discovery
UCL home » Library Services » Electronic resources » UCL Discovery

A Task-Oriented Spatial Graph Structure Learning Method for Traffic Forecasting

Wang, T; Zhao, S; Jia, W; Shi, D; (2025) A Task-Oriented Spatial Graph Structure Learning Method for Traffic Forecasting. IEEE Transactions on Intelligent Transportation Systems pp. 1-10. 10.1109/TITS.2025.3537637. Green open access

[thumbnail of A_Task-Oriented_Spatial_Graph_Structure_Learning_Method_for_Traffic_Forecasting.pdf]
Preview
Text
A_Task-Oriented_Spatial_Graph_Structure_Learning_Method_for_Traffic_Forecasting.pdf - Accepted Version

Download (1MB) | Preview

Abstract

Traffic forecasting is the foundation of intelligent transportation systems (ITS). In recent, graph neural networks (GNNs) have successfully captured spatial-temporal dependencies to forecast traffic conditions by transforming traffic data in the graph domain. Nevertheless, the existing methods focus only on learning informative graph representations and fail to model informative graph structures, which hinders the capture of dynamic spatial-temporal dependencies caused by dynamic factors such as weather, accidents, and special events. In this paper, we propose a novel task-oriented Spatial Graph Structure Learning (SGSL) method, which aims to capture dynamic dependencies by jointly learning graph structures and graph representations. Compared to methods that use spectral graph representations, we exploit a learnable spatial graph to effectively model dynamic dependencies in traffic data. Moreover, we directly define graph convolutions on spatial relations to specify different edge weights when aggregating the information of spatial neighbours. Thus, the graph structure alterations, i.e., the relation changes, and the time-varying weights of relations can be encapsulated, thereby effectively representing dynamic dependencies. The gradient descent strategy is introduced to periodically learn a spatial graph through joint optimization with a newly designed deep graph learning model named GAT-nLSTM. In this manner, the intrinsic behaviours of nodes are learned to capture correlations across periods. Notably, the optimization process is performed under the traffic forecasting constraint to ensure that the learned spatial graph is specific to this task. Compared with those of state-of-the-art baselines, the experimental results obtained on real-world traffic datasets show significant improvement, which verifies the superiority of the proposed SGSL.

Type: Article
Title: A Task-Oriented Spatial Graph Structure Learning Method for Traffic Forecasting
Open access status: An open access version is available from UCL Discovery
DOI: 10.1109/TITS.2025.3537637
Publisher version: https://doi.org/10.1109/tits.2025.3537637
Language: English
Additional information: This version is the author accepted manuscript. For information on re-use, please refer to the publisher’s terms and conditions.
Keywords: Graph structure learning, traffic forecasting, spatial-temporal dependencies, spatial graph, graph representation
UCL classification: UCL
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences > Faculty of Population Health Sciences > Institute of Health Informatics
UCL > Provost and Vice Provost Offices > School of Life and Medical Sciences > Faculty of Population Health Sciences > Institute of Health Informatics > CHIME
URI: https://discovery.ucl.ac.uk/id/eprint/10205456
Downloads since deposit
Loading...
15Downloads
Download activity - last month
Loading...
Download activity - last 12 months
Loading...
Downloads by country - last 12 months
Loading...

Archive Staff Only

View Item View Item