arrow
返回

MuGIL: A Multi-Graph Interaction Learning Network for Multi-Task Traffic Prediction

delete2024-12-01
delete0
PRE
AI
S
Shuai Liu
H
Haiyang Yu
H
Han Jiang
Z
Zhenliang Ma
Z
Zhiyong Cui
Y
Yilong Ren *
DOI:10.1016/j.knosys.2024.112709delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Recently, multi-task traffic prediction has received increasing attention, as it enables knowledge sharing between heterogeneous variables or regions, thereby improving prediction accuracy while satisfying the prediction requirements of multi-source data in Intelligent Transportation Systems (ITS). However, current studies present two significant challenges. First, they often tend to construct specialized models for a limited set of predictive parameters, which results in a lack of generality. Second, modeling the graph-based multi-task interaction and message passing processes remains difficult due to the heterogeneity of graph structures arising from multi- source traffic data. To address these challenges, this paper proposes a Multi-Graph Interaction Learning Network (MuGIL), characterized by three key innovations: 1) A flexible end-to-end multi-task prediction framework that is generalizable for varied variables or scenarios; 2) A multi-source graph representation module that aligns heterogeneous information through semantic graphs; 3) A novel message passing mechanism for multi-task graph neural networks, which enables effective knowledge among tasks. The model is validated using data from California by comparing it with the state-of-the-art prediction models. The results show that the MuGIL model achieves better prediction performance than these baselines. Ablation experiments further highlight the critical role of the designed multi-source graph representation module and message passing mechanism in the model's success. The MuGIL model we have proposed is now open-sourced at the following link: https://github. com/trafficpre/MuGIL.
Keyword:
Multi-task learning
Graph neural networks
Message passing
Traffic prediction

期刊

K
Knowledge-Based Systems
IF:
7.6
论文数:
1.3W
被引数:
4.5W

机构

B
Beihang University
学者数:
5.2W
论文数: 4.1W
被引数: 37
引用论文

引用论文

AdapGL: An adaptive graph learning algorithm for traffic prediction based on spatiotemporal neural networks
err2022-06-01
err48
PREAI
errZhang, Wei; Zhu, Fenghua; Lv, Yisheng; Tan, Chang; Liu, Wen; Zhang, Xin; Wang, Fei-Yue
err分享
err收藏
A deep learning based multitask model for network-wide traffic speed prediction
err2020-07-01
err73
PREAI
errZhang, Kunpeng; Zheng, Liang; Liu, Zijian; Jia, Ning
err分享
err收藏
Location and time embedded feature representation for spatiotemporal traffic prediction
err2024-04-01
err9
PREAI
errLi, Wei; Liu, Xin; Tao, Wei; Zhang, Lei; Zou, Junhua; Pan, Yu; Pan, Zhisong
err分享
err收藏
err分享
err收藏
学者 查看更多内容