Public transportation prediction with convolutional neural networks

Abstract

International audience; Good, efficient and reliable public transportation systems are of crucial importance for all major cities today. In this paper, we propose a concrete solution to a particular problem: improve the prediction of the bus arrival time at each bus stop station on a given itinerary, by taking to account global and local traffic contexts. The main principle consists of modeling the traffic data as an image structure, adapted for applying CNN deep neural networks. The results obtained shows that the proposed approach outperforms traditional machine learning techniques, such as OLS (Ordinary Least Squares) or SVR (Support Vector Regression) with different kernels (RBF or Polynomial), with more than 18% better accuracy prediction, while being computationally faster.