用双注意力机制提升空域飞行轨迹预测精度
DA-STGCN: 4D Trajectory Prediction Based on Spatiotemporal Feature Extraction
- 通过自注意力重构邻接矩阵,动态捕捉飞机间关系
- 在两个真实数据集上使平均位移误差降低20%,末位误差降30%
- 适合空管系统、无人机调度等需要高精度轨迹预测场景
四维(4D)轨迹预测在空中交通管理中的重要性日益凸显,冲突检测与解决、飞行异常监控及拥堵航路管理均依赖此技术,亟需智能化解决方案。机场终端区和密集空域的动态复杂且变化频繁,现有方法未能充分建模飞机间的交互关系。为此,本文提出DA-STGCN,一种融合双注意力机制的时空图卷积网络。模型通过自注意力重构邻接矩阵,增强节点关联捕捉能力,并利用图注意力提取时空特征,生成轨迹的概率分布。该重构邻接矩阵在训练过程中动态优化,比传统算法更精细地反映节点间关系。在两个ADS-B数据集(一个靠近机场终端区,一个位于密集空域)上的实验表明,相比现有4D轨迹预测方法,本模型在平均位移误差(ADE)和最终位移误差(FDE)上分别降低20%和30%。消融实验证明双注意力模块显著提升了节点相关性提取能力。
原文摘要 · Abstract (English)
The importance of four-dimensional (4D) trajectory prediction within air traffic management systems is on the rise. Key operations such as conflict detection and resolution, aircraft anomaly monitoring, and the management of congested flight paths are increasingly reliant on this foundational technology, underscoring the urgent demand for intelligent solutions. The dynamics in airport terminal zones and crowded airspaces are intricate and ever-changing; however, current methodologies do not sufficiently account for the interactions among aircraft. To tackle these challenges, we propose DA-STGCN, an innovative spatiotemporal graph convolutional network that integrates a dual attention mechanism. Our model reconstructs the adjacency matrix through a self-attention approach, enhancing the capture of node correlations, and employs graph attention to distill spatiotemporal characteristics, thereby generating a probabilistic distribution of predicted trajectories. This novel adjacency matrix, reconstructed with the self-attention mechanism, is dynamically optimized throughout the network's training process, offering a more nuanced reflection of the inter-node relationships compared to traditional algorithms. The performance of the model is validated on two ADS-B datasets, one near the airport terminal area and the other in dense airspace. Experimental results demonstrate a notable improvement over current 4D trajectory prediction methods, achieving a 20% and 30% reduction in the Average Displacement Error (ADE) and Final Displacement Error (FDE), respectively. The incorporation of a Dual-Attention module has been shown to significantly enhance the extraction of node correlations, as verified by ablation experiments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。