kth.sePublications KTH
Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
MuGIL: A Multi-Graph Interaction Learning Network for Multi-Task Traffic Prediction
School of Transportation Science and Engineering, Beihang University, Beijing, 100191, China; State Key Laboratory of Intelligent Transportation System, Beijing, 100191, China.
School of Transportation Science and Engineering, Beihang University, Beijing, 100191, China; State Key Laboratory of Intelligent Transportation System, Beijing, 100191, China; Zhongguancun Laboratory, Beijing, 100094, China; Hangzhou Innovation Institute, Beihang University, Hangzhou, 310056, China.
School of Transportation Science and Engineering, Beihang University, Beijing, 100191, China; State Key Laboratory of Intelligent Transportation System, Beijing, 100191, China.
KTH, School of Architecture and the Built Environment (ABE), Civil and Architectural Engineering, Transport planning.ORCID iD: 0000-0002-2141-0389
Show others and affiliations
2024 (English)In: Knowledge-Based Systems, ISSN 0950-7051, E-ISSN 1872-7409, Vol. 306, article id 112709Article in journal (Refereed) Published
Abstract [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.

Place, publisher, year, edition, pages
Elsevier BV , 2024. Vol. 306, article id 112709
Keywords [en]
Graph neural networks, Message passing, Multi-task learning, Traffic prediction
National Category
Computer Sciences Computer Systems
Identifiers
URN: urn:nbn:se:kth:diva-356666DOI: 10.1016/j.knosys.2024.112709ISI: 001356109100001Scopus ID: 2-s2.0-85208467234OAI: oai:DiVA.org:kth-356666DiVA, id: diva2:1914837
Note

QC 20241121

Available from: 2024-11-20 Created: 2024-11-20 Last updated: 2024-12-05Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full textScopus

Authority records

Ma, Zhenliang

Search in DiVA

By author/editor
Ma, Zhenliang
By organisation
Transport planning
In the same journal
Knowledge-Based Systems
Computer SciencesComputer Systems

Search outside of DiVA

GoogleGoogle Scholar

doi
urn-nbn

Altmetric score

doi
urn-nbn
Total: 91 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf