arXiv Machine Learning
Oct 8

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

The paper presents Context-Aware Attention-based Gaussian Mixture Models (CAA‑GMM) for predicting vehicular trajectories in complex, uncertain environments. CAA‑GMM models future motion as a probabilistic mixture conditioned on scene context and agent dynamics, using a lightweight attention mechanism to encode inter‑agent interactions and contextual salience. Experiments on nuScenes and Argoverse 2 show competitive or superior accuracy to raster‑based baselines while keeping computational complexity low, and ablation studies confirm the attention module’s importance for robust contextual reasoning.

By Arash Raftari, Babak Ebrahimi Soorchaei, Yaser P. Fallah