RL-CNN: Reinforcement Learning-designed Convolutional Neural Network for Urban Traffic Flow Estimation

Accurate prediction of urban traffic flows brings enormous advantages to big cities. Therefore, many urban traffic flow predictors have been developed in recent years. Urban traffic flow predictors aim to identify complex mobility patterns and capture urban traffic flow characteristics from large-scale historical datasets. Afterward, trained models are used to predict the future traffic volume in terms of the number of moving objects (e.g., vehicles). Convolutional Neural Networks (CNN) and other deep learning approaches are brilliant choices because of their ability to learn Spatio-temporal dependencies. Nevertheless, the extensive set of hyper-parameters tends to make these neural networks overly complex and challenging to design. In this work, we introduce RL-CNN, a framework based on Reinforcement Learning whose objective is to autonomously discover highperformance CNN architectures for the given traffic prediction task without human intervention. We examine the proposed RL-CNN model as a traffic flow estimator on a real-world and large-scale vehicular network dataset. We observe improvements of 5% - 10% in the average traffic flow prediction accuracy over the state-of-art approaches.

[1]  Quoc V. Le,et al.  Neural Architecture Search with Reinforcement Learning , 2016, ICLR.

[2]  Stephen Marshall,et al.  Activation Functions: Comparison of trends in Practice and Research for Deep Learning , 2018, ArXiv.

[3]  Dijiang Huang,et al.  Short term traffic flow prediction based on on-line sequential extreme learning machine , 2016, 2016 Eighth International Conference on Advanced Computational Intelligence (ICACI).

[4]  Hang Lei,et al.  Hyperparameter Optimization for Machine Learning Models Based on Bayesian Optimization , 2019 .

[5]  Gianluca Pollastri,et al.  A neural network approach to ordinal regression , 2007, 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence).

[6]  Yu Zheng,et al.  Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Prediction , 2016, AAAI.

[7]  D. T. Lee,et al.  Travel-time prediction with support vector regression , 2004, IEEE Transactions on Intelligent Transportation Systems.

[8]  Petro Liashchynskyi,et al.  Grid Search, Random Search, Genetic Algorithm: A Big Comparison for NAS , 2019, ArXiv.

[9]  Ramesh Raskar,et al.  Designing Neural Network Architectures using Reinforcement Learning , 2016, ICLR.

[10]  Geoffrey E. Hinton,et al.  Rectified Linear Units Improve Restricted Boltzmann Machines , 2010, ICML.

[11]  Susana Sargento,et al.  PortoLivingLab: An IoT-Based Sensing Platform for Smart Cities , 2018, IEEE Internet of Things Journal.

[12]  Wei-Chiang Hong,et al.  Traffic flow forecasting by seasonal SVR with chaotic simulated annealing algorithm , 2011, Neurocomputing.

[13]  Zili Zhang,et al.  A distributed spatial-temporal weighted model on MapReduce for short-term traffic flow forecasting , 2016, Neurocomputing.

[14]  Nicholas G. Polson,et al.  Deep Learning Predictors for Traffic Flows , 2016 .

[15]  Cyrus Shahabi,et al.  Graph Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting , 2017, ArXiv.

[16]  Tianqi Chen,et al.  XGBoost: A Scalable Tree Boosting System , 2016, KDD.

[17]  Susana Sargento,et al.  Reinforcement Learning-designed LSTM for Trajectory and Traffic Flow Prediction , 2021, 2021 IEEE Wireless Communications and Networking Conference (WCNC).

[18]  Yunpeng Wang,et al.  Large-Scale Transportation Network Congestion Evolution Prediction Using Deep Learning Theory , 2015, PloS one.

[19]  Der-Horng Lee,et al.  Short-term freeway traffic flow prediction : Bayesian combined neural network approach , 2006 .

[20]  Fei-Yue Wang,et al.  Traffic Flow Prediction With Big Data: A Deep Learning Approach , 2015, IEEE Transactions on Intelligent Transportation Systems.

[21]  Wenhao Huang,et al.  Deep Architecture for Traffic Flow Prediction: Deep Belief Networks With Multitask Learning , 2014, IEEE Transactions on Intelligent Transportation Systems.