Journal
IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS
Volume 20, Issue 3, Pages 1935-1949Publisher
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/TWC.2020.3037554
Keywords
Wireless communication; Computational modeling; Training; Minimization; Wireless sensor networks; Resource management; Data models; Federated learning; resource allocation; energy efficiency
Funding
- Engineering and Physical Sciences Research Council (EPSRC) through the Scalable Full Duplex Dense Wireless Networks (SENSE) [EP/P003486/1]
- U.S. National Science Foundation [CNS-1814477]
- EPSRC [EP/P022723/1] Funding Source: UKRI
Ask authors/readers for more resources
This paper investigates the problem of energy-efficient transmission and computation resource allocation for federated learning over wireless communication networks. An iterative algorithm is proposed to minimize energy consumption and numerical results show a reduction of up to 59.5% compared to conventional methods.
In this paper, the problem of energy efficient transmission and computation resource allocation for federated learning (FL) over wireless communication networks is investigated. In the considered model, each user exploits limited local computational resources to train a local FL model with its collected data and, then, sends the trained FL model to a base station (BS) which aggregates the local FL model and broadcasts it back to all of the users. Since FL involves an exchange of a learning model between users and the BS, both computation and communication latencies are determined by the learning accuracy level. Meanwhile, due to the limited energy budget of the wireless users, both local computation energy and transmission energy must be considered during the FL process. This joint learning and communication problem is formulated as an optimization problem whose goal is to minimize the total energy consumption of the system under a latency constraint. To solve this problem, an iterative algorithm is proposed where, at every step, closed-form solutions for time allocation, bandwidth allocation, power control, computation frequency, and learning accuracy are derived. Since the iterative algorithm requires an initial feasible solution, we construct the completion time minimization problem and a bisection-based algorithm is proposed to obtain the optimal solution, which is a feasible solution to the original energy minimization problem. Numerical results show that the proposed algorithms can reduce up to 59.5% energy consumption compared to the conventional FL method.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available