A K-nearest neighbor locally search regression algorithm for short-term traffic flow forecasting

Accurate short-term traffic flow forecasting is of great significance for real-time traffic control, guidance and management. The k-nearest neighbor (k-NN) model is a classic data-driven method which is relatively effective yet simple to implement for short-term traffic flow forecasting. For conventional prediction mechanism of k-NN model, the k nearest neighbors' outputs weighted by similarities between the current traffic flow vector and historical traffic flow vectors is directly used to generate prediction values, so that the prediction results are always not ideal. It is observed that there are always some outliers in k nearest neighbors' outputs, which may have a bad influences on the prediction value, and the local similarities between current traffic flow and historical traffic flows at the current sampling period should have a greater relevant to the prediction value. In this paper, we focus on improving the prediction mechanism of k-NN model and proposed a k-nearest neighbor locally search regression algorithm (k-LSR). The k-LSR algorithm can use locally search strategy to search for optimal nearest neighbors' outputs and use optimal nearest neighbors' outputs weighted by local similarities to forecast short-term traffic flow so as to improve the prediction mechanism of k-NN model. The proposed algorithm is tested on the actual data and compared with other algorithms in performance. We use the root mean squared error (RMSE) as the evaluation indicator. The comparison results show that the k-LSR algorithm is more successful than the k-NN and k-nearest neighbor locally weighted regression algorithm (k-LWR) in forecasting short-term traffic flow, and which prove the superiority and good practicability of the proposed algorithm.

[1]  Junwei Gao,et al.  Short-term traffic flow forecasting model of optimized BP neural network based on genetic algorithm , 2013, Proceedings of the 32nd Chinese Control Conference.

[2]  Zhaosheng Yang,et al.  Short-term traffic flow prediction model using particle swarm optimization–based combined kernel function-least squares support vector machine combined with chaos theory , 2016 .

[3]  Zhao Zhi-hong,et al.  K-nearest neighbor model of short-term traffic flow forecast , 2012 .

[4]  Jianhua Guo,et al.  Adaptive Kalman filter approach for stochastic short-term traffic flow rate prediction and uncertainty quantification , 2014 .

[5]  Andrew W. Moore,et al.  Locally Weighted Learning , 1997, Artificial Intelligence Review.

[6]  Li Pan,et al.  Predicting Short-Term Traffic Flow by Long Short-Term Memory Recurrent Neural Network , 2015, 2015 IEEE International Conference on Smart City/SocialCom/SustainCom (SmartCity).

[7]  Qiuchen Liu,et al.  An Improved K-nearest Neighbor Model for Short-term Traffic Flow Prediction , 2013 .

[8]  Rui Xue,et al.  Short-Term Bus Passenger Demand Prediction Based on Time Series Model and Interactive Multiple Model Approach , 2015 .

[9]  Hwasoo Yeo,et al.  Improvement of Search Strategy With K-Nearest Neighbors Approach for Traffic State Prediction , 2016, IEEE Transactions on Intelligent Transportation Systems.

[10]  Tao Zhang,et al.  Nonparametric regression for the short-term traffic flow forecasting , 2010, 2010 International Conference on Mechanic Automation and Control Engineering.

[11]  Gang Xiong,et al.  A k-nearest neighbor locally weighted regression method for short-term traffic flow forecasting , 2012, 2012 15th International IEEE Conference on Intelligent Transportation Systems.

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

[13]  Yunpeng Wang,et al.  A spatiotemporal correlative k-nearest neighbor model for short-term traffic multistep forecasting , 2016 .

[14]  Alessandro De Gloria,et al.  Time-Aware Multivariate Nearest Neighbor Regression Methods for Traffic Flow Prediction , 2015, IEEE Transactions on Intelligent Transportation Systems.

[15]  Mecit Cetin,et al.  Short-term traffic flow rate forecasting based on identifying similar traffic patterns , 2016 .

[16]  Bin Ran,et al.  Short-Term Traffic Prediction Based on Dynamic Tensor Completion , 2016, IEEE Transactions on Intelligent Transportation Systems.

[17]  Zuduo Zheng,et al.  Short-term traffic volume forecasting : a k-nearest neighbor approach enhanced by constrained linearly sewing principle component algorithm , 2014 .