Multi-Region Taxi Pick-Up Demand Prediction Based on Edge-GATv2-LSTM

Currently, the short-term accurate prediction of multi-region taxi pick-up demand often adopts methods that integrate graph neural networks with temporal modeling. However, most models focus solely on node features during the learning process, neglecting or simplifying edge features. This study adopts a hybrid prediction framework, Edge-GATv2-LSTM, which integrates an edge-aware attention-based graph neural network (Edge-GATv2) with a temporal modeling component (LSTM). The framework not only models spatial interactions among regions via GATv2 and temporal evolution via LSTM but also incorporates edge features into the attention computation structure, jointly representing them with node features. This enables the model to perceive both node attributes and the strength of inter-regional relationships during attention weight calculation. Experiments are conducted based on real-world taxi order data from Ningbo City, and the results demonstrate that the adopted Edge-GATv2-LSTM model exhibits favorable performance in terms of pick-up demand prediction accuracy. Specifically, the model achieves the lowest RMSE and MAE of 3.85 and 2.86, respectively, outperforming all baseline methods and confirming its effectiveness in capturing spatiotemporal demand patterns. This research can provide decision-making support for taxi drivers, platform operators, and traffic management departments—for example, by offering a reference basis for optimizing taxi pick-up route planning when vehicles are unoccupied.