← Search

Mike Rabbat

7 accepted papers

2022

Federated Learning with Buffered Asynchronous Aggregation

AISTATS 2022poster

Scalability and privacy are two critical concerns for cross-device federated learning (FL) systems. In this work, we identify that synchronous FL – cannot scale efficiently beyond a few hundred clients training in parallel. It leads to diminishing returns in model performance and training speed, ana…

Cited by 377SourcePDFScholar
2022

Federated Learning with Partial Model Personalization

ICML 2022spotlight

We consider two federated learning algorithms for training partially personalized models, where the shared and personal parameters are updated either simultaneously or alternately on the devices. Both algorithms have been proposed in the literature, but their convergence properties are not fully und…

2020

On the Convergence of Nesterov’s Accelerated Gradient Method in Stochastic Settings

ICML 2020poster

We study Nesterov’s accelerated gradient method with constant step-size and momentum parameters in the stochastic approximation setting (unbiased gradients with bounded variance) and the finite-sum setting (where randomness is due to sampling mini-batches). To build better insight into the behavior…

Cited by 75SourcePDFScholar
2019

Stochastic Gradient Push for Distributed Deep Learning

ICML 2019oral

Distributed data-parallel algorithms aim to accelerate the training of deep neural networks by parallelizing the computation of large mini-batch gradient updates across multiple nodes. Approaches that synchronize nodes using exact distributed averaging (e.g., via AllReduce) are sensitive to straggle…