华南理工大学学报(自然科学版) ›› 2020, Vol. 48 ›› Issue (11): 107-113,122.doi: 10.12141/j.issn.1000-565X.190276

• 计算机科学与技术 • 上一篇    下一篇

基于时间卷积神经网络的短时交通流预测算法

袁华 陈泽濠   

  1. 华南理工大学 计算机科学与工程学院,广东 广州 510640
  • 收稿日期:2019-05-19 修回日期:2020-05-29 出版日期:2020-11-25 发布日期:2020-11-05
  • 通信作者: 袁华(1969-),女,博士,副教授,主要从事计算机应用技术研究。 E-mail:hyuan@scut.edu.cn
  • 作者简介:袁华(1969-),女,博士,副教授,主要从事计算机应用技术研究。
  • 基金资助:
    广东省自然科学基金资助项目 (2018A030313309,2015A030308017)

Short-term Traffic Flow Prediction Based on Temporal Convolutional Networks

YUAN Hua CHEN Zehao   

  1. School of Computer Science & Engineering,South China University of Technology,Guangzhou 510640,Guangdong,China
  • Received:2019-05-19 Revised:2020-05-29 Online:2020-11-25 Published:2020-11-05
  • Contact: 袁华(1969-),女,博士,副教授,主要从事计算机应用技术研究。 E-mail:hyuan@scut.edu.cn
  • About author:袁华(1969-),女,博士,副教授,主要从事计算机应用技术研究。
  • Supported by:
    Supported by the Natural Science Foundation of Guangdong Province (2018A030313309,2015A030308017)

摘要: 短时交通流预测是智能交通系统实现交通控制与交通诱导的关键所在。传统一维卷积神经网络 (CNN) 在短时交通流预测上难以获取长时记忆,同时存在信息泄露的问题。文中提出扩张 - 因果卷积神经网络 (DCFCN),引入扩张卷积来增加感受野大
小,获取序列的长时记忆; 同时,引入因果卷积来解决信息泄露问题。DCFCN 由 6 层卷积层堆叠而成,每层通过 Padding 的方式实现因果卷积,扩张系数逐层呈指数增长。实验结果表明,文中提出的 DCFCN 在短时交通流预测上优于其他对比模型,且在 GPU上计算效率明显提升。

关键词: 短时交通流预测, 时间卷积网络, 深度学习

Abstract: Short-term traffic flow prediction is the key for Intelligent Transport Systems (ITS) to achieve traffic control and traffic guidance. Traditional convolutional neural networks are difficult to obtain long-term memory in short-term traffic flow prediction,and there is information “leakage”from future to past. Dilated-Causal Fully Convolutional Networks (DCFCN),which introduces dilated convolutions to increase the reception field and ob-tain long term memory of sequences,was proposed in this study. At the same time,the causal convolutional was introduced into DCFCN to solve the problem of information“leakage”. DCFCN is composed of 6 convolutional la-yers,each layer achieves causal convolution by padding,and the dilated increases exponentially with the depth of the network. Experiments show that the proposed DCFCN is superior to other comparison models in short-term tra-ffic flow prediction,and the computational efficiency on the GPU is significantly improved.

Key words: short-term traffic flow prediction, temporal convolutional network, deep learning

中图分类号: