Trajectory prediction models in autonomous driving are vulnerable to perturbations from non-causal agents whose actions should not affect the ego-agent's behavior. Such perturbations can lead to incorrect predictions of other agents' trajectories, potentially compromising the safety and efficiency of the ego-vehicle's decision-making process. Motivated by this challenge, we propose , a novel model that utilizes a to identify inter-agent causal relations over a window of past time steps. To incorporate discovered causal relationships, we propose a novel mechanism to selectively filter information in the proposed Transformer-based architecture. We conduct extensive experiments on two autonomous driving benchmark datasets to evaluate the robustness of our model against non-causal perturbations and its generalization capacity. Our results indicate that the robustness of predictions can be improved by up to \textbf{54%} without a significant detriment to prediction accuracy. Lastly, we demonstrate the superior domain generalizability of the proposed model, which achieves up to \textbf{29%} improvement in cross-domain performance. These results underscore the potential of our model to enhance both robustness and generalization capacity for trajectory prediction in diverse autonomous driving domains. Further details can be found on our project page:this https URL.
View on arXiv@article{ahmadi2025_2410.07191,
title={ Curb Your Attention: Causal Attention Gating for Robust Trajectory Prediction in Autonomous Driving },
author={ Ehsan Ahmadi and Ray Mercurius and Soheil Alizadeh and Kasra Rezaee and Amir Rasouli },
journal={arXiv preprint arXiv:2410.07191},
year={ 2025 }
}