FedComLoc: Communication-Efficient Distributed Training of Sparse and Quantized Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yi, Kai, Meinhardt, Georg, Condat, Laurent, Richtárik, Peter
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916943580102656
author Yi, Kai
Meinhardt, Georg
Condat, Laurent
Richtárik, Peter
author_facet Yi, Kai
Meinhardt, Georg
Condat, Laurent
Richtárik, Peter
contents Federated Learning (FL) has garnered increasing attention due to its unique characteristic of allowing heterogeneous clients to process their private data locally and interact with a central server, while being respectful of privacy. A critical bottleneck in FL is the communication cost. A pivotal strategy to mitigate this burden is Local Training, which involves running multiple local stochastic gradient descent iterations between communication phases. Our work is inspired by the innovative Scaffnew algorithm, which has considerably advanced the reduction of communication complexity in FL. We introduce FedComLoc (Federated Compressed and Local Training), integrating practical and effective compression into Scaffnew to further enhance communication efficiency. Extensive experiments, using the popular TopK compressor and quantization, demonstrate its prowess in substantially reducing communication overheads in heterogeneous settings.
format Preprint
id arxiv_https___arxiv_org_abs_2403_09904
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FedComLoc: Communication-Efficient Distributed Training of Sparse and Quantized Models
Yi, Kai
Meinhardt, Georg
Condat, Laurent
Richtárik, Peter
Machine Learning
Artificial Intelligence
Distributed, Parallel, and Cluster Computing
Federated Learning (FL) has garnered increasing attention due to its unique characteristic of allowing heterogeneous clients to process their private data locally and interact with a central server, while being respectful of privacy. A critical bottleneck in FL is the communication cost. A pivotal strategy to mitigate this burden is Local Training, which involves running multiple local stochastic gradient descent iterations between communication phases. Our work is inspired by the innovative Scaffnew algorithm, which has considerably advanced the reduction of communication complexity in FL. We introduce FedComLoc (Federated Compressed and Local Training), integrating practical and effective compression into Scaffnew to further enhance communication efficiency. Extensive experiments, using the popular TopK compressor and quantization, demonstrate its prowess in substantially reducing communication overheads in heterogeneous settings.
title FedComLoc: Communication-Efficient Distributed Training of Sparse and Quantized Models
topic Machine Learning
Artificial Intelligence
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2403.09904