Developing Multi-task Learning Models for Short-term Forecasting in Taxi and Ride-hailing System
Funded by: Miyan Research Institute, IUBAT
Ride-hailing companies require accurate spatio-temporal forecasting of both demand and supply-demand gap to reduce passenger waiting times, minimize driver search friction, and ensure efficient service management. However, such forecasting is challenging due to strong spatio-temporal dependencies and, in many cases, the lack of explicit spatial adjacency information arising from confidentiality and privacy constraints. Furthermore, designing spatio-temporal forecasting models separately in a task-wise and city-wise manner to forecast demand and supply-demand gap in a ride-hailing system poses a burden for the expanding transportation network companies. To address these issues, two complementary deep learning architectures are proposed. First, a spatio-temporal model integrating a feature importance layer, a one-dimensional convolutional neural network (CNN), and a zone-distributed independently recurrent neural network (IndRNN) is introduced to handle anonymized spatial data. Second, a multi-task learning framework, GESME-Net, is proposed to simultaneously predict demand and supply-demand gap across different tasks and cities. This framework employs a gated ensemble of spatio-temporal mixture of experts network integrating a convolutional recurrent neural network (CRNN), CNN, and recurrent neural network (RNN), as well as a task adaptation layer that reveals feature contributions while learning a joint representation for multi-task learning.
Paper 1: doi.org/10.1049/itr2.12073
Paper 2: doi.org/10.1016/j.multra.2024.100166