cs.CVMar 1, 2026

FoSS: Modeling Long Range Dependencies and Multimodal Uncertainty in Trajectory Prediction via Fourier State Space Integration

Authors: Yizhou Huang, Gengze Jiang, Yihua Cheng, Kezhi Wang

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

Explore similar work

CardsList
  1. TaPD: Temporal-adaptive Progressive Distillation for Observation-Adaptive Trajectory Forecasting in Autonomous Driving

    Mar 6, 2026Mingyu Fan, Yi Liu, Hao Zhou +3Trajectory PredictionMulti-Agent Trajectory Prediction

  2. Context-aware Attention-based Gaussian Mixture Models for Vehicular Trajectory Prediction

    Oct 6, 2026Arash Raftari, Babak Ebrahimi Soorchaei, Yaser P. FallahMultimodal Trajectory PredictionVehicle Trajectory Prediction

  3. Recover to Predict: Progressive Retrospective Learning for Variable-Length Trajectory Prediction

    Mar 11, 2026Hao Zhou, Lu Qi, Jason Li +5Trajectory PredictionAutonomous Driving