math.OCJul 2, 2026

Decentralized Stochastic Subgradient-type Methods with Communication Compression for Nonsmooth Nonconvex Optimization

Authors: Siyuan ZhangNachuan XiaoXin Liu

Abstract

In this paper, we consider the nonsmooth nonconvex decentralized optimization problem, where inter-agent communication is compressed. We propose a general framework that unifies various decentralized stochastic subgradient-type methods with unbiased compression and contractive compression with error compensation. By relating the consensus-error iterates and the averaged iterates to the trajectories of continuous-time differential inclusions, we establish global convergence for all methods encompassed by our framework when the objective functions are nonsmooth and lack Clarke regularity. Based on our framework, we further develop several compression-based methods, including decentralized stochastic subgradient methods utilizing sign-based regularization and gradient-tracking momentum. Preliminary numerical experiments empirically support our theoretical results and highlight the communication-accuracy trade-off of the newly developed methods.

Explore similar work

Jun 3, 2026math.OC

Near-Optimal Decentralized Stochastic Convex Optimization over Networks

We study decentralized stochastic smooth convex optimization, where MM workers minimize an average objective using local stochastic gradients and neighbor-only communication over a fixed gossip network. A central question in this setting is to determine the largest number of workers that can be used under a total budget of NN gradient samples while still preserving the centralized O(1/N)O(1/\sqrt N) statistical rate. We introduce an accelerated decentralized method that preserves this rate for up to MρN3/4\smash{M\lesssim \sqrtρ\,N^{3/4}} workers, where ρρ is the spectral gap of the gossip network, improving the best prior maximal scaling of MρN\smash{M\lesssim ρ\sqrt N}. The method is based on a one-step-delayed stochastic acceleration scheme that enables workers to interleave minibatching with accelerated gossip while controlling residual disagreement, and its guarantee depends only logarithmically on the optimum-local heterogeneity. We also establish a matching lower bound for linear-span decentralized first-order methods, showing that the method is optimal up to logarithmic factors.
Nitai Kluger, Amit Attia, Tomer Koren
Jun 5, 2026cs.LG

Accelerated Decentralized Stochastic Gradient Descent for Strongly Convex Optimization

Decentralized stochastic optimization is a fundamental paradigm for large-scale learning over networks, where agents communicate only with their neighbors and no central coordinator is required. For strongly convex problems, communication efficiency is mainly determined by the condition number κ=L/μκ=L/μ and the network spectral gap 1β1-β. Although deterministic decentralized methods can simultaneously achieve accelerated κ\sqrtκ and 1/1β1/\sqrt{1-β} dependences, no existing stochastic method attains both improvements at once. In this paper, we propose \emph{Multi-Gossip Accelerated DSGD} (MG-ADSGD), a decentralized stochastic algorithm that combines Nesterov-type primal--dual extrapolation with multi-round fast gossip averaging. The key idea is to couple the gossip depth with the mini-batch size so that additional communication rounds simultaneously improve consensus accuracy and reduce gradient variance. We show that MG-ADSGD achieves the communication complexity O~ ⁣(σ2μnεlog1ε+κ1βlog1ε),\widetilde{\mathcal O}\!\left( \frac{σ^2}{μnε}\log\frac{1}ε + \sqrt{\fracκ{1-β}}\log\frac{1}ε \right), where εε denotes the target accuracy, nn is the number of nodes, and σ2σ^2 is the gradient variance. To the best of our knowledge, this bound yields the best currently available communication complexity for decentralized stochastic strongly convex optimization, up to logarithmic factors that are independent of εε.
Ming Sun, Kun Yuan
Sep 16, 2026cs.LG

Revisiting Distributed Sign-Based Variance Reduction

Sign-based methods reduce communication costs in distributed environments, but aggregating local signs can introduce bias when data are heterogeneous. As a result, existing sign-based variance reduction methods fail to obtain the optimal convergence rates. In this paper, we solve this problem and obtain optimal rates for both nonconvex stochastic and finite-sum optimization. We first give a counterexample showing that majority voting can fail to approach stationary points even with exact local gradients. Motivated by this limitation, we propose tracking the global gradient at the server through unbiased compression of recursive gradient increments. As a result, we can obtain the convergence rates of O(d/K+d(a/(nK))1/3)O(\sqrt{d/K}+\sqrt d (a/(nK))^{1/3}) for the 1\ell_1-norm and O(a/K+a/(nK)1/3)O(\sqrt{a/K}+\sqrt a/(nK)^{1/3}) for the 2\ell_2-norm. Here, KK is the iteration number, nn is the number of workers, dd is the dimension, and a=1+ωa=1+ω, with ωω denoting the compressor's relative variance. For finite-sum problems with MM components, we combine periodic exact gradient refreshes with compressed component-gradient differences. The resulting total sample complexities are O(M+daMε2)O(M+d\sqrt{aM}ε^{-2}) and O(M+aM epsilon2)O(M+a\sqrt M\ epsilon^{-2}) for 1\ell_1 and 2\ell_2 gradient norms at most εε, matching the corresponding bounds in centralized settings.
Wei Jiang, Zechao Li, Lijun Zhang