TS-STNN: Spatial-temporal neural network based on tree structure for traffic flow prediction

Yang Lv, Zhiqiang Lv, Zesheng Cheng, Zhanqi Zhu and Taha Hossein Rashidi

Transportation Research Part E: Logistics and Transportation Review, 2023, vol. 177, issue C

Abstract: Traffic flow prediction effectively supports the sustainable expansion and operation of modern transport networks, one of the emerging research areas in intelligent transportation systems. Currently, most common traffic flow prediction methods use deep learning spatial–temporal models based on graph convolution theory, which cannot deeply explore the spatial hierarchy and directional information of traffic flow data due to their structural characteristics. To address this problem, a spatial–temporal neural network based on tree structure (TS-STNN) is created to anticipate future traffic flow at a specific time at a target location. The principle of this method is to use the characteristics of the tree structure to construct a plane tree matrix with hierarchical and directional features, which is finally fused into a spatial tree matrix to extract the spatial information. Meanwhile, the temporal correlation of traffic flow data in the traffic network is analyzed by TS-STNN using Gated Recurrent Units (GRUs). By comparing with the existing baseline methods, it is verified that the TS-STNN model has high prediction accuracy in both Random Uniformly Distributed (RND) and Small-Scale Aggregation of Node Distributed (SSAND) scenarios. It is further demonstrated through ablation experiments that the developed tree convolution module greatly impacts the TS-STNN accuracy.

Date: 2023
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S1366554523002399
Full text for ScienceDirect subscribers only

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:transe:v:177:y:2023:i:c:s1366554523002399

Ordering information: This journal article can be ordered from
http://www.elsevier.com/wps/find/journaldescription.cws_home/600244/bibliographic
http://www.elsevier. ... 600244/bibliographic

DOI: 10.1016/j.tre.2023.103251

Access Statistics for this article

Transportation Research Part E: Logistics and Transportation Review is currently edited by W. Talley

More articles in Transportation Research Part E: Logistics and Transportation Review from Elsevier
Bibliographic data for series maintained by Catherine Liu ().