基于KNN-LSTM的短时交通流预测

    Short-term Traffic Flow Prediction Based on KNN-LSTM

    • 摘要: 针对现有预测模型无法在交通大数据中提取交通流序列的内部规律,且未能充分利用交通流的时空相关性以实现高精度预测的问题,提出了一种基于K-最近邻(K-nearest neighbor,KNN)与长短时记忆(long short term memory,LSTM)网络模型相结合的短时交通流预测模型.采用KNN算法选择路网中与预测站点时空相关的检测站,以选择的检测站的交通流序列构造数据集,将其输入LSTM模型中进行训练及测试,并通过美国交通研究数据实验室的真实交通数据对提出的模型进行验证.结果表明:与现有的交通预测模型相比,该方法能更好地提取交通流序列的时空特性,预测准确率平均可提高12.28%,可为交通诱导与控制提供必要的依据.

       

      Abstract: To solve the problem that existing prediction models cannot extract the internal rules of the traffic flow in traffic big data and fail to make full use of spatiotemporal correlation characteristics to achieve high accurate prediction, a short-term traffic flow prediction model was proposed based on K-nearest neighbor (KNN)and long short term memory (LSTM). The KNN algorithm was used to select the stations that were related to the test station in the road network. The traffic flow data in selected stations were used as the training and testing datasets for the LSTM model. The proposed model was verified by real traffic data from the Transportation Research Data Lab in USA. Results show that the proposed method can better extract the spatiotemporal characteristics of traffic flow sequences, and the prediction accuracy can be improved by 12.28% on average compared with the existing prediction model, which can provide the necessary basis for traffic guidance and control.

       

    /

    返回文章
    返回