Privacy-Preserving Resource Allocation for Asynchronous Federated Learning

Xiaojing Chen,Zheer Zhou,Wei Ni,Guangjin Pan,Xin Wang,Shunqing Zhang,Yanzan Sun
DOI: https://doi.org/10.1109/vtc2024-spring62846.2024.10683194
2024-01-01
Abstract:This paper presents a novel two-stage deep reinforcement learning (DRL) algorithm built on a Transformer Encoder-based Deep Deterministic Policy Gradient (TEDDPG) framework, named TS-TEDDPG, which jointly optimizes the learning latency, energy consumption and model accuracy of Asynchronous Federated Learning (AFL) systems with prescribed security. The CPU configuration of local training and the transmit power of model uploading are learnt by the TEDDPG in the first stage. A linear programming-based device scheduling and cooperative jamming strategy is designed to efficiently optimize the rest of the decisions in the second stage and evaluates the immediate reward to train the TEDDPG. Experimental results based on a CNN model and the MNIST dataset demonstrate that the proposed TS-TEDDPG can reduce the training latency and energy consumption by 68.6% compared to its benchmarks, when the required test accuracy is 0.9.
What problem does this paper attempt to address?