English

Ensemble of Pre-Trained Models for Long-Tailed Trajectory Prediction

Machine Learning 2025-09-18 v1 Artificial Intelligence

Abstract

This work explores the application of ensemble modeling to the multidimensional regression problem of trajectory prediction for vehicles in urban environments. As newer and bigger state-of-the-art prediction models for autonomous driving continue to emerge, an important open challenge is the problem of how to combine the strengths of these big models without the need for costly re-training. We show how, perhaps surprisingly, combining state-of-the-art deep learning models out-of-the-box (without retraining or fine-tuning) with a simple confidence-weighted average method can enhance the overall prediction. Indeed, while combining trajectory prediction models is not straightforward, this simple approach enhances performance by 10% over the best prediction model, especially in the long-tailed metrics. We show that this performance improvement holds on both the NuScenes and Argoverse datasets, and that these improvements are made across the dataset distribution. The code for our work is open source.

Keywords

Cite

@article{arxiv.2509.13914,
  title  = {Ensemble of Pre-Trained Models for Long-Tailed Trajectory Prediction},
  author = {Divya Thuremella and Yi Yang and Simon Wanna and Lars Kunze and Daniele De Martini},
  journal= {arXiv preprint arXiv:2509.13914},
  year   = {2025}
}

Comments

Accepted 2025 IEEE International Conference on Intelligent Transportation Systems (ITSC 2025)