ProIn: Learning to Predict Trajectory Based on Progressive Interactions for Autonomous Driving

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dong, Yinke, Yuan, Haifeng, Liu, Hongkun, Jing, Wei, Li, Fangzhen, Liu, Hongmin, Fan, Bin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917621585149952
author Dong, Yinke
Yuan, Haifeng
Liu, Hongkun
Jing, Wei
Li, Fangzhen
Liu, Hongmin
Fan, Bin
author_facet Dong, Yinke
Yuan, Haifeng
Liu, Hongkun
Jing, Wei
Li, Fangzhen
Liu, Hongmin
Fan, Bin
contents Accurate motion prediction of pedestrians, cyclists, and other surrounding vehicles (all called agents) is very important for autonomous driving. Most existing works capture map information through an one-stage interaction with map by vector-based attention, to provide map constraints for social interaction and multi-modal differentiation. However, these methods have to encode all required map rules into the focal agent's feature, so as to retain all possible intentions' paths while at the meantime to adapt to potential social interaction. In this work, a progressive interaction network is proposed to enable the agent's feature to progressively focus on relevant maps, in order to better learn agents' feature representation capturing the relevant map constraints. The network progressively encode the complex influence of map constraints into the agent's feature through graph convolutions at the following three stages: after historical trajectory encoder, after social interaction, and after multi-modal differentiation. In addition, a weight allocation mechanism is proposed for multi-modal training, so that each mode can obtain learning opportunities from a single-mode ground truth. Experiments have validated the superiority of progressive interactions to the existing one-stage interaction, and demonstrate the effectiveness of each component. Encouraging results were obtained in the challenging benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2403_16374
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ProIn: Learning to Predict Trajectory Based on Progressive Interactions for Autonomous Driving
Dong, Yinke
Yuan, Haifeng
Liu, Hongkun
Jing, Wei
Li, Fangzhen
Liu, Hongmin
Fan, Bin
Machine Learning
Computer Vision and Pattern Recognition
Robotics
Accurate motion prediction of pedestrians, cyclists, and other surrounding vehicles (all called agents) is very important for autonomous driving. Most existing works capture map information through an one-stage interaction with map by vector-based attention, to provide map constraints for social interaction and multi-modal differentiation. However, these methods have to encode all required map rules into the focal agent's feature, so as to retain all possible intentions' paths while at the meantime to adapt to potential social interaction. In this work, a progressive interaction network is proposed to enable the agent's feature to progressively focus on relevant maps, in order to better learn agents' feature representation capturing the relevant map constraints. The network progressively encode the complex influence of map constraints into the agent's feature through graph convolutions at the following three stages: after historical trajectory encoder, after social interaction, and after multi-modal differentiation. In addition, a weight allocation mechanism is proposed for multi-modal training, so that each mode can obtain learning opportunities from a single-mode ground truth. Experiments have validated the superiority of progressive interactions to the existing one-stage interaction, and demonstrate the effectiveness of each component. Encouraging results were obtained in the challenging benchmarks.
title ProIn: Learning to Predict Trajectory Based on Progressive Interactions for Autonomous Driving
topic Machine Learning
Computer Vision and Pattern Recognition
Robotics
url https://arxiv.org/abs/2403.16374