Skip to main navigation Skip to search Skip to main content

SignSGD With Federated Voting

Research output: Contribution to journal › Article › peer-review

Abstract

Distributed learning is commonly used for accelerating model training by harnessing the computational capabilities of multiple-edge devices. However, in practical applications, communication delay emerges as a bottleneck due to substantial information exchange required between workers and a central parameter server. SignSGD with majority voting (signSGD-MV) is an effective distributed learning algorithm that can significantly reduce communication costs by one-bit quantization. However, due to heterogeneous computational capabilities, it fails to converge when mini-batch sizes differ among workers. To overcome this, we propose a novel signSGD optimizer with federated voting (signSGD-FV). The idea of federated voting is to exploit learnable weights to perform weighted majority voting. The server learns the weights assigned to the edge devices in an online fashion based on their computational capabilities. Subsequently, these weights are employed to decode signs of aggregated local gradients to minimize the sign decoding error probability. We provide a unified convergence rate analysis applicable to scenarios where the estimated weights are known to the parameter server either perfectly or imperfectly. We demonstrate that the proposed signSGD-FV algorithm has a theoretical convergence guarantee even when edge devices use heterogeneous mini-batch sizes. Experimental results show that signSGD-FV outperforms signSGD-MV, exhibiting a faster convergence rate, especially in heterogeneous mini-batch sizes. Furthermore, we demonstrate that variants of the proposed signSGD-FV framework can address additional challenges in distributed learning systems, such as data heterogeneity and partial participation, highlighting the extensibility of signSGD-FV.

Original languageEnglish (US)
Pages (from-to)2740-2755
Number of pages16
JournalIEEE Transactions on Signal Processing
Volume74
DOIs
StatePublished - 2026

ASJC Scopus subject areas

  • Signal Processing
  • Electrical and Electronic Engineering

Keywords

  • communication-efficient learning
  • Federated learning
  • majority voting

Cite this