UPDF AI

SPARQ-SGD: Event-Triggered and Compressed Communication in Decentralized Optimization

Navjot Singh,Deepesh Data,Jemin George,S. Diggavi

2020 · DOI: 10.1109/CDC42340.2020.9303828
IEEE Conference on Decision and Control · 0 Citations

Abstract

In this paper, we propose and analyze SPARQ-SGD, an event-triggered and compressed algorithm for decentralized training of large-scale machine learning models over a graph. Each node can locally compute a condition (event) which triggers a communication where quantized and sparsified local model parameters are sent. In SPARQ-SGD, each node first takes a fixed number of local gradient steps and then checks if the model parameters have significantly changed compared to its last update; it communicates further compressed model parameters only when there is a significant change, as specified by a (design) criterion. We prove that SPARQ-SGD converges as O(1nT)O\left( {\frac{1}{{nT}}} \right) and O(1nT)O\left( {\frac{1}{{\sqrt {nT} }}} \right) in the strongly-convex and non-convex settings, respectively, demonstrating that aggressive compression, including event-triggered communication, model sparsification and quantization does not affect the overall convergence rate compared to uncompressed decentralized training; thereby theoretically yielding communication efficiency for ‘free’. We evaluate SPARQ-SGD over real datasets to demonstrate significant savings in communication bits over the state-of-the-art.