Adaptive federated learning in resource constrained edge computing systems
File(s)AdaptiveFederatedLearning_JSAC_2019_02.pdf (1.83 MB)
Accepted version
Author(s)
Type
Journal Article
Abstract
Emerging technologies and applications including Internet of Things (IoT), social networking, and crowd-sourcing generate large amounts of data at the network edge. Machine learning models are often built from the collected data, to enable the detection, classification, and prediction of future events. Due to bandwidth, storage, and privacy concerns, it is often impractical to send all the data to a centralized location. In this paper, we consider the problem of learning model parameters from data distributed across multiple edge nodes, without sending raw data to a centralized place. Our focus is on a generic class of machine learning models that are trained using gradientdescent based approaches. We analyze the convergence bound of distributed gradient descent from a theoretical point of view, based on which we propose a control algorithm that determines the best trade-off between local update and global parameter aggregation to minimize the loss function under a given resource budget. The performance of the proposed algorithm is evaluated via extensive experiments with real datasets, both on a networked prototype system and in a larger-scale simulated environment. The experimentation results show that our proposed approach performs near to the optimum with various machine learning models and different data distributions.
Date Issued
2019-06-01
Date Acceptance
2019-02-12
Citation
IEEE Journal on Selected Areas in Communications, 2019, 37 (6), pp.1205-1221
ISSN
0733-8716
Publisher
Institute of Electrical and Electronics Engineers
Start Page
1205
End Page
1221
Journal / Book Title
IEEE Journal on Selected Areas in Communications
Volume
37
Issue
6
Copyright Statement
© 2019 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.
Sponsor
IBM United Kingdom Ltd
Identifier
https://ieeexplore.ieee.org/document/8664630
Grant Number
4603317662
Subjects
Science & Technology
Technology
Engineering, Electrical & Electronic
Telecommunications
Engineering
Distributed machine learning
federated learning
mobile edge computing
wireless networking
0805 Distributed Computing
0906 Electrical and Electronic Engineering
1005 Communications Technologies
Networking & Telecommunications
Publication Status
Published
Date Publish Online
2019-03-11