Construction of traffic state vector using mutual information for short-term traffic flow prediction.

Highlights • A new method of constructing traffic state vectors is proposed. • Variables are selected using spatio-temporal correlations based on Mutual Information. • Weighted distance is used in selecting K nearest neighbors. • The proposed method can improve the prediction accuracy of data-driven models. Abstract Short-term traffic flow prediction is an integral part in most of Intelligent Transportation Systems (ITS) research and applications. Many researchers have already developed various methods that predict the future traffic condition from the historical database. Nevertheless, there has not been sufficient effort made to study how to identify and utilize the different factors that affect the traffic flow. In order to improve the performance of short-term traffic flow prediction, it is necessary to consider sufficient information related to the road section to be predicted. In this paper, we propose a method of constructing traffic state vectors by using mutual information (MI). First, the variables with different time delays are generated from the historical traffic time series, and the spatio-temporal correlations between the road sections in urban road network are evaluated by the MI. Then, the variables with the highest correlation related to the target traffic flow are selected by using a greedy search algorithm to construct the traffic state vector. The K-Nearest Neighbor (KNN) model is adapted for the application of the proposed state vector. Experimental results on real-world traffic data show that the proposed method of constructing traffic state vector provides good prediction accuracy in short-term traffic prediction. [ABSTRACT FROM AUTHOR]