Computer Science ›› 2021, Vol. 48 ›› Issue (10): 177-184.doi: 10.11896/jsjkx.200800077

• Database & Big Data & Data Science • Previous Articles     Next Articles

Urban Traffic Flow Completion with Multi-view Attention Mechanism

KANG Yan, CHEN Tie, LI Hao, YANG Bing, ZHANG Ya-chuan, BU Rong-jing   

  1. School of Software,Yunnan University,Kunming 650500,China
  • Received:2020-08-12 Revised:2020-11-11 Online:2021-10-15 Published:2021-10-18
  • About author:KANG Yan,born in 1972,Ph.D,asso-ciate professor.Her main research interests include software engineering,system optimization,big data processing and mining.
  • Supported by:
    Research on Dynamic Multi-objective Requirement Optimization based on Transfer Learning(61762092) and Major Science and Technology Project in Yunnan Province(202002AB080001).

Abstract: Traffic flow information is an important basis for intelligent transportation systems and urban computing.Traffic flow data is a new type of time series data.Due to the data collection method and the influence of external complex factors,the phenomenon of data loss is common and unavoidable.How to effectively mine the spatial-temporal characteristics of traffic flow data and the correlation between the data becomes the key to improve the missing data completion accuracy.Traditional statistical methods cannot meet the increasingly complex data requirements,and the application of deep learning promotes the development of missing data completion methods to higher accuracy.The article deeply analyzes the spatial-temporal characteristics of traffic flow,makes assumptions about the missing traffic flow,and proposes a UMAtNet (U-net with Multi-view Attention Mechanisms) traffic flow complement model.The model fuses closeness,trend and period time data with spatial data,and adopts diffe-rent data correlation measurement methods to fuse a multi-view attention mechanism,which can optimize the impact of the model on the spatial correlation of missing data.In order to verify the model,we use the open source data set of Beijing traffic data in the experiment,and analyzes in detail the influence of each part of the model and the loss function on the completion accuracy.The experimental results show that the fusion of UMAtNet and corresponding components can further improve the completion accuracy.

Key words: Attention mechanism, Completion of traffic flow, Multi-view, Spatial-temporal characteristics, U-Net

CLC Number: 

  • TP311
[1]LU Y,HU J,XU J,et al.Urban Traffic Flow Forecasting Based on Adaptive Hinging Hyperplanes[C]//International Confe-rence on Artificial Intelligence and Computational Intelligence.Berlin,Heidelberg:Springer,2009:658-667.
[2]LV Y,DUAN Y,KANG W,et al.Traffic Flow Prediction With Big Data:A Deep Learning Approach[J].IEEE Transactions on Intelligent Transportation Systems,2015,16(2):865-873.
[3]SMITH B,SCHERER W,CCONKLIN J.Exploring ImputationTechniques for Missing Data in Transportation Management Systems[J].Transportation Research Record Journal of the Transportation Research Board,2003,1836:132-142.
[4]HAN W G,WANG J F,HU J J.Interpolation Method of Mis-sing Value in Traffic Flow Data[J].Journal of Transport information and safety,2005,23(1):39-42.
[5]BOX G E P,JENKINS G M,REINSELG C,et al.Time Series Analysis:Forecasting and Control,5th Edition[J].Journal of the Operational Research Society,2015,22(2):199-201.
[6]MING Z,SATISH S,PAWAN L.Matching Patterns for Updating Missing Values of Traffic Counts[J].Transportation Planning and Technology,2006,29:141-156.
[7]ZHONG M,LINGRAS P,SHARMA S.Estimation of missing traffic counts using factor,genetic,neural,and regression techniques[J].Transportation Research Part C Emerging Technologies,2004,12(2):139-166.
[8]TAN H,FENG G,FENG J,et al.A Tensor Based Method for Missing Traffic Data Completion[J].Transportation Research Part Cemerging Technologies,2013,12(2):139-166.
[9]RAN B,TAN H,WU Y,et al.Tensor based missing traffic data completion with spatial-temporal correlation[J].Statistical Mechanics and its Applications,2016,446:54-63.
[10]TIAN Y,ZHANG K,LI J,et al.LSTM-based traffic flow prediction with missing data[J].Neurocomputing,2018,318(27):297-305.
[11]KU W C,JAGADEESH G R,PRAKASH A,et al.A Clustering-Based Approach for Data-Driven Imputation of Missing Traffic Data[C]//IEEE Forum on Integrated and Sustainable Transportation Systems.IEEE,2016.
[12]YU F,WEI D,ZHANG S,et al.3D CNN-based Accurate Prediction for Large-scale Traffic Flow[C]//2019 4th International Conference on Intelligent Transportation Engineering (ICITE).2019:99-103.
[13]LI Z,ZHENG H,FENG X.3D Convolutional Generative Adversarial Networks for Missing Traffic Data Completion[C]//2018 10th International Conference on Wireless Communications and Signal Processing (WCSP).2018.
[14]WANG L,LI M,YAN J Q,et al,Urban Traffic Flow Data Recovery Method Based on Generative Adversarial Network[J].Journal of Transportation Systems Engineering and Information Technology.2018,18(6):63-71.
[15]YUAN Y Y,KANG Y,LI H,et al.Timing traffic flow datacompletion based on ST-DCGAN[J].Computer Engineering and Applications,2020,56(15):140-146.
[16]HONG Y L,BIN J,YI X,et al.Coherent Semantic Attention for Image Inpainting [C]//Proceedings of the IEEE/CVF International Conference on Computer Vision.2019:4170-4179.
[17]GUILIN L,FITSUM A,KEVIN J,et al.Image Inpainting forIrregular Holes Using Partial Convolutions[C]//European Con-ference on Computer Vision.2018(11):89-105.
[18]WEN X,LI T,HAN Z,et al.Point Cloud Completion by Skip-attention Network with Hierarchical Folding[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).IEEE,2020.
[19]OLAF R,PHILIPP F,THOMAS B.U-Net:Convolutional Networks for Biomedical Image Segmentation[C]//International Conference on Medical Image Computing and Computer-assisted Intervention.Springer,Cham,2015:234-241.
[20]ZHANG J,ZHENG Y,QI D,et al.Predicting Citywide Crowd Flows Using Deep Spatio-Temporal Residual Networks[J].Artificial Intelligence,2018,259:147-166.
[21]WANG L,GENG X,MA X,et al.Crowd Flow Prediction byDeep Spatio-Temporal Transfer Learning[J].arXiv:1802.00386,2018.
[22]VU D L,VU N T.Vietnamese speech recognition using Dyna-mic Time Warping and Coefficient of Correlation[C]//International Conference on Control,Automation and Information Sciences (ICCAIS).2013:64-67.
[23]TANG R,KANAMORI R,YAMAMOTO T.Short-Term Ur-ban Link Travel Time Prediction using Dynamic Time Warping with Disaggregate Probe Data[J].IEEE Access,2019,99:1.
[24]ASHISH V,NOAM S,NIKI P,et al.Attention is All you Need[C]//Conference and Workshop on Neural Information Proces-sing Systems.2017:5998-6008.
[25]AHMED K,KESKAR N S,SOCHER R.Weighted transformer network for machine translation[J].arXiv:1711.02132,2017.
[26]ZHU G,ZHANG L,SHEN P,et al.Continuous Gesture Segmentation and Recognition Using 3DCNN and Convolutional LSTM[J].IEEE Transactions on Multimedia,2018,21:1011-1021.
[27]LEA C,FLYNN M D,VIDAL R,et al.Temporal convolutional networks for action segmentation and detection[C]//Procee-dings of the IEEE Conference on Computer Vision and Pattern Recognition.2017:156-165.
[28]XIONG T,QI Y,ZHANG W B.Short-term Traffic Flow Prediction Based on DCGRU-RF Model for Road Network[J].Computer Science,2020,47(5):84-89.
[29]RONG B,WU Z H,LIU X H,et al.Flow Prediction of Traffic Stations Based on Spatio-Temporal Multi-Graph Convolutional Network[J].Computer Engineering,2020,46(5):26-33.
[30]KONG F Y,ZHOU Y F,CHEN G.Traffic Flow PredictionMethod Based on Spatio-Temporal Feature Mining[J].Compu-ter Science,2019,46(7):322-326.
[1] ZHOU Fang-quan, CHENG Wei-qing. Sequence Recommendation Based on Global Enhanced Graph Neural Network [J]. Computer Science, 2022, 49(9): 55-63.
[2] DAI Yu, XU Lin-feng. Cross-image Text Reading Method Based on Text Line Matching [J]. Computer Science, 2022, 49(9): 139-145.
[3] ZHOU Le-yuan, ZHANG Jian-hua, YUAN Tian-tian, CHEN Sheng-yong. Sequence-to-Sequence Chinese Continuous Sign Language Recognition and Translation with Multi- layer Attention Mechanism Fusion [J]. Computer Science, 2022, 49(9): 155-161.
[4] XIONG Li-qin, CAO Lei, LAI Jun, CHEN Xi-liang. Overview of Multi-agent Deep Reinforcement Learning Based on Value Factorization [J]. Computer Science, 2022, 49(9): 172-182.
[5] RAO Zhi-shuang, JIA Zhen, ZHANG Fan, LI Tian-rui. Key-Value Relational Memory Networks for Question Answering over Knowledge Graph [J]. Computer Science, 2022, 49(9): 202-207.
[6] LI Bin, WAN Yuan. Unsupervised Multi-view Feature Selection Based on Similarity Matrix Learning and Matrix Alignment [J]. Computer Science, 2022, 49(8): 86-96.
[7] ZHU Cheng-zhang, HUANG Jia-er, XIAO Ya-long, WANG Han, ZOU Bei-ji. Deep Hash Retrieval Algorithm for Medical Images Based on Attention Mechanism [J]. Computer Science, 2022, 49(8): 113-119.
[8] SUN Qi, JI Gen-lin, ZHANG Jie. Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection [J]. Computer Science, 2022, 49(8): 172-177.
[9] YAN Jia-dan, JIA Cai-yan. Text Classification Method Based on Information Fusion of Dual-graph Neural Network [J]. Computer Science, 2022, 49(8): 230-236.
[10] WANG Ming, PENG Jian, HUANG Fei-hu. Multi-time Scale Spatial-Temporal Graph Neural Network for Traffic Flow Prediction [J]. Computer Science, 2022, 49(8): 40-48.
[11] JIANG Meng-han, LI Shao-mei, ZHENG Hong-hao, ZHANG Jian-peng. Rumor Detection Model Based on Improved Position Embedding [J]. Computer Science, 2022, 49(8): 330-335.
[12] JIN Fang-yan, WANG Xiu-li. Implicit Causality Extraction of Financial Events Integrating RACNN and BiLSTM [J]. Computer Science, 2022, 49(7): 179-186.
[13] XIONG Luo-geng, ZHENG Shang, ZOU Hai-tao, YU Hua-long, GAO Shang. Software Self-admitted Technical Debt Identification with Bidirectional Gate Recurrent Unit and Attention Mechanism [J]. Computer Science, 2022, 49(7): 212-219.
[14] PENG Shuang, WU Jiang-jiang, CHEN Hao, DU Chun, LI Jun. Satellite Onboard Observation Task Planning Based on Attention Neural Network [J]. Computer Science, 2022, 49(7): 242-247.
[15] ZHANG Ying-tao, ZHANG Jie, ZHANG Rui, ZHANG Wen-qiang. Photorealistic Style Transfer Guided by Global Information [J]. Computer Science, 2022, 49(7): 100-105.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!