Abstract: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.

Velocity Prediction Method of Quadrotor UAV Based on BP Neural Network

Vibration Forecasting Using BP Neural Network for Image Stabilization and an Improving Method

Robotic velocity generation using neural network

Efficient Learning-based Trajectory Tacker for Quadrotor at High-speed Flight

Neural Network Based Model Predictive Control for a Quadrotor UAV

Backstepping control for quadrotor with BP neural network based thrust model

Neural Network Based PID Control for Quadrotor Aircraft.

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

A Velocity Controller for Quadrotors Based on Reinforcement Learning.

A Trajectory Prediction Method Based on Maneuver Recognition

Application of Modified BP Neural Network in Identification of Unmanned Surface Vehicle Dynamics

A Multilayer Neural Dynamic Controller Design Method of Quadrotor UAV for Completing Time-Varying Tasks

Application of the Improved BP Neural Network in Ship and Ocean Engineering

Modeling and PID control of quadrotor UAV based on machine learning

A New Neural-Dynamic Control Method Of Position And Angular Stabilization For Autonomous Quadrotor Uavs

Predictor-Based Neural Attitude Control of A Quadrotor with Disturbances

DIVE: Deep Inertial-Only Velocity Aided Estimation for Quadrotors

Physics-Inspired Temporal Learning of Quadrotor Dynamics for Accurate Model Predictive Trajectory Tracking

Neural Network Adaptive Inverse Model Control Method for Quadrotor UAV

A Revisited Approach To Lateral Acceleration Modeling For Quadrotor Uavs State Estimation

Wind estimation using quadcopter motion: A machine learning approach