An improved transformer‐based model for long‐term 4D trajectory prediction in civil aviation
Aofeng Luo,Yuxing Luo,Hong Liu,Wenchao Du,Xiping Wu,Hu Chen,Hongyu Yang
DOI: https://doi.org/10.1049/itr2.12530
IF: 2.7
2024-06-25
IET Intelligent Transport Systems
Abstract:Here, a novel transformer‐based approach is proposed for long‐term air traffic trajectory prediction, overcoming the limitations of long short‐term memory‐based methods. The model employs self‐attention for feature extraction, includes a trajectory stabilization module for enhanced predictability, and utilizes a one‐step inference strategy for improved precision and speed. Experimental results with real data from China's Southwest Air Traffic Management Bureau demonstrate superior performance compared to benchmark models. Four‐dimensional trajectory prediction is a crucial component of air traffic management, and its accuracy is closely related to the efficiency and safety of air transportation. Although long short‐term memory (LSTM) or its variants have been widely used in recent studies, they may produce unacceptable results in long‐term prediction due to the iterative output that accumulates error. To address this issue, a transformer‐based long‐term trajectory prediction model is proposed here, which utilizes the self‐attention mechanism to extract time series features from historical trajectory data. For long‐term prediction scenarios, we a trajectory stabilization module is introduced to ensure the stationarity of the time series for better predictability. Additionally, the transformer output strategy is improved to generate the prediction sequence by a single step instead of serial dynamic decoding, thus effectively enhancing the precision and inference speed. The proposed model is validated using real data obtained from China's Southwest Air Traffic Management Bureau. The experimental results demonstrate that this model outperforms the benchmark model. Further ablation experiments and visualizations are performed to analyze the impact of trajectory stabilization and one‐step inference strategy.
engineering, electrical & electronic,transportation science & technology