Deep Learning Meets Swarm Intelligence for UAV-Assisted IoT Coverage in Massive MIMO

Mobeen Mahmood,MohammadMahdi Ghadaksaz,Asil Koc,Tho Le-Ngoc
2023-09-21
Abstract:This study considers a UAV-assisted multi-user massive multiple-input multiple-output (MU-mMIMO) systems, where a decode-and-forward (DF) relay in the form of an unmanned aerial vehicle (UAV) facilitates the transmission of multiple data streams from a base station (BS) to multiple Internet-of-Things (IoT) users. A joint optimization problem of hybrid beamforming (HBF), UAV relay positioning, and power allocation (PA) to multiple IoT users to maximize the total achievable rate (AR) is investigated. The study adopts a geometry-based millimeter-wave (mmWave) channel model for both links and proposes three different swarm intelligence (SI)-based algorithmic solutions to optimize: 1) UAV location with equal PA; 2) PA with fixed UAV location; and 3) joint PA with UAV deployment. The radio frequency (RF) stages are designed to reduce the number of RF chains based on the slow time-varying angular information, while the baseband (BB) stages are designed using the reduced-dimension effective channel matrices. Then, a novel deep learning (DL)-based low-complexity joint hybrid beamforming, UAV location and power allocation optimization scheme (J-HBF-DLLPA) is proposed via fully-connected deep neural network (DNN), consisting of an offline training phase, and an online prediction of UAV location and optimal power values for maximizing the AR. The illustrative results show that the proposed algorithmic solutions can attain higher capacity and reduce average delay for delay-constrained transmissions in a UAV-assisted MU-mMIMO IoT systems. Additionally, the proposed J-HBF-DLLPA can closely approach the optimal capacity while significantly reducing the runtime by 99%, which makes the DL-based solution a promising implementation for real-time online applications in UAV-assisted MU-mMIMO IoT systems.
Information Theory,Signal Processing
What problem does this paper attempt to address?
The paper primarily investigates several key issues in UAV-assisted multi-user massive multiple-input multiple-output (MU-mMIMO) systems. Specifically, its main objectives include: 1. **Improving the total achievable rate (AR)**: By jointly optimizing hybrid beamforming (HBF), UAV relay positioning, and power allocation (PA) to maximize the system's total achievable rate. 2. **Reducing computational complexity**: Proposing a low-complexity joint optimization scheme based on deep learning (J-HBF-DLLPA), which can significantly reduce runtime (by up to 99%), making it a feasible solution for real-time online applications. 3. **Signal buffering**: Analyzing the performance of UAV relays with and without buffering, significantly improving transmission performance by adopting signal buffering techniques at the UAV relay, especially under delay-constrained conditions. To achieve these goals, the authors propose three different algorithmic schemes: 1. **Optimal power allocation for fixed UAV location (J-HBF-PSOPA-FL)**: In this case, the UAV's position is fixed, and the power allocation is optimal. This approach designs the RF and baseband (BB) stages to achieve maximum capacity. 2. **UAV location optimization for equal power allocation (J-HBF-PSOL-EQPA)**: The goal here is to optimize the UAV's position so that all IoT users receive the same power allocation. 3. **Joint UAV location optimization and optimal power allocation (J-HBF-PSOLPA)**: This is the most comprehensive method, considering both the optimization of the UAV's position and the optimal power allocation for each IoT user. The paper also proposes an algorithm based on particle swarm optimization (PSO) to solve the above optimization problems and introduces deep learning techniques to further reduce computational complexity and enhance practical applicability. Overall, this paper aims to utilize UAVs as relay nodes to enhance information transmission efficiency in MU-mMIMO systems through a series of optimization methods and demonstrates the effectiveness of the proposed algorithms through practical results.