Algorithm Research & Explore
|
770-773,779

Traffic flow prediction model based on attention mechanism of temporal graph

Yao Xiaomin
Zhang Xinlan
Zhang Zhenguo
Intelligent Information Processing Lab, College of Engineering, Yanbian University, Yanji Jilin 133002, China

Abstract

Traffic condition prediction is an important part of intelligent transportation system, and traffic flow is the most direct embodiment of traffic condition. Therefore, traffic flow prediction has important application value. On the one hand, the roads in the city have spatial topological properties, on the other hand, the traffic flow changes dynamically with time. Therefore, the key to the prediction of traffic flow is to model the time and space dependence in the data. In view of this characteristic, this paper used neural network model and attention mechanism to explore the temporal and spatial dependence relationship in traffic flow data, and proposed a traffic flow prediction model based on time map attention. In terms of spatial dependence, it used a learning algorithm combining graph convolution network and attention to assign different weights to nodes with diffe-rent influence degrees, and added node adaptive learning to effectively extract spatial features. In terms of time dependence, it used the temporal convolution network to extract the temporal features, and expanded the sensing domain by expanding convolution, so as to capture the features of longer time series data. A spatial-temporal network layer was composed of graph attention network and time convolution network, which was finally connected to the output layer to output the prediction results. The model used the combination of graph convolution neural network and attention mechanism to extract spatial features, fully considered the spatial relationship between roads, and used temporal convolution network to capture temporal features. After experiments on two real datasets, it is found that it has good performance in the next 15, 30 and 60 minutes, and the results are better than the existing baselines.

Foundation Support

国家自然科学基金资助项目(62162062)

Publish Information

DOI: 10.19734/j.issn.1001-3695.2021.08.0344
Publish at: Application Research of Computers Printed Article, Vol. 39, 2022 No. 3
Section: Algorithm Research & Explore
Pages: 770-773,779
Serial Number: 1001-3695(2022)03-021-0770-04

Publish History

[2021-11-15] Accepted Paper
[2022-03-05] Printed Article

Cite This Article

姚晓敏, 张心蓝, 张振国. 基于时间图注意力的交通流量预测模型 [J]. 计算机应用研究, 2022, 39 (3): 770-773,779. (Yao Xiaomin, Zhang Xinlan, Zhang Zhenguo. Traffic flow prediction model based on attention mechanism of temporal graph [J]. Application Research of Computers, 2022, 39 (3): 770-773,779. )

About the Journal

  • Application Research of Computers Monthly Journal
  • Journal ID ISSN 1001-3695
    CN  51-1196/TP

Application Research of Computers, founded in 1984, is an academic journal of computing technology sponsored by Sichuan Institute of Computer Sciences under the Science and Technology Department of Sichuan Province.

Aiming at the urgently needed cutting-edge technology in this discipline, Application Research of Computers reflects the mainstream technology, hot technology and the latest development trend of computer application research at home and abroad in a timely manner. The main contents of the journal include high-level academic papers in this discipline, the latest scientific research results and major application results. The contents of the columns involve new theories of computer discipline, basic computer theory, algorithm theory research, algorithm design and analysis, blockchain technology, system software and software engineering technology, pattern recognition and artificial intelligence, architecture, advanced computing, parallel processing, database technology, computer network and communication technology, information security technology, computer image graphics and its latest hot application technology.

Application Research of Computers has many high-level readers and authors, and its readers are mainly senior and middle-level researchers and engineers engaged in the field of computer science, as well as teachers and students majoring in computer science and related majors in colleges and universities. Over the years, the total citation frequency and Web download rate of Application Research of Computers have been ranked among the top of similar academic journals in this discipline, and the academic papers published are highly popular among the readers for their novelty, academics, foresight, orientation and practicality.


Indexed & Evaluation

  • The Second National Periodical Award 100 Key Journals
  • Double Effect Journal of China Journal Formation
  • the Core Journal of China (Peking University 2023 Edition)
  • the Core Journal for Science
  • Chinese Science Citation Database (CSCD) Source Journals
  • RCCSE Chinese Core Academic Journals
  • Journal of China Computer Federation
  • 2020-2022 The World Journal Clout Index (WJCI) Report of Scientific and Technological Periodicals
  • Full-text Source Journal of China Science and Technology Periodicals Database
  • Source Journal of China Academic Journals Comprehensive Evaluation Database
  • Source Journals of China Academic Journals (CD-ROM Version), China Journal Network
  • 2017-2019 China Outstanding Academic Journals with International Influence (Natural Science and Engineering Technology)
  • Source Journal of Top Academic Papers (F5000) Program of China's Excellent Science and Technology Journals
  • Source Journal of China Engineering Technology Electronic Information Network and Electronic Technology Literature Database
  • Source Journal of British Science Digest (INSPEC)
  • Japan Science and Technology Agency (JST) Source Journal
  • Russian Journal of Abstracts (AJ, VINITI) Source Journals
  • Full-text Journal of EBSCO, USA
  • Cambridge Scientific Abstracts (Natural Sciences) (CSA(NS)) core journals
  • Poland Copernicus Index (IC)
  • Ulrichsweb (USA)