Huang, H., Mao, J., Liu, R. orcid.org/0000-0003-0627-3184 et al. (3 more authors) (2024) MTLMetro: A Deep Multi-Task Learning Model for Metro Passenger Demands Prediction. IEEE Transactions on Intelligent Transportation Systems. ISSN 1524-9050
Abstract
Accurate prediction of passenger demand is essential for the efficient operation and management of metro systems. In practical scenarios, strategies to enhance metro service quality often require passenger demand information on multiple fronts, such as inflow to a station, outflow from a station, as well as transition flow between entry/exit stations. While predictions for a single type of passenger demand have been extensively studied, limited attention was paid to jointly predicting multiple demands. This problem is challenging due to the complex relationships among multiple demands (e.g., inflow is only correlated with historical inflow, while the outflow is not only correlated with outflow but also determined by the inflow) and the imbalanced training issue of multiple prediction tasks. To address these challenges, this paper proposes a deep multi-task learning (MTL) model called MTLMetro to co-predict multiple demands in metro systems. More specifically, we deploy the message-passing schemes in graph neural networks (GNNs) as the knowledge-sharing mechanisms in the MTL model to capture the inherent relationships among multiple demands. To balance the training of multiple tasks, we introduce a novel weighting scheme named dynamic weight average (DWA), which can dynamically adapt relative weight for each task. In addition, the partial observability problem of transition flow is also considered in MTLMetro in an end-to-end manner. Empirical evaluation on a real-world dataset demonstrates MTLMetro’s superior performance across the different demand prediction tasks when compared to several benchmarks. Further ablation experiments verify the effectiveness of the proposed modules and the weighting method.
Metadata
Item Type: | Article |
---|---|
Authors/Creators: |
|
Copyright, Publisher and Additional Information: | © 2024 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works. |
Keywords: | Metro demand prediction; multi-task learning; deep learning; graph neural network; dynamic weight average |
Dates: |
|
Institution: | The University of Leeds |
Academic Units: | The University of Leeds > Faculty of Environment (Leeds) > Institute for Transport Studies (Leeds) > ITS: Spatial Modelling and Dynamics (Leeds) |
Depositing User: | Symplectic Publications |
Date Deposited: | 22 Mar 2024 10:48 |
Last Modified: | 25 Mar 2024 12:09 |
Status: | Published online |
Publisher: | Institute of Electrical and Electronics Engineers (IEEE) |
Identification Number: | 10.1109/tits.2024.3373565 |
Open Archives Initiative ID (OAI ID): | oai:eprints.whiterose.ac.uk:210698 |