An Attention-based AI Model for 3D Beam Prediction in THz Unmanned Aerial Vehicle Communication

M. S. Munir,Sachin Shetty,Muhammad Enayetur Rahman
DOI: https://doi.org/10.1109/ICNC59896.2024.10556146
2024-02-19
Abstract:In this study, we considered the Terahertz (THz) Unmanned Aerial Vehicle (UAV) communication scenario, where we are dealing with the problem of beam prediction. UAVs have emerged as a critical component in evolving sixth-generation (6G) wireless networks, bringing seamless communication and extensive coverage. The usage of the THz spectrum allows for high-speed data throughput for the envisioned 6G era. However, the dynamic mobility of UAVs exacerbates the inherent challenges of THz communications, such as channel impairments and obstruction-related effects. We highlight the potential of Reconfigurable Intelligent Surface (RIS), which offers a flexible technique to boost coverage by adapting to channel fluctuations. We propose a unique deep learning paradigm based on self-attention networks in integrating RIS technology with THz UAV communication. In particular, we design and develop an attention-based mechanism that can predict the ideal serving beam for UAVs. This is based on sequential communication data such as UAV's beam trajectories, accurate position information, and Line of Sight (LoS) information. The proposed attention-based beam prediction mechanism performs well compared to the baseline Long Short-Term Memory (LSTM) and Gated recurrent units (GRUs) models, indicating superior performance and effectiveness. The findings, supported by comprehensive evaluation criteria such as Root-Mean-Square Error (RMSE) and Mean Absolute Error (MAE), demonstrate our proposed model's competency in capturing intricate temporal relationships, outperforming its competitors in predicting reliability.
Computer Science,Physics,Engineering
What problem does this paper attempt to address?