An Ensemble Learning Framework for Vehicle Trajectory Prediction in Interactive Scenarios

Zirui Li, Yunlong Lin, Gong Cheng, Xinwei Wang, Qi Liu, Jianwei Gong*, Chao Lu

*Corresponding author for this work

Research output: Chapter in Book/Conference proceedings/Edited volumeConference contributionScientificpeer-review

1 Citation (Scopus)
11 Downloads (Pure)

Abstract

Precisely modeling interactions and accurately predicting trajectories of surrounding vehicles are essential to the decision-making and path-planning of intelligent vehicles. This paper proposes a novel framework based on ensemble learning to improve the performance of trajectory predictions in interactive scenarios. The framework is termed Interactive Ensemble Trajectory Predictor (IETP). IETP assembles interaction-aware trajectory predictors as base learners to build an ensemble learner. Firstly, each base learner in IETP observes historical trajectories of vehicles in the scene. Then each base learner handles interactions between vehicles to predict trajectories. Finally, an ensemble learner is built to predict trajectories by applying two ensemble strategies on the predictions from all base learners. Predictions generated by the ensemble learner are final outputs of IETP. In this study, three experiments using different data are conducted based on the NGSIM dataset. Experimental results show that IETP improves the predicting accuracy and decreases the variance of errors compared to base learners. In addition, IETP exceeds baseline models with 50% of the training data, indicating that IETP is data-efficient. Moreover, the implementation of IETP is publicly available at https://github.com/BIT-Jack/IETP.

Original languageEnglish
Title of host publicationProceedings of the 2022 IEEE Intelligent Vehicles Symposium, IV 2022
PublisherIEEE
Pages51-57
Number of pages7
ISBN (Electronic)978-1-6654-8821-1
ISBN (Print)978-1-6654-8822-8
DOIs
Publication statusPublished - 2022
Event2022 IEEE Intelligent Vehicles Symposium, IV 2022 - Aachen, Germany
Duration: 5 Jun 20229 Jun 2022

Conference

Conference2022 IEEE Intelligent Vehicles Symposium, IV 2022
Country/TerritoryGermany
CityAachen
Period5/06/229/06/22

Bibliographical note

Accepted author manuscript

Fingerprint

Dive into the research topics of 'An Ensemble Learning Framework for Vehicle Trajectory Prediction in Interactive Scenarios'. Together they form a unique fingerprint.

Cite this