When edge meets learning: adaptive control for resource-constrained distributed machine learning
File(s)Infocom_2018_distributed_ML.pdf (661.73 KB)
Accepted version
Author(s)
Type
Conference Paper
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 gradient-
descent based approaches. We analyze the convergence rate 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.
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 gradient-
descent based approaches. We analyze the convergence rate 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
2018-10-11
Date Acceptance
2017-11-27
Citation
IEEE INFOCOM 2018 - IEEE Conference on Computer Communications, 2018
Publisher
IEEE
Journal / Book Title
IEEE INFOCOM 2018 - IEEE Conference on Computer Communications
Copyright Statement
© 2018 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
Grant Number
4603317662
Source
IEEE Infocom 2018
Subjects
Science & Technology
Technology
Computer Science, Hardware & Architecture
Engineering, Electrical & Electronic
Telecommunications
Computer Science
Engineering
Publication Status
Published
Start Date
2018-04-16
Finish Date
2018-04-19
Coverage Spatial
Hawaii, USA