用隐空间生成新航迹数据,提升空管模型训练质量
Aircraft Trajectory Dataset Augmentation in Latent Space
- 将航迹数据映射到隐空间,用Transformer学习模式
- 在降维空间建模分布,生成高质量新航迹样本
- 适合空管系统研发者用于扩充训练数据
航空器航迹建模在空中交通管理(ATM)中至关重要,对冲突检测、着陆时间预测等下游任务具有重要意义。通过添加合成航迹数据进行数据集增强,有助于构建更鲁棒的航迹模型并确保数据集充分且均衡。本文提出一种名为ATRADA的新框架,首先使用Transformer编码器学习原始航迹数据中的潜在模式,并将每个数据点转换为学习到的隐空间中的上下文向量。随后,利用主成分分析(PCA)对数据集进行降维,并在低维空间中使用高斯混合模型(GMM)拟合数据点的概率分布。最后,从拟合的GMM中采样新样本,将其维度还原至原始维度,并通过多层感知机(MLP)解码生成新航迹。实验表明,该框架能有效生成高质量的合成航迹数据,性能优于多个基线方法。
原文摘要 · Abstract (English)
Aircraft trajectory modeling plays a crucial role in air traffic management (ATM) and is important for various downstream tasks, including conflict detection and landing time prediction. Dataset augmentation by adding synthetically generated trajectory data is necessary to develop a more robust aircraft trajectory model and ensure that the trajectory dataset is sufficient and balanced. We propose a novel framework called ATRADA for aircraft trajectory dataset augmentation. In the proposed framework, a Transformer encoder learns the underlying patterns in the original trajectory dataset and converts each data point into a context vector in the learned latent space. The converted dataset is projected to reduced dimensions using principal component analysis (PCA), and a Gaussian mixture model (GMM) is applied to fit the probability distribution of the data points in the reduced-dimensional space. Finally, new samples are drawn from the fitted GMM, the dimension of the samples is reverted to the original dimension, and the samples are decoded with a multi-layer perceptron (MLP). Several experiments demonstrate that the framework effectively generates new, high-quality synthetic aircraft trajectory data, which were compared to the results of several baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。