TSTNet: A Sequence to Sequence Transformer Network for Spatial-Temporal Traffic Prediction

Song，Xiaozhuang1,2; Wu，Ying1,2,3; Zhang，Chenhan1,2,4

10.1007/978-3-030-86362-3_28

2021

0302-9743

1611-3349

LECT NOTES ARTIF INT

12891

343-354

Making accurate traffic forecasting is of great importance in smart city-related researches. However, as the traffic features like traffic speed have a complex spatial-temporal characteristics, how to build an accurate traffic prediction model is still an open challenge. In this work, we propose TSTNet, a Sequence to Sequence (Seq2Seq) spatial-temporal traffic prediction model. TSTNet adopts Graph Attention Network (GAT), which can learn the spatial feature aggregation, to build spatial dependency. For temporal dependency, TSTNet applies a Seq2Seq Transformer structure to establish temporal dependency. As a GAT layer’s operation only aggregate the attribute information for neighbor nodes, it does not involve any spatial positional information. Similarly, if we apply the Transformer model on sequence learning tasks, the Transformer model also does not involve any temporal positional information as it does not know the exact time slot of different inputs. To solve the above problems, TSTNet implements a spatial-temporal embedding method to obtain the spatial-temporal positional representation for each input data. We evaluate TSTNet on traffic speed prediction tasks with other baselines upon two real-world datasets, the results show that TSTNet outperforms all the baseline models.

Sequence to sequence model Spatial-temporal forecasting Time series data

第一 ; 通讯

英语

CPCI-S ; EI

Computer Science

Computer Science, Artificial Intelligence ; Computer Science, Theory & Methods

WOS:000711965200028

20213910941671

Speed ; Traffic control

2-s2.0-85115444206

Scopus

1.Southern University of Science and Technology,Shenzhen,China
2.Guangdong Provincial Key Laboratory of Brain-inspired Intelligent Computation,Shenzhen,China
3.University of Leeds,Leeds,United Kingdom
4.University of Technology Sydney,Sydney,Australia

Song，Xiaozhuang,Wu，Ying,Zhang，Chenhan. TSTNet: A Sequence to Sequence Transformer Network for Spatial-Temporal Traffic Prediction[C],2021:343-354.