Energy-efficient federated learning with resource allocation for green IoT edge intelligence in B5G.

Abstract

An edge intelligence-aided Internet-of-Things (IoT) network has been proposed to accelerate the response of IoT services by deploying edge intelligence near IoT devices. The transmission of data from IoT devices to the edge nodes leads to large network traffic in the wireless connections. Federated Learning (FL) is proposed to solve the high computational complexity by training the model locally on IoT devices and sharing the model parameters in the edge nodes. This paper focuses on developing an efficient integration of joint edge intelligence nodes depending on investigating an energy-efficient bandwidth allocation, computing Central Processing Unit (CPU) frequency, optimization transmission power, and the desired level of learning accuracy to minimize the energy consumption and satisfy the FL time requirement for all IoT devices. The proposal efficiently optimized the computation frequency allocation and reduced energy consumption in IoT devices by solving the bandwidth optimization problem in closed form. The remaining computational frequency allocation, transmission power allocation, and loss could be resolved with an Alternative Direction Algorithm (ADA) to reduce energy consumption and complexity at every iteration of FL time from IoT devices to edge intelligence nodes. The simulation results indicated that the proposed ADA can adapt the central processing unit frequency and power transmission control to reduce energy consumption at the cost of a small growth of FL time.