Timezone: »

 
Poster
Preserved central model for faster bidirectional compression in distributed settings
Constantin Philippenko · Aymeric Dieuleveut

Thu Dec 09 12:30 AM -- 02:00 AM (PST) @
in Poster Session 5 »

We develop a new approach to tackle communication constraints in a distributed learning problem with a central server. We propose and analyze a new algorithm that performs bidirectional compression and achieves the same convergence rate as algorithms using only uplink (from the local workers to the central server) compression. To obtain this improvement, we design MCM, an algorithm such that the downlink compression only impacts local models, while the global model is preserved. As a result, and contrary to previous works, the gradients on local servers are computed on perturbed models. Consequently, convergence proofs are more challenging and require a precise control of this perturbation. To ensure it, MCM additionally combines model compression with a memory mechanism. This analysis opens new doors, e.g. incorporating worker dependent randomized-models and partial participation.

Keywords: Optimization · Federated Learning

Author Information

Constantin Philippenko (Ecole Polytechnique, IPParis)
Aymeric Dieuleveut (Ecole Polytechnique, IPParis)

More from the Same Authors