TY - JOUR
T1 - Bayesian Spatio-Temporal grAph tRansformer network (B-STAR) for multi-aircraft trajectory prediction
AU - Pang, Yutian
AU - Zhao, Xinyu
AU - Hu, Jueming
AU - Yan, Hao
AU - Liu, Yongming
N1 - Publisher Copyright:
© 2022 Elsevier B.V.
PY - 2022/8/5
Y1 - 2022/8/5
N2 - Multi-Agent Trajectory Prediction is a critical and challenging component across different safety–critical engineering applications, e.g., autonomous driving and flight systems. Trajectory prediction tools are required for the next-generation air transportation system (NextGen). In practice, the prediction of aircraft trajectories needs to consider the impact of various sources, such as environmental conditions, pilot/controller behaviors, and potential conflicts with nearby aircraft. Huge uncertainties associated with these factors lead to the untrustworthiness of a deterministic trajectory prediction model. Moreover, the safety assurance in the near-terminal area is of specific interest due to the increased airspace complexity, where the instrument/visual flight rules are applied. In this work, we propose the Bayesian Spatio-Temporal grAph tRansformer (B-STAR) architecture to model the spatial and temporal relationship of multiple agents under uncertainties. It is shown that the proposed B-STAR achieves state-of-the-art performance on the ETH/UCY pedestrian dataset with UQ competence. Then, multi-aircraft near-terminal interactive trajectory prediction model is trained and validated with real-world flight recording data. The sensitivity study on the prediction/observation horizon and the graph neighboring distance threshold are performed. The code is available at https://github.com/ymlasu/para-atm-collection/tree/master/air-traffic-prediction/MultiAircraftTP.
AB - Multi-Agent Trajectory Prediction is a critical and challenging component across different safety–critical engineering applications, e.g., autonomous driving and flight systems. Trajectory prediction tools are required for the next-generation air transportation system (NextGen). In practice, the prediction of aircraft trajectories needs to consider the impact of various sources, such as environmental conditions, pilot/controller behaviors, and potential conflicts with nearby aircraft. Huge uncertainties associated with these factors lead to the untrustworthiness of a deterministic trajectory prediction model. Moreover, the safety assurance in the near-terminal area is of specific interest due to the increased airspace complexity, where the instrument/visual flight rules are applied. In this work, we propose the Bayesian Spatio-Temporal grAph tRansformer (B-STAR) architecture to model the spatial and temporal relationship of multiple agents under uncertainties. It is shown that the proposed B-STAR achieves state-of-the-art performance on the ETH/UCY pedestrian dataset with UQ competence. Then, multi-aircraft near-terminal interactive trajectory prediction model is trained and validated with real-world flight recording data. The sensitivity study on the prediction/observation horizon and the graph neighboring distance threshold are performed. The code is available at https://github.com/ymlasu/para-atm-collection/tree/master/air-traffic-prediction/MultiAircraftTP.
KW - Air traffic management
KW - Graph neural network
KW - Multi-agent trajectory prediction
KW - Transformer
UR - http://www.scopus.com/inward/record.url?scp=85130787596&partnerID=8YFLogxK
UR - http://www.scopus.com/inward/citedby.url?scp=85130787596&partnerID=8YFLogxK
U2 - 10.1016/j.knosys.2022.108998
DO - 10.1016/j.knosys.2022.108998
M3 - Article
AN - SCOPUS:85130787596
SN - 0950-7051
VL - 249
JO - Knowledge-Based Systems
JF - Knowledge-Based Systems
M1 - 108998
ER -