Deep and Embedded Learning Approach for Traffic Flow Prediction in Urban Informatics

• Published in: IEEE transactions on intelligent transportation systems Vol. 20; no. 10; pp. 3927 - 3939
• Main Authors: Zheng, Zibin, Yang, Yatao, Liu, Jiahao, Dai, Hong-Ning, Zhang, Yan
• Format: Journal Article
• Language: English
• Published: New York IEEE 01-10-2019; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Artificial neural networks; Deep learning; Embedding; embedding neural networks; Informatics; Learning; Meteorology; Predictive models; Roads; Sensors; Traffic congestion; Traffic flow; traffic flow prediction; Traffic information; Two dimensional flow; Urban areas; Urban informatics; Weather
• ISSN: 1524-9050; 1558-0016
• DOI: 10.1109/TITS.2019.2909904

Traffic flow prediction has received extensive attention recently, since it is a key step to prevent and mitigate traffic congestion in urban areas. However, most previous studies on traffic flow prediction fail to capture fine-grained traffic information (like link-level traffic) and ignore the impacts from other factors, such as route structure and weather conditions. In this paper, we propose a deep and embedding learning approach (DELA) that can help to explicitly learn from fine-grained traffic information, route structure, and weather conditions. In particular, our DELA consists of an embedding component, a convolutional neural network (CNN) component and a long short-term memory (LSTM) component. The embedding component can capture the categorical feature information and identify correlated features. Meanwhile, the CNN component can learn the 2-D traffic flow data while the LSTM component has the benefits of maintaining a long-term memory of historical data. The integration of the three models together can improve the prediction accuracy of traffic flow. We conduct extensive experiments on realistic traffic flow dataset to evaluate the performance of our DELA and make comparison with other existing models. The experimental results show that the proposed DELA outperforms the existing methods in terms of prediction accuracy.