Parallel Interactive Attention Network for Short-Term Origin–Destination Prediction in Urban Rail Transit

Abstract

Short-term origin–destination (termed as OD) prediction is crucial to improve the operation of urban rail transit (termed as URT). The latest research results show that deep learning can effectively improve the performance of short-term OD prediction and meet the real-time requirements. However, many advanced neural network design ideas have not been fully applied in the field of short-term OD prediction in URT. In this paper, a novel parallel interactive attention network (termed as PIANet) for short-term OD prediction in URT is proposed to further improve the short-term OD prediction accuracy. In the proposed PIANet, a novel omnidirectional attention module (termed as OAM) is proposed to improve the representational power of the network by calculating the feature weights in the channel–spatial dimension. Moreover, a simple yet effective feature interaction is proposed to improve the feature utilization. Based on the two real-world datasets from the Beijing subway, the comparative experiments demonstrate that the proposed PIANet outperforms the state-of-the-art deep learning methods for short-term OD prediction in URT, and the ablation studies demonstrate that the proposed OAMs and feature interaction play an important role in improving the short-term OD prediction accuracy.