Shadowheart SGD: Distributed Asynchronous SGD with Optimal Time Complexity Under Arbitrary Computation and Communication Heterogeneity
Alexander Tyurin, Marta Pozzi, Ivan Ilin, Peter Richtárik
Abstract
We consider nonconvex stochastic optimization problems in the asynchronous centralized distributed setup where the communication times from workers to a server can not be ignored, and the computation and communication times are potentially different for all workers. Using an unbiassed compression technique, we develop a new method—Shadowheart SGD—that provably improves the time complexities of all previous centralized methods. Moreover, we show that the time complexity of Shadowheart SGD is optimal in the family of centralized methods with compressed communication. We also consider the bidirectional setup, where broadcasting from the server to the workers is non-negligible, and develop a corresponding method.
BibTeX
@inproceedings{
tyurin2024shadowheart,
title={Shadowheart {SGD}: Distributed Asynchronous {SGD} with Optimal Time Complexity Under Arbitrary Computation and Communication Heterogeneity},
author={Alexander Tyurin and Marta Pozzi and Ivan Ilin and Peter Richt{\'a}rik},
booktitle={The Thirty-eighth Annual Conference on Neural Information Processing Systems},
year={2024},
url={https://openreview.net/forum?id=O8yHsRLwPl}
}