Cargando…

MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction

The spatial–temporal prediction of traffic flow is very important for traffic management and planning. The most difficult challenges of traffic flow prediction are the temporal feature extraction and the spatial correlation extraction of nodes. Due to the complex spatial correlation between differen...

Descripción completa

Detalles Bibliográficos
Autores principales: Huang, Xiaohui, Wang, Junyang, Lan, Yuanchun, Jiang, Chaojie, Yuan, Xinhua
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9861593/
https://www.ncbi.nlm.nih.gov/pubmed/36679639
http://dx.doi.org/10.3390/s23020841
_version_ 1784874880356843520
author Huang, Xiaohui
Wang, Junyang
Lan, Yuanchun
Jiang, Chaojie
Yuan, Xinhua
author_facet Huang, Xiaohui
Wang, Junyang
Lan, Yuanchun
Jiang, Chaojie
Yuan, Xinhua
author_sort Huang, Xiaohui
collection PubMed
description The spatial–temporal prediction of traffic flow is very important for traffic management and planning. The most difficult challenges of traffic flow prediction are the temporal feature extraction and the spatial correlation extraction of nodes. Due to the complex spatial correlation between different roads and the dynamic trend of time patterns, traditional forecasting methods still have limitations in obtaining spatial–temporal correlation, which makes it difficult to extract more valid information. In order to improve the accuracy of the forecasting, this paper proposes a multi-scale temporal dual graph convolution network for traffic flow prediction (MD-GCN). Firstly, we propose a gated temporal convolution based on a channel attention and inception structure to extract multi-scale temporal dependence. Then, aiming at the complexity of the traffic spatial structure, we develop a dual graph convolution module including the graph sampling and aggregation submodule (GraphSAGE) and the mix-hop propagation graph convolution submodule (MGCN) to extract the local correlation and global correlation between neighbor nodes. Finally, extensive experiments are carried out on several public traffic datasets, and the experimental results show that our proposed algorithm outperforms the existing methods.
format Online
Article
Text
id pubmed-9861593
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-98615932023-01-22 MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction Huang, Xiaohui Wang, Junyang Lan, Yuanchun Jiang, Chaojie Yuan, Xinhua Sensors (Basel) Article The spatial–temporal prediction of traffic flow is very important for traffic management and planning. The most difficult challenges of traffic flow prediction are the temporal feature extraction and the spatial correlation extraction of nodes. Due to the complex spatial correlation between different roads and the dynamic trend of time patterns, traditional forecasting methods still have limitations in obtaining spatial–temporal correlation, which makes it difficult to extract more valid information. In order to improve the accuracy of the forecasting, this paper proposes a multi-scale temporal dual graph convolution network for traffic flow prediction (MD-GCN). Firstly, we propose a gated temporal convolution based on a channel attention and inception structure to extract multi-scale temporal dependence. Then, aiming at the complexity of the traffic spatial structure, we develop a dual graph convolution module including the graph sampling and aggregation submodule (GraphSAGE) and the mix-hop propagation graph convolution submodule (MGCN) to extract the local correlation and global correlation between neighbor nodes. Finally, extensive experiments are carried out on several public traffic datasets, and the experimental results show that our proposed algorithm outperforms the existing methods. MDPI 2023-01-11 /pmc/articles/PMC9861593/ /pubmed/36679639 http://dx.doi.org/10.3390/s23020841 Text en © 2023 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Huang, Xiaohui
Wang, Junyang
Lan, Yuanchun
Jiang, Chaojie
Yuan, Xinhua
MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction
title MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction
title_full MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction
title_fullStr MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction
title_full_unstemmed MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction
title_short MD-GCN: A Multi-Scale Temporal Dual Graph Convolution Network for Traffic Flow Prediction
title_sort md-gcn: a multi-scale temporal dual graph convolution network for traffic flow prediction
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC9861593/
https://www.ncbi.nlm.nih.gov/pubmed/36679639
http://dx.doi.org/10.3390/s23020841
work_keys_str_mv AT huangxiaohui mdgcnamultiscaletemporaldualgraphconvolutionnetworkfortrafficflowprediction
AT wangjunyang mdgcnamultiscaletemporaldualgraphconvolutionnetworkfortrafficflowprediction
AT lanyuanchun mdgcnamultiscaletemporaldualgraphconvolutionnetworkfortrafficflowprediction
AT jiangchaojie mdgcnamultiscaletemporaldualgraphconvolutionnetworkfortrafficflowprediction
AT yuanxinhua mdgcnamultiscaletemporaldualgraphconvolutionnetworkfortrafficflowprediction