TTPNet: A Neural Network for Travel Time Prediction Based on Tensor Decomposition and Graph Embedding

Yibin Shen,Cheqing Jin,Jiaxun Hua,Dingjiang Huang
DOI: https://doi.org/10.1109/tkde.2020.3038259
IF: 9.235
2021-01-01
IEEE Transactions on Knowledge and Data Engineering
Abstract:Travel time prediction of a given trajectory plays an indispensable role in intelligent transportation systems. Although many prior researches have struggled for accurate prediction results, most of them achieve inferior performance due to insufficient feature extraction of travel speed and road network structure from the trajectory data, which confirms the challenges involved in this topic. To overcome those issues, we propose a novel neural Network for Travel Time Prediction based on tensor decomposition and graph embedding, named TTPNet, which can extract travel speed and representation of road network structure effectively from historical trajectories, as well as predict the travel time with better accuracy. Specifically, TTPNet consists of three components: the first module (Travel Speed Features Layer) leverages non-negative tensor decomposition to restore travel speed distributions on different roads in the previous hour, and integrates a CNN-RNN model to extract both long-term and short-term travel speed features of the query trajectory; the second module (Road Network Structure Features Layer) utilizes graph embedding to generate the representation of local and global road network structure; the last module (Deep LSTM Prediction Layer) completes the final predicting task. Empirical results over two real-world large-scale datasets show that our proposed TTPNet model can achieve significantly better performance and remarkable robustness.
computer science, information systems, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?