arXiv:2410.23305cs.ROcs.LG2024-10被引 26

用速度数据提升无人机轨迹预测精度,实现实时高精度3D预测。

VECTOR: Velocity-Enhanced GRU Neural Network for Real-Time 3D UAV Trajectory Prediction

  • 基于历史速度而非位置数据训练GRU模型,捕捉飞行动态。
  • 在合成与真实数据上均实现低至2×10⁻⁸的均方误差。
  • 适合需要高精度实时轨迹预测的无人机系统开发。

本文针对无人机实时3D轨迹预测难题,提出一种基于门控循环单元(GRU)的速度增强型神经网络方法。传统依赖位置数据的模型在训练分布外表现不佳,本研究通过引入速度估计这一一阶动力学信息,显著提升预测精度与泛化能力。所提方法基于历史速度序列预测未来速度与位置,避免对位置域的依赖。实验使用Gazebo仿真器与PX4自动驾驶系统生成合成数据,结合UZH-FPV和Mid-Air无人机竞速数据集进行验证。结果表明,该方法在多种飞行模式、速度与机动性场景下均优于现有RNN模型,均方误差低至2×10⁻⁸。研究还开源了包含5000条轨迹的数据集及ROS 2集成包,便于与现有无人机系统对接。

原文摘要 · Abstract (English)

This paper tackles the challenge of real-time 3D trajectory prediction for UAVs, which is critical for applications such as aerial surveillance and defense. Existing prediction models that rely primarily on position data struggle with accuracy, especially when UAV movements fall outside the position domain used in training. Our research identifies a gap in utilizing velocity estimates, first-order dynamics, to better capture the dynamics and enhance prediction accuracy and generalizability in any position domain. To bridge this gap, we propose a new trajectory prediction method using Gated Recurrent Units (GRUs) within sequence-based neural networks. Unlike traditional methods that rely on RNNs or transformers, this approach forecasts future velocities and positions based on historical velocity data instead of positions. This is designed to enhance prediction accuracy and scalability, overcoming challenges faced by conventional models in handling complex UAV dynamics. The methodology employs both synthetic and real-world 3D UAV trajectory data, capturing a wide range of flight patterns, speeds, and agility. Synthetic data is generated using the Gazebo simulator and PX4 Autopilot, while real-world data comes from the UZH-FPV and Mid-Air drone racing datasets. The GRU-based models significantly outperform state-of-the-art RNN approaches, with a mean square error (MSE) as low as 2 x 10^-8. Overall, our findings confirm the effectiveness of incorporating velocity data in improving the accuracy of UAV trajectory predictions across both synthetic and real-world scenarios, in and out of position data distributions. Finally, we open-source our 5000 trajectories dataset and a ROS 2 package to facilitate the integration with existing ROS-based UAV systems.

无人机轨迹预测GRU实时系统

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。