测试科学与仪器2021,Vol.12Issue(2):195-207,13.DOI:10.3969/j.issn.1674-8042.2021.02.009
基于小波去噪和LSTM模型的短时交通流预测
Short-time prediction for traffic flow based on wavelet de-noising and LSTM model
摘要
Abstract
Aiming at the problem that some existing traffic flow prediction models are only for a single road segment and the model input data are not pre-processed,a heuristic threshold algorithm is used to de-noise the original traffic flow data after wavelet decomposition.The correlation coefficients of road traffic flow data are calculated and the data compression matrix of road traffic flow is constructed.Data de-noising minimizes the interference of data to the model,while the correlation analysis of road network data realizes the prediction at the road network level.Utilizing the advantages of long short term memory (LSTM)network in time series data processing,the compression matrix is input into the constructed LSTM model for short-term traffic flow prediction.The LSTM-1 and LSTM-2 models were respectively trained by de-noising processed data and original data.Through simulation experiments,different prediction times were set,and the prediction results of the prediction model proposed in this paper were compared with those of other methods.It is found that the accuracy of the LSTM-2 model proposed in this paper increases by 10.278% on average compared with other prediction methods,and the prediction accuracy reaches 95.58%,which proves that the short-term traffic flow prediction method proposed in this paper is efficient.关键词
短时交通流预测/深度学习/小波去噪/路网矩阵压缩/长短时记忆(LSTM)网络Key words
short-term traffic flow prediction/deep learning/wavelet denoising/network matrix compression/long short term memory (LSTM)network引用本文复制引用
王庆荣,李彤伟,朱昌锋..基于小波去噪和LSTM模型的短时交通流预测[J].测试科学与仪器,2021,12(2):195-207,13.基金项目
National Natural Science Foundation of China(No.71961016) (No.71961016)
Planning Fund for the Humanities and Social Sciences of the Ministry of Education(Nos.15XJAZH002,18YJAZH148) (Nos.15XJAZH002,18YJAZH148)
Natural Science Foundation of Gansu Province(No.18JR3RA125) (No.18JR3RA125)