FoSS: Modeling Long Range Dependencies and Multimodal Uncertainty in Trajectory Prediction via Fourier State Space Integration
Organizations: Brunel University of London · University of Birmingham
Abstract
Accurate trajectory prediction is vital for safe autonomous driving, yet existing approaches struggle to balance modeling power and computational efficiency. Attention-based architectures incur quadratic complexity with increasing agents, while recurrent models struggle to capture long-range dependencies and fine-grained local dynamics. Building upon this, we present FoSS, a dual-branch framework that unifies frequency-domain reasoning with linear-time sequence modeling. The frequency-domain branch performs a discrete Fourier transform to decompose trajectories into amplitude components encoding global intent and phase components capturing local variations, followed by a progressive helix reordering module that preserves spectral order; two selective state-space submodules, Coarse2Fine-SSM and SpecEvolve-SSM, refine spectral features with O(N) complexity. In parallel, a time-domain dynamic selective SSM reconstructs self-attention behavior in linear time to retain long-range temporal context. A cross-attention layer fuses temporal and spectral representations, while learnable queries generate multiple candidate trajectories, and a weighted fusion head expresses motion uncertainty. Experiments on Argoverse 1 and Argoverse 2 benchmarks demonstrate that FoSS achieves state-of-the-art accuracy while reducing computation by 22.5% and parameters by over 40%. Comprehensive ablations confirm the necessity of each component.
Figures & tables
| Method | b-minFDE | minADE | minFDE | MR | #Params (M) |
|---|---|---|---|---|---|
| LaneGCN [ 19 ] | 2.06 | 0.87 | 1.36 | 0.16 | - |
| DenseTNT [ 11 ] | 1.98 | 0.88 | 1.28 | 0.13 | - |
| BANet [ 36 ] | 1.92 | 0.71 | 1.36 | 0.19 | 9.49 |
| SceneTransformer [ 23 ] | 1.89 | 0.80 | 1.23 | 0.13 | 15.30 |
| GOHOME [ 8 ] | 1.86 | 0.89 | 1.29 | 0.08 | - |
| HiVT [ 44 ] | 1.84 | 0.77 | 1.17 | 0.13 | - |
Trajectory Prediction
Trajectory prediction forecasts how people, vehicles, robots, and other dynamic systems will move from observed histories and context.
See all 186 papers