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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Thuremella, Divya, Yang, Yi, Wanna, Simon, Kunze, Lars, De Martini, Daniele
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914042558283776
author Thuremella, Divya
Yang, Yi
Wanna, Simon
Kunze, Lars
De Martini, Daniele
author_facet Thuremella, Divya
Yang, Yi
Wanna, Simon
Kunze, Lars
De Martini, Daniele
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2509_13914
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Ensemble of Pre-Trained Models for Long-Tailed Trajectory Prediction
Thuremella, Divya
Yang, Yi
Wanna, Simon
Kunze, Lars
De Martini, Daniele
Machine Learning
Artificial Intelligence
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.
title Ensemble of Pre-Trained Models for Long-Tailed Trajectory Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.13914