← Search

Ran Ben-Basat

3 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
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…