Hierarchical Neural Architecture Search for Travel Time Estimation

We propose a novel automated deep learning framework, namely Automated Spatio-Temporal Dual Graph Convolutional Networks (Auto-STDGCN), for travel time estimation. Specifically, a hierarchical neural architecture search approach is introduced to capture the joint spatio-temporal correlations of intersections and road segments, whose search space is composed of internal and external search space. In the internal search space, spatial graph convolution and temporal convolution operations are adopted to capture the spatio-temporal correlations of the dual graphs. In the external search space, the node-wise and edge-wise graph convolution operations from the internal architecture search are built to capture the interaction patterns between the intersections and road segments. We conduct several experiments on two real-world datasets, and the results demonstrate that Auto-STDGCN is significantly superior to the state-of-art methods.

[1]  Eric R. Ziegel,et al.  The Elements of Statistical Learning , 2003, Technometrics.

[2]  Zheng Wang,et al.  Learning to Estimate the Travel Time , 2018, KDD.

[3]  Haifeng Wang,et al.  ConSTGAT: Contextual Spatial-Temporal Graph Attention Network for Travel Time Estimation at Baidu Maps , 2020, KDD.

[4]  Yong Yu,et al.  AutoSTG: Neural Architecture Search for Predictions of Spatio-Temporal Graph✱ , 2021, WWW.

[5]  Depeng Jin,et al.  Dynamic Graph Convolutional Recurrent Network for Traffic Prediction: Benchmark and Solution , 2021, ACM Trans. Knowl. Discov. Data.

[6]  Yu Zheng,et al.  AutoST: Efficient Neural Architecture Search for Spatio-Temporal Prediction , 2020, KDD.

[7]  Guangyin Jin,et al.  Spatial-Temporal Dual Graph Neural Networks for Travel Time Estimation , 2021, ArXiv.

[8]  Cyrus Shahabi,et al.  Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting , 2017, ICLR.

[9]  Yiming Yang,et al.  DARTS: Differentiable Architecture Search , 2018, ICLR.

[10]  Joan Bruna,et al.  Community Detection with Graph Neural Networks , 2017, 1705.08415.

[11]  Cho-Jui Hsieh,et al.  Rethinking Architecture Selection in Differentiable NAS , 2021, ICLR.

[12]  Wei Cao,et al.  When Will You Arrive? Estimating Travel Time Based on Deep Neural Networks , 2018, AAAI.

[13]  Xianfeng Tang,et al.  A Simple Baseline for Travel Time Estimation using Large-scale Trip Data , 2019, ACM Trans. Intell. Syst. Technol..

[14]  Yu Liu,et al.  T-GCN: A Temporal Graph Convolutional Network for Traffic Prediction , 2018, IEEE Transactions on Intelligent Transportation Systems.

[15]  Zheng Wang,et al.  CompactETA: A Fast Inference System for Travel Time Prediction , 2020, KDD.