中文核心期刊
CSCD来源期刊
中国科技核心期刊
RCCSE中国核心学术期刊

重庆交通大学学报(自然科学版) ›› 2024, Vol. 43 ›› Issue (6): 39-46.DOI: 10.3969/j.issn.1674-0696.2024.06.06

• 交通+大数据人工智能 • 上一篇    下一篇

融合滞后极限学习机的IDBiLSTM短时交通流预测

张阳1,王梓良1,姚芳钰1,许浩越2,杨书敏3   

  1. (1.福建理工大学 交通运输学院,福建 福州 350118;2.哈尔滨理工大学 计算机科学与技术学院,黑龙江 哈尔滨 150080;3.同济大学 交通运输工程学院,上海 201804)
  • 收稿日期:2022-12-16 修回日期:2023-11-01 发布日期:2024-06-24

IDBiLSTM Short-Term Traffic Flow Prediction with Fused Hysteretic Extreme Learning Machine

ZHANG Yang1,WANG Ziliang1,YAO Fangyu1,XU Haoyue2,YANG Shumin3   

  1. (1. School of Transportation,Fujian University of Technology, Fuzhou 350118, Fujian, China; 2.School of Computer Science and Technology,Harbin University of Science and Technology,Harbin 150080, Heilongjiang, China; 3.School of Transportation Engineering,Tongji University,Shanghai 201804,China)
  • Received:2022-12-16 Revised:2023-11-01 Published:2024-06-24

摘要: 深度学习短时交通流预测中,存在数据处理实时性较弱,以及算法对交通流数据的复用和修正能力不足导致预测性能较差的问题。针对这一问题,提出一种融合滞后极限学习机的深度双向长短时记忆神经网络短时交通流预测方法。首先,引入权值共享机制对双向长短时记忆网络模型进行结构优化,在模型训练过程中不断进行权重更新和偏置更新,从而充分利用逆序逆转数据增强数据的复用和修正能力;其次,为了进一步提高算法实时性,引入极限学习机模型,并在其神经元激活函数中嵌入生物神经系统中的滞后参数进行优化,加速了运算效率,提升算法的整体实时性。实验结果表明:提出的方法预测精度和算法实时性均有提升,与经典方法CNN-BiLSTM和多元集合CNN-LSTM相比,平均绝对误差分别减少了6.82、6.47,计算速度分别提高了12、19 s,具备良好的短时交通流预测能力和实时性。

Abstract: In deep learning short-term traffic flow prediction, there are some problems, such as the weak real-time performance in data processing and the poor prediction performance caused by the insufficient reuse and correction ability of the algorithm for traffic flow data. To address these problems, a kind of short term traffic flow prediction method using deep bidirectional long short-term memory neural network which incorporated a hysteretic extreme learning machine was proposed. Firstly, a weight-sharing mechanism was introduced to optimize the structure of the bidirectional long and short-term memory network model, and the weights and biases were continuously updated during the training process of the model, so as to make full use of the reverse-order data to enhance the reuse and correction ability of the data. Secondly, in order to further improve the real-time performance of the algorithm, the extreme learning machine model was introduced and the hysteretic parameters of the biological neural system were embedded in the activation function of its neurons, which optimized the computational efficiency and enhanced the overall real-time performance of the algorithm. The experimental results show that the proposed method has improved prediction accuracy and algorithm real-time performance, its average absolute error is respectively reduced by 6.82 and 6.47 and its computation speed is respectively improved by 12 and 19 seconds, compared with the classical method CNN-BiLSTM and multivariate ensemble CNN-LSTM, which has good short-term traffic flow prediction capability and real-time performance.