Search papers, labs, and topics across Lattice.
This paper addresses the limitations of current trajectory forecasting models in autonomous driving, which often optimize for specific benchmark metrics that can conflict with one another. By introducing a metric-agnostic training approach, the authors propose the Trajectory Distribution Evaluation (TraDiE) policies to map predictive distributions to required trajectories and confidences, allowing for more robust performance across various metrics. The proposed DONUT-NLL model, which adapts the state-of-the-art DONUT model, achieves superior results on the Waymo motion prediction benchmark, demonstrating the effectiveness of this new paradigm.
Training trajectory forecasting models with a metric-agnostic approach can lead to state-of-the-art performance across all evaluation metrics, challenging the notion that metric-specific optimization is necessary.
Accurate trajectory forecasting of surrounding traffic participants is a core capability for autonomous driving, enabling vehicles to anticipate behavior and plan safe maneuvers. We observe that current state-of-the-art forecasting models on Argoverse 2 and the Waymo Open Motion Dataset tailor their training objectives to the different benchmark metrics. Because these metrics encourage conflicting behavior, we propose a paradigm change for trajectory forecasting: training models with metric-agnostic probabilistic objectives and treating metric optimization as a downstream task applied to the predictive distribution. Concretely, we introduce Trajectory Distribution Evaluation (TraDiE) policies, metric-specific policies that map a predictive distribution to the set of $K$ trajectories and confidences required by trajectory forecasting metrics. We evaluate this framework by introducing DONUT-NLL, which adapts the training objective of the state-of-the-art trajectory forecasting model DONUT to directly optimize the predictive distribution. Using our policies, DONUT-NLL achieves state-of-the-art results on all metrics of the Waymo motion prediction benchmark.
