Abstract:Federated learning (FL) is a popular edge learning approach that utilizes local data and computing resources of network edge devices to train machine learning (ML) models while preserving users' privacy. Nevertheless, performing efficient learning tasks on the devices and achieving longer battery life are primary challenges faced by federated learning. In this paper, we are the first to study the application of heterogeneous computing (HC) and wireless power transfer (WPT) to federated learning to address these challenges. Especially, we propose a heterogeneous computation and resource allocation framework based on a heterogeneous mobile architecture to achieve effective implementation of FL. To minimize the energy consumption of smart devices and maximize their harvesting energy simultaneously, we formulate an optimization problem featuring multidimensional control, which jointly considers time splitting for WPT, dataset size allocation, transmit power allocation and subcarrier assignment during communications, and processor frequency of processing units (central processing unit (CPU) and graphics processing unit (GPU)). However, the major obstacle is how to design a proper algorithm to solve this optimization problem efficiently. For this purpose, we decouple the optimization variables so as to achieve high efficiency in deriving its solution. Particularly, we first compute the optimal processor frequency and dataset size allocation via employing the Lagrangian dual method, followed by finding the closed-form solution to the optimal time splitting allocation, and finally attain the optimal subcarrier assignment as well as transmit power for transmissions through an iteration algorithm. To evaluate the performance of our proposed scheme, we set up four baseline schemes as comparison, and simulation results show that the proposed scheme converges quite fast and better enhance the energy efficiency of the wireless powered FL system compared with t-e baseline schemes.

Optimize Scheduling of Federated Learning on Battery-powered Mobile Devices

Towards Efficient Scheduling of Federated Mobile Devices under Computational and Statistical Heterogeneity

Energy Optimization for Federated Learning on Consumer Mobile Devices with Asynchronous SGD and Application Co-Execution

Energy Minimization for Federated Asynchronous Learning on Battery-Powered Mobile Devices via Application Co-running

Adaptive Batchsize Selection and Gradient Compression for Wireless Federated Learning

Energy-efficient Federated Learning Via Stabilization-aware On-device Update Scaling.

Edge-Based Communication Optimization for Distributed Federated Learning

BEFL: Balancing Energy Consumption in Federated Learning for Mobile Edge IoT

Energy-Efficient Federated Training on Mobile Device

Heterogeneous Computation and Resource Allocation for Wireless Powered Federated Edge Learning Systems

Online Optimization for Over-the-Air Federated Learning with Energy Harvesting

Communication-Efficient Device Scheduling for Federated Learning Using Stochastic Optimization

Accelerating Wireless Federated Learning with Adaptive Scheduling over Heterogeneous Devices

Experience-Driven Computational Resource Allocation of Federated Learning by Deep Reinforcement Learning

Optimizing Federated Learning on Device Heterogeneity with A Sampling Strategy

Communicational and Computational Efficient Federated Domain Adaptation

Device Scheduling and Resource Allocation for Federated Learning under Delay and Energy Constraints

Online Client Scheduling and Resource Allocation for Efficient Federated Edge Learning

Energy Efficient Federated Learning over Heterogeneous Mobile Devices via Joint Design of Weight Quantization and Wireless Transmission

Joint Gradient Sparsification and Device Scheduling for Federated Learning

Resource-Efficient and Convergence-Preserving Online Participant Selection in Federated Learning