← Search

shay vargaftik

5 accepted papers

2024

Accelerating Federated Learning with Quick Distributed Mean Estimation

ICML 2024poster

Distributed Mean Estimation (DME), in which $n$ clients communicate vectors to a parameter server that estimates their average, is a fundamental building block in communication-efficient federated learning. In this paper, we improve on previous DME techniques that achieve the optimal $O(1/n)$ Normal…

Cited by 13SourcePDFScholar
2024

Optimal and Approximate Adaptive Stochastic Quantization

NeurIPS 2024poster

Quantization is a fundamental optimization for many machine learning (ML) use cases, including compressing gradients, model weights and activations, and datasets. The most accurate form of quantization is adaptive, where the error is minimized with respect to a given input rather than optimizing for…

Cited by 2SourcePDFScholar
2023

DoCoFL: Downlink Compression for Cross-Device Federated Learning

ICML 2023poster

Many compression techniques have been proposed to reduce the communication overhead of Federated Learning training procedures. However, these are typically designed for compressing model updates, which are expected to decay throughout training. As a result, such methods are inapplicable to downlink…

Cited by 20SourcePDFScholar
2022

EDEN: Communication-Efficient and Robust Distributed Mean Estimation for Federated Learning

ICML 2022spotlight

Distributed Mean Estimation (DME) is a central building block in federated learning, where clients send local gradients to a parameter server for averaging and updating the model. Due to communication constraints, clients often use lossy compression techniques to compress the gradients, resulting in…

2021

DRIVE: One-bit Distributed Mean Estimation

NeurIPS 2021poster

We consider the problem where $n$ clients transmit $d$-dimensional real-valued vectors using $d(1+o(1))$ bits each, in a manner that allows the receiver to approximately reconstruct their mean. Such compression problems naturally arise in distributed and federated learning. We provide novel mathemat…