UAV-Assisted Covert Federated Learning Over mmWave Massive MIMO
Z. Tong, Jingjing Wang, Xiangwang Hou, Chunxiao Jiang, Jianwei Liu
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 optimization problem involving the UAV’s trajectory, transmitting power, analog beamforming, and the required accuracy of FL. 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.