针对事故场景中长尾轨迹预测难题,提出融合风险信息的扩散变换器模型。
Risk-Informed Diffusion Transformer for Long-Tail Trajectory Prediction in the Crash Scenario
- 结合图结构风险与扩散机制,用Transformer建模轨迹演化过程。
- 在尾部10%数据上minADE=0.016m,minFDE=2.667m,表现优异。
- 适用于高危罕见场景,提升自动驾驶系统安全性,适合安全关键领域研究者。
轨迹预测广泛应用于自动驾驶技术中。尽管整体预测精度较高,但训练数据中关键场景(如事故)的轨迹样本稀缺,导致长尾分布问题。本文从真实事故场景中提取轨迹数据,包含更多长尾样本。基于此,我们融合图结构风险信息与扩散模型,提出风险感知扩散变换器(RI-DiT)方法。在真实事故场景数据上的大量实验表明,该方法性能优越:在预测尾部10%(Top 10%)数据时,minADE为0.016米,minFDE为2.667米。同时分析了不同长尾分布下的轨迹特征,发现越接近尾部的数据,轨迹越不平滑。通过引入碰撞前时间(ITTC)和交通流特征,该方法显著提升了长尾轨迹预测准确性,推动了自动驾驶系统在极端场景下的安全性发展。
原文摘要 · Abstract (English)
Trajectory prediction methods have been widely applied in autonomous driving technologies. Although the overall performance accuracy of trajectory prediction is relatively high, the lack of trajectory data in critical scenarios in the training data leads to the long-tail phenomenon. Normally, the trajectories of the tail data are more critical and more difficult to predict and may include rare scenarios such as crashes. To solve this problem, we extracted the trajectory data from real-world crash scenarios, which contain more long-tail data. Meanwhile, based on the trajectory data in this scenario, we integrated graph-based risk information and diffusion with transformer and proposed the Risk-Informed Diffusion Transformer (RI-DiT) trajectory prediction method. Extensive experiments were conducted on trajectory data in the real-world crash scenario, and the results show that the algorithm we proposed has good performance. When predicting the data of the tail 10\% (Top 10\%), the minADE and minFDE indicators are 0.016/2.667 m. At the same time, we showed the trajectory conditions of different long-tail distributions. The distribution of trajectory data is closer to the tail, the less smooth the trajectory is. Through the trajectory data in real-world crash scenarios, Our work expands the methods to overcome the long-tail challenges in trajectory prediction. Our method, RI-DiT, integrates inverse time to collision (ITTC) and the feature of traffic flow, which can predict long-tail trajectories more accurately and improve the safety of autonomous driving systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。