Motion Prediction in Airports using Heterogeneous Map Representations

Motion ForecastingTransformerGraph Neural NetAviation

Motivation

Ensuring safety in airport ground operations is crucial to prevent catastrophic collisions, known as incursions. Human monitoring is currently relied upon to identify and correct potential risks. Motion forecasting models, underexplored in aviation, offer promise in automating the detection of safety-critical situations. This project focuses on studying the impact of different map representations on these models for enhancing situational awareness in aviation safety systems.

Data

The trajectory data, collected through the Federal Aviation Administration's (FAA) System Wide Information Management (SWIM) platform, spans 200 days of movement between Seattle-Tacoma Airport (KSEA) and Newark Airport (KEWR).

KSEA Vector Map
KEWR Vector Map

Specifically, we utilized the SWIM Terminal Data Distribution System (STDDS) for trajectory data, collecting information about aircraft and operational vehicles from 6 FAA systems — including Airport Surface Detection Equipment Model X, Airport Surface Surveillance Capability, Standard Terminal Automation Replacement System, Runway Visual Range, Electronic Flight Strip Transfer System, and Tower Data Link Services. The data spans 200 days from December 1 to June 19, 2023, providing details on altitude, speed, heading, vehicle type, and 2D geographical coordinates, grouped into 30-second-long scenarios.

Airport Map Representations

We processed the airport maps into two different representations: (1) rasterized images of HD maps and (2) graph-based vectorized maps, with semantic information imbued in their nodes and edges.

Raster Representation

We viewed the polygons formed by map elements as shapes on an image. To retain semantic information, a specific color is assigned to each semantic class.

Semantic Map

Graph Representation

We processed the airport maps by vectorizing them: (a) the original graph before processing, (b) remaining centerlines after categorizing each node into Runway, Taxiway, Hold-short Line, or Exit, (c) the graph after connecting each runway and eliminating redundant nodes, (d) the vectorization process of the graph.

Vectorization Process

Model & Pipeline

The motion forecasting model architecture adopts a multi-modal joint prediction approach, utilizing an agent-centric representation for trajectories and local context patches. The agent encoder incorporates a linear layer, while the context encoder varies based on the chosen map representation. In this project, we compared CNN-based and MLP-based encoders with and without attention layers for rasterized and graph representations. A GPT-2 style causal encoder performs masked attention over trajectories. Lastly, a Gaussian Mixture Model (GMM) serves as the prediction header, representing future trajectories as a probabilistic distribution for each time step.

Model Pipeline

Evaluation

We assessed motion forecasting models by inputting historical aircraft trajectories and corresponding map patches in both representations. In both cases, raster-based context slightly outperforms the vector-based context. In KSEA, raster-based context has a 0.9 lower ADE and a 3.07 lower FDE than vector-based. In KEWR, ADE differs by 0.98 and FDE by 4.42. Despite the scale of aircraft, these differences are relatively low, suggesting that both context encoding methods are learning similar features from the provided information.

ADE FDE Table
Raster SWIM Grid
Vector SWIM Grid

Acknowledgement

I would like to express my sincere gratitude to Pablo Ortega-Kral for his substantial contributions to this project. Pablo played a pivotal role in the execution of the research, including the primary work and creation of figures presented in this project. His dedication and expertise significantly enriched the outcome of this study.