UAV-Assisted Covert Federated Learning over Mmwave Massive MIMO

Ziheng Tong,Jingjing Wang,Xiangwang Hou,Chunxiao Jiang,Jianwei Liu
DOI: https://doi.org/10.1109/twc.2024.3384957
IF: 10.4
2024-01-01
IEEE Transactions on Wireless Communications
Abstract:Unmanned aerial vehicles (UAVs) associated with federated learning (FL) have been deemed as a prospective framework by utilizing private data generated in the edge devices. However, despite various privacy-preserving and cryptography technologies adopted at the data level, FL still faces a range of security threats to raw data considering the broadcast nature of wireless channel. In this paper, to facilitate the communication-efficiency and privacy-preservation capability, we propose a UAV-enhanced covert federated learning architecture over mmWave massive multiple input multiple output (MIMO) channel, where we harness the covert communication technique in FL in order to avoid eavesdropping of illegal wardens. To achieve a trade-off between the security performance and training cost, we formulate a joint UAV's trajectory, transmitting power, analog beamforming as well as the required accuracy of FL optimization problem. Furthermore, we propose the multi-agent deep deterministic policy gradient (MADDPG) algorithm to solve the above-mentioned problem. Numerous simulations have been performed to demonstrate both the effectiveness and convergence of the proposed algorithm.
What problem does this paper attempt to address?