MTTPRE: A Multi-Scale Spatial-Temporal Model for Travel Time Prediction

Travel time prediction is a critical task in intelligent transportation system and location-based service. Existing studies build models based on the features extracted from trajectories, but few of them consider the sparsity of trajectory data from both temporal and spatial dimensions, as well as the spatial structure and heterogeneity. To address these issues, we propose a novel Multi-scale spatial-temporal model for Travel Time Prediction, abbreviated as MTTPRE. Specifically, the study area is represented as a flexible Voronoi graph according to a variable-sized partition scheme and the missing features on it are recovered via a spatial-temporal context-based method. Subsequently, a geospatial network with POI information is established to represent the spatial structure based on the Voronoi graph. Next, the multi-dimensional traffic condition features and graph-trajectory-POI multilevel features are extracted as spatial-temporal features. Finally, these features are fed into a hierarchical multi-task learning layer to complete the travel time prediction task. Extensive experiments on two real-world datasets show that the MTTPRE outperforms all the competitors with significant improvement and remarkable robustness.