Lane-changing trajectory prediction based on multi-task learning

Abstract

As a complex driving behavior, lane-changing (LC) behavior has a great influence on traffic flow. Improper lane-changing behavior often leads to traffic accidents. Numerous studies are currently being conducted to predict lane change trajectories to minimize dangers. However, most of their models focus on how to optimize input variables without considering the interaction between output variables. This study proposes a LC trajectory prediction model based on a multi-task deep learning framework to improve driving safety. Concretely, in this work, the coupling effect of lateral and longitudinal movement is considered in the LC process. Trajectory changes in two directions will be modeled separately, and the information interaction is completed under the multi-task learning framework. In addition, the trajectory fragments are clustered by the driving features, and trajectory type recognition is added to the trajectory prediction framework as an auxiliary task. Finally, the prediction process of lateral and longitudinal trajectory and LC style is completed by Long Short-Term Memory (LSTM). The model training and testing are conducted with the data collected by the driving simulator, and that the proposed method expresses better performance in the LC trajectory prediction compared with several traditional models. The result showed in this study can enhance the trajectory prediction accuracy of Advanced Driving Assistance System (ADAS) and reduce the traffic accidents caused by lane changes.