Adaptive Configuration for Heterogeneous Participants in Decentralized Federated Learning
Preprint
Abstract
Data generated at the network edge can be processed locally by leveraging the paradigm of edge computing (EC). Aided by EC, decentralized federated learning (DFL), which overcomes the single-point-of-failure problem in the parameter server (PS) based federated learning, is becoming a practical and popular approach for machine learning over distributed data. However, DFL faces two critical challenges, \ie, system heterogeneity and statistical heterogeneity introduced by edge devices. To ensure fast convergence with the existence of slow edge devices, we present an efficient DFL method, termed FedHP, which integrates adaptive control of both local updating frequency and network topology to better support the heterogeneous participants. We establish a theoretical relationship between local updating frequency and network topology regarding model training performance and obtain a convergence upper bound. Upon this, we propose an optimization algorithm, that adaptively determines local updating frequencies and constructs the network topology, so as to speed up convergence and improve the model accuracy. Evaluation results show that the proposed FedHP can reduce the completion time by about 51% and improve model accuracy by at least 5% in heterogeneous scenarios, compared with the baselines.Keywords
All Related Versions
This publication has 17 references indexed in Scilit:
- Decentralized Machine Learning Through Experience-Driven Method in Edge NetworksIEEE Journal on Selected Areas in Communications, 2021
- FedSA: A Semi-Asynchronous Federated Learning Mechanism in Heterogeneous Edge ComputingIEEE Journal on Selected Areas in Communications, 2021
- Federated Learning in Mobile Edge Networks: A Comprehensive SurveyIEEE Communications Surveys & Tutorials, 2020
- Optimal Online Data Partitioning for Geo-Distributed Machine Learning in Edge of Wireless NetworksIEEE Journal on Selected Areas in Communications, 2019
- Adaptive Federated Learning in Resource Constrained Edge Computing SystemsIEEE Journal on Selected Areas in Communications, 2019
- Multi-Timescale Decentralized Online Orchestration of Software-Defined NetworksIEEE Journal on Selected Areas in Communications, 2018
- An Adaptive Synchronous Parallel Strategy for Distributed Machine LearningIEEE Access, 2018
- Edge Computing: Vision and ChallengesIEEE Internet of Things Journal, 2016
- Fast linear iterations for distributed averagingSystems & Control Letters, 2004
- Distributed asynchronous deterministic and stochastic gradient optimization algorithmsIEEE Transactions on Automatic Control, 1986