Hybrid machine learning-based 3-dimensional UAV node localization for UAV-assisted wireless networks

Abstract

This paper presents a hybrid machine-learning framework for optimizing 3-Dimensional (3D) Unmanned Aerial Vehicles (UAV) node localization and resource distribution in UAV-assisted THz 6G networks to ensure efficient coverage in dynamic, high-density environments. The proposed model efficiently managed interference, adapted to UAV mobility, and ensured optimal throughput by dynamically optimizing UAV trajectories. The hybrid framework combined the strengths of Graph Neural Networks (GNN) for feature aggregation, Deep Neural Networks (DNN) for efficient resource allocation, and Double Deep Q-Networks (DDQN) for distributed decision-making. Simulation results demonstrated that the proposed model outperformed traditional machine learning models, significantly improving energy efficiency, latency, and throughput. The hybrid model achieved an optimized energy efficiency of 90 Tbps/J, reduced latency to 0.0105 ms, and delivered a network throughput of approximately 96 Tbps. The model adapts to varying link densities, maintaining stable performance even in high-density scenarios. These findings underscore the framework's potential to address key challenges in UAV-assisted 6G networks, paving the way for scalable and efficient communication in next-generation wireless systems.