Article (Scientific journals)
Efficient Federated Learning Algorithm for Resource Allocation in Wireless IoT Networks
Nguyen, van Dinh; Sharma, Shree Krishna; Vu, Thang Xuan et al.
2021In IEEE Internet of Things Journal, 8 (5), p. 3394 - 3409
Peer Reviewed verified by ORBi
 

Files


Full Text
FL for Wireless IoT Network_Final version.pdf
Author postprint (2.11 MB)
Download

All documents in ORBilu are protected by a user license.

Send to



Details



Keywords :
Energy-efficiency; IoT; federated learning
Abstract :
[en] Federated learning (FL) allows multiple edge computing nodes to jointly build a shared learning model without having to transfer their raw data to a centralized server, thus reducing communication overhead. However, FL still faces a number of challenges such as non-iid distributed data and heterogeneity of user equipments (UEs). Enabling a large number of UEs to join the training process in every round raises a potential issue of the heavy global communication burden. To address these issues, we generalize the current state-of-the-art Federated Averaging (FedAvg) by adding a weight-based proximal term to the local loss function. The proposed FL algorithm runs stochastic gradient descent in parallel on a sampled subset of the total UEs with replacement during each global round. We provide a convergence upper bound characterizing the trade-off between convergence rate and global rounds, showing that a small number of active UEs per round still guarantees convergence. Next, we employ the proposed FL algorithm in wireless Internet-of-Things (IoT) networks to minimize either total energy consumption or completion time of FL, where a simple yet efficient path-following algorithm is developed for its solutions. Finally, numerical results on unbalanced datasets are provided to demonstrate the performance improvement and robustness on the convergence rate of the proposed FL algorithm over FedAvg. They also reveal that the proposed algorithm requires much less training time and energy consumption than the FL algorithm with full user participation. These observations advocate the proposed FL algorithm for a paradigm shift in bandwidth- constrained learning wireless IoT networks.
Research center :
Interdisciplinary Centre for Security, Reliability and Trust (SnT) > SIGCOM
Disciplines :
Electrical & electronics engineering
Author, co-author :
Nguyen, van Dinh ;  University of Luxembourg > Interdisciplinary Centre for Security, Reliability and Trust (SNT) > SigCom
Sharma, Shree Krishna ;  University of Luxembourg > Interdisciplinary Centre for Security, Reliability and Trust (SNT) > SigCom
Vu, Thang Xuan  ;  University of Luxembourg > Interdisciplinary Centre for Security, Reliability and Trust (SNT) > SigCom
Chatzinotas, Symeon  ;  University of Luxembourg > Interdisciplinary Centre for Security, Reliability and Trust (SNT) > SigCom
Ottersten, Björn ;  University of Luxembourg > Interdisciplinary Centre for Security, Reliability and Trust (SNT)
External co-authors :
no
Language :
English
Title :
Efficient Federated Learning Algorithm for Resource Allocation in Wireless IoT Networks
Publication date :
01 March 2021
Journal title :
IEEE Internet of Things Journal
eISSN :
2327-4662
Publisher :
Institute of Electrical and Electronics Engineers
Volume :
8
Issue :
5
Pages :
3394 - 3409
Peer reviewed :
Peer Reviewed verified by ORBi
European Projects :
H2020 - 742648 - AGNOSTIC - Actively Enhanced Cognition based Framework for Design of Complex Systems
Name of the research project :
the ERC project AGNOSTIC
Funders :
CER - Conseil Européen de la Recherche [BE]
CE - Commission Européenne [BE]
Available on ORBilu :
since 02 November 2020

Statistics


Number of views
533 (78 by Unilu)
Number of downloads
420 (71 by Unilu)

Scopus citations®
 
66
Scopus citations®
without self-citations
62
WoS citations
 
56

Bibliography


Similar publications



Contact ORBilu