From Time Series to State: Situation-Aware Modeling for Air Traffic Flow Prediction
Organizations: Ocean University of China
Abstract
Accurate air traffic prediction in the terminal airspace (TA) is pivotal for proactive air traffic management (ATM). However, existing data-driven approaches predominantly rely on time series-based forecasting paradigms, which inherently overlook critical aircraft state information, such as real-time kinematics and proximity to airspace boundaries. To address this limitation, we propose AeroSense, a direct state-to-flow modeling framework for air traffic prediction. Unlike classical time series-based methods that first aggregate aircraft trajectories into macroscopic flow sequences before modeling, AeroSense explicitly represents the real-time airspace situation as a dynamic set of aircraft states, enabling the direct processing of a variable number of aircraft instead of time series as inputs. Specifically, we introduce a situation-aware state representation that enables AeroSense to sense the instantaneous terminal airspace situation directly from microscopic aircraft states. Furthermore, we design a model architecture that incorporates masked self-attention to capture inter-aircraft interactions, together with two decoupled prediction heads to model heterogeneous flow dynamics across two key functional areas of the TA. Extensive experiments on a large-scale real-world airport dataset demonstrate that AeroSense consistently achieves state-of-the-art performance, validating that direct modeling of microscopic aircraft states yields substantially higher predictive fidelity than time series-based baselines. Moreover, the proposed framework exhibits superior robustness during peak traffic periods, achieves Pareto-optimal performance under dayparting multi-object evaluation, and provides meaningful interpretability through attention-based visualizations.
Figures & tables
| Model | AP Area | AR Area | ||||
|---|---|---|---|---|---|---|
| MAE | RMSE | R 2 | MAE | RMSE | R 2 | |
| Autoformer | ||||||
| w Time Embedding | 2.052 | 2.668 | 0.898 | 4.361 | 5.572 | 0.943 |
| w/o Time Embedding | 2.048 | 2.603 | 0.903 | 4.238 | 5.624 | 0.942 |
| Fedformer | ||||||
| w time embedding | 1.872 | 2.404 | 0.917 | 3.170 | 4.078 | 0.969 |
| Variant | AP Area | AR Area | ||||
|---|---|---|---|---|---|---|
| MAE | RMSE | R 2 | MAE | RMSE | R 2 | |
| AeroSense | 1.506 | 2.040 | 0.935 | 2.100 | 2.806 | 0.984 |
| Effect of Module Design | ||||||
| w/o Masked Attention | 1.567 | 2.109 | 0.931 | 2.149 | 2.865 | 0.983 |
| w/o Decoupled Prediction Heads | 1.552 | 2.093 | 0.932 | 2.183 | 2.921 | 0.983 |
| Effect of Situation-aware State Representation | ||||||