Improved adaptive graph local spatial–temporal multi-head self-attention network: a deep learning framework for flight delay prediction

Ming Wei,Ziqing Xu,Ruifeng Gao
DOI: https://doi.org/10.1007/s00521-024-10559-7
2024-12-09
Neural Computing and Applications
Abstract:Flight delay prediction is a challenging problem due to the sophisticated spatial–temporal correlations between airports and difficult-to-fit delay patterns. This paper proposed an improved adaptive graph local spatial–temporal multi-head self-attention network to predict flight arrival punctuality. First, an adaptive graph can learn about changing airport spatial–temporal correlations and capture their hidden spatial relationships. After that, local spatial–temporal multi-head self-attention was used to capture the flight delay's spatial–temporal feature, and the attention receptive field was set to the spatially neighboring nodes to reduce the time complexity. Finally, dilated inception convolution was used to merge more features in input data to improve long-term forecasting performance. A case study of the flight arrival punctuality of 50 airports in the USA from 2009 to 2019 validated this model's feasibility, which outperformed other models in predicting long-term flight delays.
computer science, artificial intelligence
What problem does this paper attempt to address?