ControlledShifts: Towards Standardizing Robustness Evaluation in Trajectory Prediction Under Distribution Shifts
Organizations: Robotics Institute, School of Computer Science, Carnegie Mellon University · Work done as part of an internship at Lavoro AI. · Stack AV · Bosch Center for Artificial Intelligence
Abstract
Trajectory prediction is central to safety in autonomous driving, yet learning-based predictors tend to degrade sharply when encountering scenarios poorly represented by their training data. Many methods attempt to mitigate distribution shift degradation through data-centric or test-time adaptation approaches; however, they are typically validated along fragmented axes of generalization, leaving the field without a standardized way to compare robustness across shifts a model may encounter. To address this, we introduce ControlledShifts, a framework and benchmark suite that systematically re-splits existing trajectory datasets into in-distribution (seen) and out-of-distribution (unseen) partitions, via a shared characterization-and-splitting formulation, in which a characterization function fixes the axis of variation a benchmark probes and a splitting function fixes how the tail of that axis is withheld. The suite comprises three benchmarks targeting key topological and behavioral distribution shifts. Furthermore, to aggregate multi-dimensional performance metrics across these benchmarks, we propose a unified robustness score that evaluates models along two complementary dimensions: prediction quality (relative performance gain) and prediction stability (performance preservation under shift). We showcase ControlledShifts by benchmarking prominent transformer-based architectures, exposing critical differences in how models of varying capacities handle latent relevance and environmental structure.
Figures & tables
| Benchmark | Model | In Distribution ( seen , ) | Distribution Shift ( unseen , ) | Robustness Scores ( ) | |||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| BrierFDE | MinFDE 6 | MinADE 6 | MissRate | CollisionRate | BrierFDE | MinFDE 6 | MinADE 6 | MissRate | CollisionRate | Quality | Stability | ||||||
| Uniform | Naive | 11.228 | 10.561 | 3.649 | 0.640 | 0.048 | 11.357 ( +1.15% ) | 10.692 ( +1.24% ) | 3.689 ( +1.09% ) | 0.632 ( -1.28% ) | 0.046 ( -4.18% ) | 1.000 | 1.000 | ||||
| AutoBot | 3.213 | 2.519 | 2.480 | 0.368 | 0.017 | 3.195 ( -0.58% ) | 2.500 ( -0.74% ) | 2.453 ( -1.05% ) | 0.375 ( +1.85% ) | 0.016 ( -7.27% ) | 2.758 | 1.000 | |||||
| SceneTransformer | 4.748 | 4.276 | 1.547 | 0.491 | 0.067 | 4.793 ( +0.94% ) | 4.330 ( +1.26% ) | 1.566 ( +1.24% ) | 0.490 ( -0.31% ) | 0.069 ( +3.96% ) | 1.837 | 1.000 | |||||
| Wayformer | 2.792 | 2.306 | 0.880 | 0.385 | 0.060 | 2.806 ( +0.51% ) | 2.327 ( +0.91% ) | 0.887 ( +0.77% ) | 0.404 ( +4.99% ) | 0.061 ( +3.21% ) | 3.034 | 1.000 | |||||
| MTR | 3.839 | 3.316 | 1.210 | 0.424 | 0.054 | 3.841 ( +0.04% ) | 3.323 ( +0.20% ) | 1.209 ( -0.09% ) | 0.428 ( +1.00% ) | 0.056 ( +3.31% ) | 2.306 | 1.000 | |||||
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.